o h|L@sdZddlmZddlmZddlmZmZddlmZm Z ddl m Z ddl mZddlmZd d lmZz dd lZdd lmZWn eyOd ZZYnwd ZeddZeeZGdddeZd S)z"AWS DynamoDB result store backend.) namedtuple) ip_address)sleeptime)AnyDict) _parse_url)ImproperlyConfigured) get_logger)KeyValueStoreBackendN) ClientError)DynamoDBBackendDynamoDBAttributename data_typecspeZdZdZdZdZdZdZdZdZ dZ e dddZ e d d dZ e d d dZe d d dZe dd dZdZdZd@fdd ZeddZd@ddZddZddZddZddZddZdd Zd!d"ZdAd$d%Zd&d'Zd(d)Z d*e!d+e"e!e#ffd,d-Z$d*e!d+e"e!e#ffd.d/Z%d0d1Z&e'd2d3Z(d4d5Z)d6d7Z*d8d9Z+d:d;Z,d*e-d+e.fdd?Z0Z1S)BrzAWS DynamoDB result backend. Raises: celery.exceptions.ImproperlyConfigured: if module :pypi:`boto3` is not available. celeryr NTidSrresultB chord_countN timestampttlc stj|i|||_|p|j|_tstdd}d}d}|durt|\}} } } } } }| }| }|du}|du}||krAtd|}| dksLt| rbd| d| |_ d|_ t d |j n| |_ |jjj}|d }|rs||_ t|d |j|_t|d |j|_|d |j}|rzt||_Wnty}z t jd |d|d|d}~ww| p|j|_|j|j|jf|_d|_|r|j||ddSdS)NzBYou need to install the boto3 library to use the DynamoDB backend.Fz6You need to specify both the Access Key ID and Secret. localhostzhttp://:z us-east-1z*Using local-only DynamoDB endpoint URL: {}dynamodb_endpoint_urlreadwrite ttl_secondszTTL must be a number; got "")exc_info) access_key_idsecret_access_key)super__init__url table_nameboto3r parse_urlr _is_valid_ip endpoint_url aws_regionloggerwarningformatappconfgetintread_capacity_unitswrite_capacity_unitstime_to_live_seconds ValueErrorerror _key_field _value_field_timestamp_field_available_fields_client _get_client)selfr(r)argskwargsaws_credentials_givenaws_access_key_idaws_secret_access_keyschemeregionportusernamepasswordtablequeryaccess_key_givensecret_key_given_getconfig_endpoint_urlre __class__Z/var/www/html/optinet_system/venv/lib/python3.10/site-packages/celery/backends/dynamodb.pyr'Ds     zDynamoDBBackend.__init__cCs$zt|WdStyYdSw)NTF)rr9)iprUrUrVr,s  zDynamoDBBackend._is_valid_ipcCs~|jdur=1.9.178 and botocore>=1.12.178 are installed,)methodsz#boto3 method(s) {methods} not foundN) listhasattrr?appendr/r:r1joinAttributeError)rArequired_methodsmissing_methodsmethodrUrUrVr^s&   z%DynamoDBBackend._validate_ttl_methodscCs|j||ddS)zBGet the boto3 structure describing the DynamoDB TTL specification.)Enabledra)rhTimeToLiveSpecification)r)r])rA ttl_attr_namerUrUrV_get_ttl_specifications z&DynamoDBBackend._get_ttl_specificationc Cspz |jj|jd}W|Sty7}z |jddd}|jddd}tdj|j||d|d}~ww)NrmrnrorpMessagezJError describing Time to Live on DynamoDB table {table}: {code}: {message})rLcodemessage) r?r|r)r rtr4r/r:r1)rA descriptionrRry error_messagerUrUrV_get_table_ttl_description!s$z*DynamoDBBackend._get_table_ttl_descriptionc Csn|}|dd}|dvr2|dd}|r1||jjkr1tdj|dkr(dnd|jd |Sn'|d vrN|sMtdj|d krDd nd |jd |Sn tdj||jd|dkr_|n|jj}z |j j di|j |d}t dj|j||jjd|WSt y}z'|jddd}|jddd}tdj|rdnd|j||d|d}~ww)z,Enable or disable Time to Live on the table.TimeToLiveDescriptionTimeToLiveStatus)ENABLEDENABLINGraz5DynamoDB Time to Live is {situation} on table {table}rzalready enabledzcurrently being enabled) situationrL)DISABLED DISABLINGrzalready disabledzcurrently being disabledzWUnknown DynamoDB Time to Live status {status} on table {table}. Attempting to continue.)statusrL)rzUDynamoDB table Time to Live updated: table={table} enabled={enabled} attribute={attr})rLenabledattrrnrorprzHError {action} Time to Live on DynamoDB table {table}: {code}: {message}enabling disabling)actionrLrrNrU)rr] _ttl_fieldrr/debugr1r)r0r?r{rrvr rtr4r:) rArr cur_attr_name attr_name specificationrRryrrUrUrVr_6s     &   zDynamoDBBackend._set_table_ttlrrcCsTd}|s(|jj|jd}td|j||dd}||k}td|rdSdS)z#Poll for the expected table status.Frmz+Waiting for DynamoDB table {} to become {}.Table TableStatusr N)r[rsr)r/rr1r)rAexpectedachieved_staterzcurrent_statusrUrUrVrws  z&DynamoDBBackend._wait_for_table_statuscCs|j|jj|jj|iidS)z0Construct the item retrieval request parameters.)rhKey)r)r;rrrAkeyrUrUrV_prepare_get_requests z$DynamoDBBackend._prepare_get_requestc Cs~t}|j|jj|jj|i|jj|jj|i|jj|jjt|iid}|r=|d |j j|j jtt ||j ii|S)z/Construct the item creation request parameters.rhItemr) rr)r;rrr<r=strr]rZrr5r8)rArvaluer put_requestrUrUrV_prepare_put_requests* z$DynamoDBBackend._prepare_put_requestrreturnc CsFt}|j|jj|jj|i|jj|jjdi|jj|jjt|iidS)z7Construct the counter initialization request parameters0r)rr)r;rr _count_filedr=r)rArrrUrUrV_prepare_init_count_requests z+DynamoDBBackend._prepare_init_count_requestcCs@|j|jj|jj|iid|jjd|jjddddiiddS) z2Construct the counter increment request parameterszset z = z + :numz:numr1 UPDATED_NEW)rhrUpdateExpressionExpressionAttributeValues ReturnValues)r)r;rrrrrUrUrV_prepare_inc_count_requestsz*DynamoDBBackend._prepare_inc_count_requestcs dvriSfdd|jDS)z1Convert get_item() response to field-value pairs.rcs$i|]}|jd|j|jqS)rr).0field raw_responserUrV sz1DynamoDBBackend._item_to_dict..)r>)rArrUrrV _item_to_dicts  zDynamoDBBackend._item_to_dictcCs|S)N)r@rkrUrUrVr[szDynamoDBBackend.clientcCs<t|}||}|jjdi|}||}||jjSNrU)rrr[get_itemrr4r<r)rArrequest_parameters item_responseitemrUrUrVr4 s   zDynamoDBBackend.getcCs*t|}|||}|jjdi|dSr)rrr[put_item)rArrrrUrUrVsets zDynamoDBBackend.setcsfdd|DS)Ncsg|]}|qSrU)r4)rrrkrUrV sz(DynamoDBBackend.mget..rU)rAkeysrUrkrVmgetszDynamoDBBackend.mgetcCs(t|}||}|jjdi|dSr)rrr[ delete_item)rArrrUrUrVdeletes zDynamoDBBackend.deletecCsDt|}||}|jjdi|}|d|jj|jj}t|S)zimplements_incrr' staticmethodr,r@rlr\r]r^rrr_rwrrrrrrrrpropertyr[r4rrrbytesr5rr __classcell__rUrUrSrVrsR     Y     p  r)r collectionsr ipaddressrrrtypingrrkombu.utils.urlrr+celery.exceptionsr celery.utils.logr baser r*botocore.exceptionsr ImportError__all__rrr/rrUrUrUrVs&