U ]g@sddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZddlmZGdd d e ZdS) N) ModuleType)cast)ConfigurationError)Storage) CallableListMemcachedClientPOptionalPRTupleTypeUnion)get_dependencycs.eZdZdZdgZdgZd$eeeee ge ffddfdd Z e ee eee ed ffd d d Zeeeeefee d ddZe eefejejedddZe e d ddZeedddZeddddZd%eeeeedddZeedddZed d d!Ze ed d"d#Z!Z"S)&MemcachedStoragez[ Rate limit storage with memcached as backend. Depends on :pypi:`pymemcache`. Z memcached pymemcacheFN)uriwrap_exceptionsoptionsreturnc stj|}g|_|jdD],}|s,q"|d\}}|j|t|fq"|j rl|jsl|j sl|j g|_|j dj |_ t|dd|_t|dd|_tttttttfgtf|d|j|_||_t|jstd |jt|_d |j_ t!j"||d d S) an :param uri: memcached location of the form ``memcached://host:port,host:port``, ``memcached:///var/tmp/path/to/sock`` :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. :param options: all remaining keyword arguments are passed directly to the constructor of :class:`pymemcache.client.base.PooledClient` or :class:`pymemcache.client.hash.HashClient` (if there are more than one hosts specified) :raise ConfigurationError: when :pypi:`pymemcache` is not available ,:rlibraryzpymemcache.clientcluster_libraryzpymemcache.client.hash client_getterz7memcached prerequisite not available. please install %sN)r)#urllibparseurlparsehostsnetlocstripsplitappendintpathportZ dependenciesmodule dependencystrpoprrrrrrr r get_clientrrrr threadinglocal local_storagestoragesuper__init__)selfrrrparsedlochostr% __class__rrrrrrrr)r1r'r7r7r8r.{szMemcachedStorage.storage)keyrcCst|j|pdS)zB :param key: the key to get the counter value for r)r#r.getr1rGr7r7r8rHszMemcachedStorage.getcCs|j|dS)z> :param key: the key to clear rate limits for N)r.deleterIr7r7r8clearszMemcachedStorage.clearr=)rGexpiryelastic_expiryamountrcCs|j|jj|||ddsh|j||p(|}|rd||jj|||j|jj|d|t|dd|S|j|jj|d|t|dd|S)aD increments the counter for a given rate limit key :param key: the key to increment :param expiry: amount in seconds for the key to expire in :param elastic_expiry: whether to keep extending the rate limit window every hit. :param amount: the number to increment by F)rA/expires)ZexpirerA)rEr.addincrtouchsettime)r1rGrLrMrNvaluer7r7r8rQs4   zMemcachedStorage.incrcCs tt|j|dptS)z; :param key: the key to get the expiry for rO)r#floatr.rHrTrIr7r7r8 get_expiryszMemcachedStorage.get_expirycCs*z||jjdWdSYdSXdS)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` z limiter-checkTFN)rEr.rHr:r7r7r8checks zMemcachedStorage.checkcCstdSr9)NotImplementedErrorr:r7r7r8resetszMemcachedStorage.reset)F)Fr=)#__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEZ DEPENDENCIESr(boolrrrr0propertyr Exceptionr r;rrr#r*r r r@r<rEr.rHrKrQrWrXr rZ __classcell__r7r7r5r8rsN4    ( r)rBr+rT urllib.parsertypesrtypingrZ limits.errorsrZlimits.storage.baserZ limits.typingrrrr r r r r rZ limits.utilrrr7r7r7r8s    ,