hfCdZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z  dd lZdd lmZn#e$rd xZZYnwxYwd Zed d Ze eZGdde Zd S)z"AWS DynamoDB result store backend.) namedtuple)sleeptime) _parse_url)ImproperlyConfigured) get_logger)KeyValueStoreBackendN) ClientError)DynamoDBBackendDynamoDBAttributename data_typec&eZdZdZdZdZdZdZdZdZ dZ e ddZ e d d Z e d d Ze d d ZdZd!fd Zd!dZdZdZdZdZdZdZdZd"dZdZdZdZedZdZ dZ!dZ"d Z#xZ$S)#r zAWS DynamoDB result backend. Raises: celery.exceptions.ImproperlyConfigured: if module :pypi:`boto3` is not available. celeryr NTidSrresultB timestampNttlctj|i|||_|p|j|_tst dd}d}d}|{t |\}} } } } } }| }| }|du}|du}||krt d|}| dkrDd| |_d|_t d |jn| |_|j j j}|d}|r||_t|d |j|_t|d |j|_|d |j}|rF t||_n0#t&$r#}td | |d}~wwxYw| p|j|_|j|j|jf|_d|_|r|||dSdS)NzBYou need to install the boto3 library to use the DynamoDB backend.Fz6You need to specify both the Access Key ID and Secret. localhostzhttp://localhost:z us-east-1z*Using local-only DynamoDB endpoint URL: {}dynamodb_endpoint_urlreadwrite ttl_secondsz!TTL must be a number; got "{ttl}")exc_info) access_key_idsecret_access_key)super__init__url table_nameboto3r parse_url 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__s T/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/celery/backends/dynamodb.pyr$zDynamoDBBackend.__init__>s$)&)))$7 %&$%% %!&  $ ?# CFFD(HeU!) $, !0< 4D@ #333*"###%5 !$$$>$>$>!"-@GG) #)8=$D"&$'>"?"? " 8$7!'* ,((D $ ), -))D %))M4+DEEC 03CD--!LL;!"!G $6tDO O    !"       /"7        s7F F9F44F9cT|jd|ji}||||d|j |j|d<t j di||_||(|| |jS)zGet client connection.N region_name)rArBr)dynamodb)rS) r;r*updater)r'client_get_or_create_table_has_ttl_validate_ttl_methods_set_table_ttl)r=r!r"client_parameterss rPr<zDynamoDBBackend._get_clients < t! (!(()6->**  ,484E!.1 <#DL  % % ' ' '}}***,,,##%%%|c|jj|jjdg|j|jjddg|j|jddS)z=Get the boto3 structure describing the DynamoDB table schema.) AttributeName AttributeTypeHASH)r]KeyType)ReadCapacityUnitsWriteCapacityUnits)AttributeDefinitions TableName KeySchemaProvisionedThroughput)r7rrr&r2r3r=s rP_get_table_schemaz!DynamoDBBackend._get_table_schemasm &*_%9%)_%>% &*_%9%&*%=&*&?&&   r[c|} |jjd i|}td|j|dtd|j|S#t$rS}|j d dd}|dkr%|j |jcYd }~S|d }~wwxYw) z=Create table if not exists, otherwise return the description.z*DynamoDB Table {} did not exist, creating.ACTIVEz#DynamoDB Table {} is now available.ErrorCodeUnknownResourceInUseExceptionrdN) rhr; create_tabler+infor-r&_wait_for_table_statusr responser0describe_table)r= table_schematable_descriptionrN error_codes rPrVz$DynamoDBBackend._get_or_create_tables:--//   9 9 I IL I I  KK<CCO     ' ' 1 1 1 KK5<<O    % $   G,00CCJ555|22"o3 s%B B## D-AC;3D9C;;Dc*|jdn |jdkS)zReturn the desired Time to Live config. - True: Enable TTL on the table; use expiry. - False: Disable TTL on the table; don't use expiry. - None: Ignore TTL on the table; don't use expiry. Nr)r4rgs rPrWzDynamoDBBackend._has_ttls!08tt*a/ 0r[cxd}g}t|D],}t|j|s||-|rwtdd|tdd|dS)z:Verify boto support for the DynamoDB Time to Live methods.)update_time_to_livedescribe_time_to_livezdboto3 method(s) {methods} not found; ensure that boto3>=1.9.178 and botocore>=1.12.178 are installed,)methodsz#boto3 method(s) {methods} not foundN) listhasattrr;appendr+r6r-joinAttributeError)r=required_methodsmissing_methodsmethods rPrXz%DynamoDBBackend._validate_ttl_methodss  +,, / /F4<00 /&&v...   LLJ&HH_55    !5<<HH_55=   r[c@|j||ddS)zBGet the boto3 structure describing the DynamoDB TTL specification.)Enabledr])rdTimeToLiveSpecification)r&rW)r= ttl_attr_names rP_get_ttl_specificationz&DynamoDBBackend._get_ttl_specification s0==??!.((   r[c` |j|j}n#t$r~}|jddd}|jddd}t d|j|||d}~wwxYw|S)NrorkrlrmMessagezJError describing Time to Live on DynamoDB table {table}: {code}: {message})rHcodemessage) r;r|r&r rtr0r+r6r-)r= descriptionrNrx error_messages rP_get_table_ttl_descriptionz*DynamoDBBackend._get_table_ttl_descriptions ,<</=KK   G,00CCJJw/33IyIIM LL$fo%   G s # B+A9B&&B+c |}|dd}|dvrq|dd}|rN||jjkr>td|dkrdnd|j |Sn|d vrS|s>td|d krd nd |j |Sn4td||j|dkr|n |jj} |j j di| |}t d|j||jj|S#t$r}|jddd}|jddd}td|rdnd|j|||d}~wwxYw)z,Enable or disable Time to Live on the table.TimeToLiveDescriptionTimeToLiveStatus)ENABLEDENABLINGr]z5DynamoDB Time to Live is {situation} on table {table}rzalready enabledzcurrently being enabled) situationrH)DISABLED DISABLINGrzalready disabledzcurrently being disabledzWUnknown DynamoDB Time to Live status {status} on table {table}. Attempting to continue.)statusrH)rzUDynamoDB table Time to Live updated: table={table} enabled={enabled} attribute={attr})rHenabledattrrkrlrmrzHError {action} Time to Live on DynamoDB table {table}: {code}: {message}enabling disabling)actionrHrrNrp)rrW _ttl_fieldrr+debugr-r&r,r;r{rrrr rtr0r6) r=rr cur_attr_name attr_name specificationrNrxrs rPrYzDynamoDBBackend._set_table_ttl+s5577 456HI , , ,34_E }} ' DO$888LL+f!Y..#4"36"o '& 0 0 0==?? # '&++103/ #" #$ NN<fo   L$y00MMdo6J  z1DynamoDBBackend._item_to_dict..s@    J V,UZ8I   r[)r:)r=rs `rP _item_to_dictzDynamoDBBackend._item_to_dictsA  % %I    /    r[c*|S)N)r<rgs rPrUzDynamoDBBackend.clients!!!r[ct|}||}|jjdi|}||}||jjSNrp)rrrUget_itemrr0r8r)r=rrequest_parameters item_responseitems rPr0zDynamoDBBackend.getsg#hh!66s;;, ,BB/ABB !!-00xx).///r[ctt|}|||}|jjdi|dSr)rrrUput_item)r=rrrs rPsetzDynamoDBBackend.setsE#hh!66sEBB 22122222r[c fd|DS)Nc:g|]}|Srp)r0)rrr=s rP z(DynamoDBBackend.mget..s#...# ...r[rp)r=keyss` rPmgetzDynamoDBBackend.mgets........r[crt|}||}|jjdi|dSr)rrrU delete_item)r=rrs rPdeletezDynamoDBBackend.deletesC#hh!66s;; 55"455555r[)NN)rj)%__name__ __module__ __qualname____doc__r&r2r3r*r)r4supports_autoexpirer r7r8r9rr:r$r<rhrVrWrXrrrYrsrrrpropertyrUr0rrr __classcell__)rOs@rPr r sJJL ""<<rs4((""""""333333222222''''''&&&&&&LLL///////EKKK J24IJJ H  U6U6U6U6U6*U6U6U6U6U6s 5 AA