o

wetN�@s0dZd
dl
mZ
d
dlmZ
Gdd�de�ZdS)z
Represents an SQS Queue
�)
�urllib)
�Messagec@sj
eZ
dZd
d
efdd�
Zdd�Zdd�Zee�
Zdd	�Z	ee	�
Z
d
d�Zee�
Zdd
�Z
dd�Zdd�ZdKdd�
Zdd�Zdd�Zdd�Zdd�Zdd�Z	
	
dLdd �
ZdMd!d"�
Zd#d$�ZdNd&d'�
Z	
	
	
dOd)d*�
Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�ZdPd6d7�
Z dPd8d9�
Z!dPd:d;�
Z"dQd=d>�
Z#dRd?d@�
Z$dRdAdB�
Z%e%Z&dCdD�Z'dMdEdF�
Z(dRdGdH�
Z)dRdIdJ�
Z*e*Z+d
S)S�QueueNcCs|
|_||_
||_d|_dS�
N)�
connection�url�
message_class�visibility_timeout)�selfrrr�r�G/home/arjun/projects/env/lib/python3.10/site-packages/boto/sqs/queue.py�__init__s




zQueue.__init__c

Cs
d
|jS)Nz	Queue(%s))
r�
r
rrr�__repr__%s

zQueue.__repr__c
Cs&|jrt
j�|j�
d
}
|
S|j}
|
S)N�)rr�parse�urlparse�r
�valrrr�_id(s


�
z	Queue._idc
Cs0|jrt
j�|j�
d
�d�
d
}
|
S|j}
|
S)Nr�
/)rrrr�splitrrrr�_name0s


�
zQueue._namec
CsB|j�
d
�
}
|jjjdkrd}nd}d||jjj|
d|
dfS)Nrz
cn-north-1zaws-cn�awszarn:%s:sqs:%s:%s:%s�
r)�idrr�region�name)r
�parts�	partitionrrr�_arn8s




�z
Queue._arnc
CsdSrr)r
r�attrsrrrr�startElementBs
zQueue.startElementcCs8|
d
kr	||_dS|
dkrt
|�
|_dSt||
|�
dS)N�QueueUrl�VisibilityTimeout)r�intr	�setattr)r
r�valuerrrr�
endElementEs





zQueue.endElementcCs
|
|_d
S)aq

        Set the message class that should be used when instantiating
        messages read from the queue.  By default, the class
        :class:`boto.sqs.message.Message` is used but this can be overriden
        with any class that behaves like a message.

        :type message_class: Message-like class
        :param message_class:  The new Message class
        N)
r)r
rrrr�set_message_classMs

zQueue.set_message_class�AllcC�|j�
||
�S)
a�
        Retrieves attributes about this queue object and returns
        them in an Attribute instance (subclass of a Dictionary).

        :type attributes: string
        :param attributes: String containing one of:
                           ApproximateNumberOfMessages,
                           ApproximateNumberOfMessagesNotVisible,
                           VisibilityTimeout,
                           CreatedTimestamp,
                           LastModifiedTimestamp,
                           Policy
                           ReceiveMessageWaitTimeSeconds
        :rtype: Attribute object
        :return: An Attribute object which is a mapping type holding the
                 requested name/value pairs
        )r�get_queue_attributes)r
