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 RequestPolicyResponsePolicyZ explaincli)Zrequest_policyZresponse_policyZsuglenZprofileZ dropindexZ aliasupdateZalterZ aggregateZsyndumpcreateexplainZsuggetZdictdelZaliasaddZdictaddZ synupdateZdropinfo)ZsugaddZdictdumpcursorsearchZtagvalsZaliasdelZsugdelZ 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. NselfrrrN/home/app/Keep/.python/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. Nrrrrrr with_fallbackrzPolicyResolver.with_fallbackNrrrr __name__ __module__ __qualname__rstrrrrr!rrrrr|  rc@r) AsyncPolicyResolverrrcsdSrrrrrrrs zAsyncPolicyResolver.resolvercCr)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. Nrr rrrr!rz!AsyncPolicyResolver.with_fallbackNrr)rr)r#rrrrr)r(r)c@ReZdZdZ ddedeeddfddZdedee fd d Z e dd d Z dS)BasePolicyResolverz* Base class for policy resolvers. NpoliciesrrcC||_||_dSN _policies _fallbackrr-rrrr__init__ zBasePolicyResolver.__init__rcCs|d}t|dkrtd|t|dkr|nd|df\}}|j|ddur8|jdur6|j|SdS|j||ddurQ|jdurO|j|SdS|j||SN.zWrong command or module name: rrsplitlen ValueErrorr1getr2rrrpartsmodulerrrrrs       zBasePolicyResolver.resolvercCdSr/rr rrrr!z BasePolicyResolver.with_fallbackr/r") r$r%r&__doc__rrrr4r'rrrr!rrrrr, r,c@r+)AsyncBasePolicyResolverz0 Async base class for policy resolvers. Nr-rrcCr.r/r0r3rrrr4r5z AsyncBasePolicyResolver.__init__rcs|d}t|dkrtd|t|dkr|nd|df\}}|j|ddur<|jdur:|j|IdHSdS|j||ddurX|jdurV|j|IdHSdS|j||Sr6r9r>rrrrs     zAsyncBasePolicyResolver.resolver)cCrAr/rr rrrr!rBz%AsyncBasePolicyResolver.with_fallbackr/r*) r$r%r&rCrrr)r4r'rrrr!rrrrrErDrEc@eZdZdZ d dedeeddffdd Zd d d ZZ S) DynamicPolicyResolverzB Resolves policy dynamically based on the COMMAND output. Ncommands_parserrrcs||_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_parsersuperr4Zget_command_policies)rrHr __class__rrr4s zDynamicPolicyResolver.__init__rcC t|j|Sr/)rGrIr rrrr! z#DynamicPolicyResolver.with_fallbackr/r") r$r%r&rCrrrr4r! __classcell__rrrKrrGs rGc:eZdZdZd deeddffdd Zd dd ZZS) StaticPolicyResolverz? Resolves policy from a static list of policy records. Nrrctt|dS)z Parameters: fallback (Optional[PolicyResolver]): An optional fallback policy resolver used for resolving policies if static policies are inadequate. NrJr4rr rKrrr4zStaticPolicyResolver.__init__rcCt|Sr/)rQr rrrr!"z"StaticPolicyResolver.with_fallbackr/r") r$r%r&rCrrr4r!rOrrrKrrQrQcrF) AsyncDynamicPolicyResolverz1 Async version of DynamicPolicyResolver. Npolicy_recordsrrcst||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)rJr4)rrYrrKrrr4+s z#AsyncDynamicPolicyResolver.__init__r)cCrMr/)rXr1r rrrr!8rNz(AsyncDynamicPolicyResolver.with_fallbackr/r*) r$r%r&rCrrr)r4r!rOrrrKrrX&s rXcrP) AsyncStaticPolicyResolverz0 Async version of StaticPolicyResolver. NrrcrR)z Parameters: fallback (Optional[AsyncPolicyResolver]): An optional fallback policy resolver used for resolving policies if static policies are inadequate. NrSr rKrrr4ArTz"AsyncStaticPolicyResolver.__init__r)cCrUr/)rZr rrrr!IrVz'AsyncStaticPolicyResolver.with_fallbackr/r*) r$r%r&rCrr)r4r!rOrrrKrrZ<rWrZN)abcrrtypingrZredis._parsers.commandsrrrrr ZDEFAULT_KEYLESSZ DEFAULT_KEYEDZSPECIALr__annotations__rr)r,rErGrQrXrZrrrrs    !%)-159=AEgp&&