o i, @sUddlmZmZddlmZddlmZmZmZm Z m Z idee j e j ddee j e j ddee j e j ddee j e j dd ee j e j dd ee j e j dd ee j e j dd ee j e j dd ee j e j ddee j e j ddee j e j ddee j e j ddee j e j ddee j e j ddee j e j ddee j e j ddee j e j dee j e j dee j e j dee j e j dee j e j dee j e j dee j e j dee j e j dee j e j dddee j e j didZeed<GdddeZGdddeZGdddeZGd d!d!eZGd"d#d#eZGd$d%d%eZGd&d'd'eZGd(d)d)eZd*S)+)ABCabstractmethod)Optional)CommandPoliciesCommandsParser PolicyRecords RequestPolicyResponsePolicy explaincli)request_policyresponse_policysuglenprofile dropindex aliasupdatealter aggregatesyndumpcreateexplainsuggetdictdelaliasadddictadd synupdatedropinfo)sugadddictdumpcursorsearchtagvalsaliasdelsugdel spellcheckcommand)ftcoreSTATIC_POLICIESc@4eZdZededeefddZed ddZdS) PolicyResolver command_namereturncCdSa  Resolves the command name and determines the associated command policies. Args: command_name: The name of the command to resolve. Returns: CommandPolicies: The policies associated with the specified command. Nselfr+r/r/K/home/ubuntu/.local/lib/python3.10/site-packages/redis/commands/policies.pyresolve} zPolicyResolver.resolvefallbackcCr-)a Factory method to instantiate a policy resolver with a fallback resolver. Args: fallback: Fallback resolver Returns: PolicyResolver: Returns a new policy resolver with the specified fallback resolver. Nr/r1r5r/r/r2 with_fallbackr4zPolicyResolver.with_fallbackNr5r*r,r* __name__ __module__ __qualname__rstrrrr3r7r/r/r/r2r*|  r*c@r)) AsyncPolicyResolverr+r,csdSr.r/r0r/r/r2r3s zAsyncPolicyResolver.resolver5cCr-)a Factory method to instantiate an async policy resolver with a fallback resolver. Args: fallback: Fallback resolver Returns: AsyncPolicyResolver: Returns a new policy resolver with the specified fallback resolver. Nr/r6r/r/r2r7r4z!AsyncPolicyResolver.with_fallbackNr5r?r,r?r9r/r/r/r2r?r>r?c@ReZdZdZ ddedeeddfddZdedee fd d Z e dd d Z dS)BasePolicyResolverz* Base class for policy resolvers. Npoliciesr5r,cC||_||_dSN _policies _fallbackr1rCr5r/r/r2__init__ zBasePolicyResolver.__init__r+cCs|d}t|dkrtd|t|dkr|nd|df\}}|j|ddur8|jdur6|j|SdS|j||ddurQ|jdurO|j|SdS|j||SN.zWrong command or module name: r'rsplitlen ValueErrorrGgetrHr3r1r+partsmoduler%r/r/r2r3s       zBasePolicyResolver.resolver*cCdSrEr/r6r/r/r2r7z BasePolicyResolver.with_fallbackrEr8) r:r;r<__doc__rrr*rJr=rr3rr7r/r/r/r2rB rBc@rA)AsyncBasePolicyResolverz0 Async base class for policy resolvers. NrCr5r,cCrDrErFrIr/r/r2rJrKz AsyncBasePolicyResolver.__init__r+cs|d}t|dkrtd|t|dkr|nd|df\}}|j|ddur<|jdur:|j|IdHSdS|j||ddurX|jdurV|j|IdHSdS|j||SrLrOrTr/r/r2r3s     zAsyncBasePolicyResolver.resolver?cCrWrEr/r6r/r/r2r7rXz%AsyncBasePolicyResolver.with_fallbackrEr@) r:r;r<rYrrr?rJr=rr3rr7r/r/r/r2r[rZr[c@eZdZdZ d dedeeddffdd Zd d d ZZ S) DynamicPolicyResolverzB Resolves policy dynamically based on the COMMAND output. Ncommands_parserr5r,cs||_t||dS)z Parameters: commands_parser (CommandsParser): COMMAND output parser. fallback (Optional[PolicyResolver]): An optional resolver to be used when the primary policies cannot handle a specific request. N)_commands_parsersuperrJget_command_policies)r1r^r5 __class__r/r2rJs zDynamicPolicyResolver.__init__r*cC t|j|SrE)r]r_r6r/r/r2r7 z#DynamicPolicyResolver.with_fallbackrEr8) r:r;r<rYrrr*rJr7 __classcell__r/r/rbr2r]s r]c:eZdZdZd deeddffdd Zd dd ZZS) StaticPolicyResolverz? Resolves policy from a static list of policy records. Nr5r,ctt|dS)z Parameters: fallback (Optional[PolicyResolver]): An optional fallback policy resolver used for resolving policies if static policies are inadequate. Nr`rJr(r6rbr/r2rJzStaticPolicyResolver.__init__r*cCt|SrE)rhr6r/r/r2r7"z"StaticPolicyResolver.with_fallbackrEr8) r:r;r<rYrr*rJr7rfr/r/rbr2rhrhcr\) AsyncDynamicPolicyResolverz1 Async version of DynamicPolicyResolver. Npolicy_recordsr5r,cst||dS)z Parameters: policy_records (PolicyRecords): Policy records. fallback (Optional[AsyncPolicyResolver]): An optional resolver to be used when the primary policies cannot handle a specific request. N)r`rJ)r1rpr5rbr/r2rJ+s z#AsyncDynamicPolicyResolver.__init__r?cCrdrE)rorGr6r/r/r2r78rez(AsyncDynamicPolicyResolver.with_fallbackrEr@) r:r;r<rYrrr?rJr7rfr/r/rbr2ro&s rocrg) AsyncStaticPolicyResolverz0 Async version of StaticPolicyResolver. Nr5r,cri)z Parameters: fallback (Optional[AsyncPolicyResolver]): An optional fallback policy resolver used for resolving policies if static policies are inadequate. Nrjr6rbr/r2rJArkz"AsyncStaticPolicyResolver.__init__r?cCrlrE)rqr6r/r/r2r7Irmz'AsyncStaticPolicyResolver.with_fallbackrEr@) r:r;r<rYrr?rJr7rfr/r/rbr2rq<rnrqN)abcrrtypingrredis._parsers.commandsrrrrr DEFAULT_KEYLESS DEFAULT_KEYEDSPECIALr(__annotations__r*r?rBr[r]rhrorqr/r/r/r2s    !%)-159=AEgp&&