�
attributesrrr�get_attributesYszQueue.get_attributescCs|j�
||
|�S)
a�

        Set a new value for an attribute of the Queue.

        :type attribute: String
        :param attribute: The name of the attribute you want to set.

        :param value: The new value for the attribute must be:


            * For `DelaySeconds` the value must be an integer number of
            seconds from 0 to 900 (15 minutes).
                >>> queue.set_attribute('DelaySeconds', 900)

            * For `MaximumMessageSize` the value must be an integer number of
            bytes from 1024 (1 KiB) to 262144 (256 KiB).
                >>> queue.set_attribute('MaximumMessageSize', 262144)

            * For `MessageRetentionPeriod` the value must be an integer number of
            seconds from 60 (1 minute) to 1209600 (14 days).
                >>> queue.set_attribute('MessageRetentionPeriod', 1209600)

            * For `Policy` the value must be an string that contains JSON formatted
            parameters and values.
                >>> queue.set_attribute('Policy', json.dumps({
                ...     'Version': '2008-10-17',
                ...     'Id': '/123456789012/testQueue/SQSDefaultPolicy',
                ...     'Statement': [
                ...        {
                ...            'Sid': 'Queue1ReceiveMessage',
                ...            'Effect': 'Allow',
                ...            'Principal': {
                ...                'AWS': '*'
                ...            },
                ...            'Action': 'SQS:ReceiveMessage',
                ...            'Resource': 'arn:aws:aws:sqs:us-east-1:123456789012:testQueue'
                ...        }
                ...    ]
                ... }))

            * For `ReceiveMessageWaitTimeSeconds` the value must be an integer number of
            seconds from 0 to 20.
                >>> queue.set_attribute('ReceiveMessageWaitTimeSeconds', 20)

            * For `VisibilityTimeout` the value must be an integer number of
            seconds from 0 to 43200 (12 hours).
                >>> queue.set_attribute('VisibilityTimeout', 43200)

            * For `RedrivePolicy` the value must be an string that contains JSON formatted
            parameters and values. You can set maxReceiveCount to a value between 1 and 1000.
            The deadLetterTargetArn value is the Amazon Resource Name (ARN) of the queue that
            will receive the dead letter messages.
                >>> queue.set_attribute('RedrivePolicy', json.dumps({
                ...    'maxReceiveCount': 5,
                ...    'deadLetterTargetArn': "arn:aws:aws:sqs:us-east-1:123456789012:testDeadLetterQueue"
                ... }))

        :rtype: bool
        :return: True if successful, otherwise False.
        )r�set_queue_attribute)r
�	attributer'rrr�
set_attributems<zQueue.set_attributec
Cs|�d
�
}
t
|
d
�
S)z�
        Get the visibility timeout for the queue.

        :rtype: int
        :return: The number of seconds as an integer.
        r$�r.r%)r
�
arrr�get_timeout�s

zQueue.get_timeoutcCs|�d
|
�}|r|
|_
|S)z�
        Set the visibility timeout for the queue.

        :type visibility_timeout: int
        :param visibility_timeout: The desired timeout in seconds
        r$)r1r	)r
r	�retvalrrr�set_timeout�s


zQueue.set_timeoutcCs|j�
||
||�S)
as
        Add a permission to a queue.

        :type label: str or unicode
        :param label: A unique identification of the permission you are setting.
            Maximum of 80 characters ``[0-9a-zA-Z_-]``
            Example, AliceSendMessage

        :type aws_account_id: str or unicode
        :param principal_id: The AWS account number of the principal who
            will be given permission.  The principal must have an AWS account,
            but does not need to be signed up for Amazon SQS. For information
            about locating the AWS account identification.

        :type action_name: str or unicode
        :param action_name: The action.  Valid choices are:
            SendMessage|ReceiveMessage|DeleteMessage|
            ChangeMessageVisibility|GetQueueAttributes|*

        :rtype: bool
        :return: True if successful, False otherwise.

        )r�add_permission)r
�label�aws_account_id�action_namerrrr7�s
�zQueue.add_permissioncCr+)
a

        Remove a permission from a queue.

        :type label: str or unicode
        :param label: The unique label associated with the permission
            being removed.

        :rtype: bool
        :return: True if successful, False otherwise.
        )r�remove_permission)r
r8rrrr;�szQueue.remove_permissioncCs*|jd
|
||d�}t
|�
d
kr|dSdS)aD
        Read a single message from the queue.

        :type visibility_timeout: int
        :param visibility_timeout: The timeout for this message in seconds

        :type wait_time_seconds: int
        :param wait_time_seconds: The duration (in seconds) for which the call
            will wait for a message to arrive in the queue before returning.
            If a message is available, the call will return sooner than
            wait_time_seconds.

        :type message_attributes: list
        :param message_attributes: The name(s) of additional message
            attributes to return. The default is to return no additional
            message attributes. Use ``['All']`` or ``['.*']`` to return all.

        :rtype: :class:`boto.sqs.message.Message`
        :return: A single message or None if queue is empty
        r)�wait_time_seconds�message_attributesr
