U ]g@sbddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z eddGdddeZ dS) N) versionadded)Storage)EmcacheClientPOptionalTupleTypeUnionz2.1)versioncseZdZdZdgZdgZdeeee eefddfdd Z e ee e ee e d ffd d d Zed d dZeedddZeddddZdeeeeedddZeedddZed ddZeed ddZZS) MemcachedStoragezW Rate limit storage with memcached as backend. Depends on :pypi:`emcache` zasync+memcachedemcacheFN)uriwrap_exceptionsoptionsreturnc stj|}g|_dd|jdDD]\}}|j|t|fq,||_ d|_ t j |fd|i||j dj|_dS)a :param uri: memcached location of the form ``async+memcached://host:port,host:port`` :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:`emcache.Client` :raise ConfigurationError: when :pypi:`emcache` is not available css |]}|r|dVqdS):N)stripsplit).0locr@/tmp/pip-unpacked-wheel-kizsipjx/limits/aio/storage/memcached.py )sz,MemcachedStorage.__init__..,Nr r )urllibparseurlparsehostsnetlocrrappendint_options_storagesuper__init__Z dependenciesmodule dependency)selfr r rparsedhostport __class__rrr#s zMemcachedStorage.__init__.)rcCs|jj|jjfSN)r%ZClusterNoAvailableNodes CommandErrorr&rrrbase_exceptions3sz MemcachedStorage.base_exceptionscs@js0jjfddjDfjIdH_js:tjS)Ncsg|]\}}j||qSr)r%ZMemcachedHostAddress)rhpr.rr ?sz0MemcachedStorage.get_storage..)r!r%Z create_clientrr AssertionErrorr.rr.r get_storage<s  zMemcachedStorage.get_storage)keyrcs2|IdH|dIdH}|r.t|jp0dS)zB :param key: the key to get the counter value for Nutf-8r)r4getencodervalue)r&r5itemrrrr7Es zMemcachedStorage.getcs$|IdH|dIdHdS)z> :param key: the key to clear rate limits for Nr6)r4deleter8)r&r5rrrclearNszMemcachedStorage.clear)r5expiryelastic_expiryamountrc s|IdH}|d}|d}d}z |j|||dIdHWn*|jjk rtd}|IdH}YnX|s|||IdHp|} |r|j||dIdH|j|t|t d|ddIdH| S|j|t|t d|ddIdH|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 Nr6/expiresT)exptimeF)rBZnoreply) r4r8addr%ZNotStoredStorageCommandError incrementtouchsetstrtime) r&r5r>r?r@storageZ limit_keyZ expire_keyaddedr9rrrincrTs6     zMemcachedStorage.incrcsB|IdH}||dIdH}t|r8t|jp>tS)z; :param key: the key to get the expiry for NrA)r4r7r8rfloatr9rH)r&r5rIr:rrr get_expiryszMemcachedStorage.get_expirycs8z$|IdH}|dIdHWdSYdSXdS)zq Check if storage is healthy by calling the ``get`` command on the key ``limiter-check`` Ns limiter-checkTF)r4r7)r&rIrrrchecks zMemcachedStorage.checkcstdSr,)NotImplementedErrorr.rrrresetszMemcachedStorage.reset)F)Fr=)__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEZ DEPENDENCIESrGboolrrLr#propertyr Exceptionrr/rr4rr7r<rKrMrNrrP __classcell__rrr*rr s8    -  r )rH urllib.parserZdeprecated.sphinxrZlimits.aio.storage.baserZ limits.typingrrrrrr rrrrs