N)�get_messages�len)r
r	r<r=�rsrrr�read�s

�
z
Queue.readcCs.|jj
||
��||
jd
�}|j|
_|j|
_|
S)a

        Add a single message to the queue.

        :type message: Message
        :param message: The message to be written to the queue

        :rtype: :class:`boto.sqs.message.Message`
        :return: The :class:`boto.sqs.message.Message` object that was written.
        )�
delay_secondsr=)r�send_message�get_body_encodedr=r�md5)r
�messagerB�new_msgrrr�write
s


�

zQueue.writecCr+)
a�
        Delivers up to 10 messages in a single request.

        :type messages: List of lists.
        :param messages: A list of lists or tuples.  Each inner
            tuple represents a single message to be written
            and consists of and ID (string) that must be unique
            within the list of messages, the message body itself
            which can be a maximum of 64K in length, an
            integer which represents the delay time (in seconds)
            for the message (0-900) before the message will
            be delivered to the queue, and an optional dict of
            message attributes like those passed to ``send_message``
            in the connection class.
        )r�send_message_batch�r
�messagesrrr�write_batch
szQueue.write_batch�cKs|j||
fi|�
�
}||_
|S)
z�
        Create new message of appropriate class.

        :type body: message body
        :param body: The body of the newly created message (optional).

        :rtype: :class:`boto.sqs.message.Message`
        :return: A new Message object
        )r�queue)r
�body�kwargs�
mrrr�new_message*
s


zQueue.new_messagercCs|jj
||
||||d
�S)a
        Get a variable number of messages.

        :type num_messages: int
        :param num_messages: The maximum number of messages to read from
            the queue.

        :type visibility_timeout: int
        :param visibility_timeout: The VisibilityTimeout for the messages read.

        :type attributes: str
        :param attributes: The name of additional attribute to return
            with response or All if you want all attributes.  The
            default is to return no additional attributes.  Valid
            values: All SenderId SentTimestamp ApproximateReceiveCount
            ApproximateFirstReceiveTimestamp

        :type wait_time_seconds: int
        :param wait_time_seconds: The duration (in seconds) for which the call
            will wait for a message to arrive in the queue before returning.
            If a message is available, the call will return sooner than
            wait_time_seconds.

        :type message_attributes: list
        :param message_attributes: The name(s) of additional message
            attributes to return. The default is to return no additional
            message attributes. Use ``['All']`` or ``['.*']`` to return all.

        :rtype: list
        :return: A list of :class:`boto.sqs.message.Message` objects.
        )�number_messagesr	r-r<r=)r�receive_message)r
�num_messagesr	r-r<r=rrrr>9
s"



�zQueue.get_messagescCr+)
a

        Delete a message from the queue.

        :type message: :class:`boto.sqs.message.Message`
        :param message: The :class:`boto.sqs.message.Message` object to delete.

        :rtype: bool
        :return: True if successful, False otherwise
        )r�delete_message)r
rFrrrrVa
�
zQueue.delete_messagecCr+)
z�
        Deletes a list of messages in a single request.

        :type messages: List of :class:`boto.sqs.message.Message` objects.
        :param messages: A list of message objects.
        )r�delete_message_batchrJrrrrXm
szQueue.delete_message_batchcCr+)
ar

        A batch version of change_message_visibility that can act
        on up to 10 messages at a time.

        :type messages: List of tuples.
        :param messages: A list of tuples where each tuple consists
            of a :class:`boto.sqs.message.Message` object and an integer
            that represents the new visibility timeout for that message.
        )r�change_message_visibility_batchrJrrrrYv
rWz%Queue.change_message_visibility_batchc

C�|j�
|�
S)
z#
        Delete the queue.
        )r�delete_queuerrrr�delete�
�zQueue.deletec

CrZ)
z2
        Purge all messages in the queue.
        )r�purge_queuerrrr�purge�
r]zQueue.purge�
cCs|��S)
z?Deprecated utility function to remove all messages from a queue)
r_)r
�	page_size�vtimeoutrrr�clear�
szQueue.clearcCs|�d
�
}t
|d
�
S)z�
        Utility function to count the number of messages in a queue.
        Note: This function now calls GetQueueAttributes to obtain
        an 'approximate' count of the number of messages in a queue.
        �ApproximateNumberOfMessagesr2)r
rarbr3rrr�count�
s

zQueue.countcCs:d
}|�|
|�}|r|D]}|d7}q|�|
|�}|s
|S)a�

        Deprecated.  This is the old 'count' method that actually counts
        the messages by reading them all.  This gives an accurate count but
        is very slow for queues with non-trivial number of messasges.
        Instead, use get_attributes('ApproximateNumberOfMessages') to take
        advantage of the new SQS capability.  This is retained only for
        the unit tests.
        r
r)
r>)r
rarb�
n�
lrQrrr�
count_slow�
s	





�zQueue.count_slow�

c	Csht|
d
�}d}|�
||�}|r.|D]}|�|���

|r!|�|�

|d7}q|�
||�}|s|��
|S)zoUtility function to dump the messages in a queue to a file
        NOTE: Page size must be < 10 else SQS errors�wbr
r)�openr>rH�get_body�close)	r
�	file_namerarb�sep�fprfrgrQrrr�dump�
s











�
z
Queue.dumpcCsNd
}|��}|r%|d7}|
�
|���

|r|
�
|�

|�|�

|��}|s|S)a:

        Read all messages from the queue and persist them to file-like object.
        Messages are written to the file and the 'sep' string is written
        in between messages.  Messages are deleted from the queue after
        being written to the file.
        Returns the number of messages saved.
        r
r)rArHrlrV)r
rprorfrQrrr�save_to_file�
s









�zQueue.save_to_filecC�"t|
d
�}|�
||�}|��
|S)a4

        Read all messages from the queue and persist them to local file.
        Messages are written to the file and the 'sep' string is written
        in between messages.  Messages are deleted from the queue after
        being written to the file.
        Returns the number of messages saved.
        rj)rkrrrm�r
rnrorprfrrr�save_to_filename�
s



zQueue.save_to_filenamecCsVd
}|��}|r)|d7}|
�
d|j|jf�
}|�|���

|�|�

|��}|s|S)a+

        Read all messages from the queue and persist them to S3.
        Messages are stored in the S3 bucket using a naming scheme of::

            <queue_id>/<message_id>

        Messages are deleted from the queue after being saved to S3.
        Returns the number of messages saved.
        r
rz%s/%s)rA�new_keyr�set_contents_from_stringrlrV)r
�bucketrfrQ�keyrrr�
save_to_s3�
s








�zQueue.save_to_s3cCs^d
}|r	d|}n	d|jdd�}|
j
|d�
}|D]}|d7}|�|���
}|�|�

q|S)z7
        Load messages previously saved to S3.
        r
z%s/rN)
�prefix)r�listrR�get_contents_as_stringrH)r
rxr{rfr@ryrQrrr�load_from_s3�
s








zQueue.load_from_s3cCsbd
}d}|
��}|r/||kr%t
||�}|�|�

|d7}td|�

d}n||}|
��}|s
|S)zDUtility function to load messages from a file-like object to a queuer
rMrzwriting message %d)�readlinerrH�print)r
rprorfrOrgrQrrr�load_from_files











�
zQueue.load_from_filecCrs)zBUtility function to load messages from a local filename to a queue�rb)rkr�rmrtrrr�load_from_filenames



zQueue.load_from_filename)
r*)NNNr)
rM)rNNNN)r`r`)r`r`ri)
ri),�__name__�
__module__�__qualname__rr
rr�propertyrrrr �arnr"r(r)r.r1r4r6r7r;rArHrLrRr>rVrXrYr\r_rcrerhrqrrru�saverzr~r�r��loadrrrrrsZ
>



�




�(	


	





rN)�__doc__�boto.compatr�boto.sqs.messager�objectrrrrr�<module>
s