o ;h4K@sddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z ddl m Z mZmZddlmZe dZedZed d ed d d Gddde e e ZdS)) annotationsN) versionaddedversionchanged)MovingWindowSupportSlidingWindowCounterSupportStorage) ParamSpecTypeVarcast)get_dependencyPRz2.1)versionz3.14.0zEAdded option to select custom collection names for windows & counters)rreasoncseZdZdZddgZ ddgZ    d@dAfdd ZedBddZeddZ dCddZ dDdd Z dEd"d#Z dFd%d&Z dGd(d)ZdHdId-d.ZdJd/d0ZdKd3d4Z *dHdLd5d6Z *dHdLd7d8ZdMd:d;ZdNdd?ZZS)OMongoDBStoragezS Rate limit storage with MongoDB as backend. Depends on :pypi:`motor` async+mongodbzasync+mongodb+srvmotor.motor_asynciopymongolimitscounterswindowsFuristr database_namecounter_collection_namewindow_collection_namewrap_exceptionsbooloptionsfloat | str | boolreturnNonec s|ddd}tj|fd|i||jd|_|jd|_td\|_}|jjj |fi||_ t j |j _ ||_||d|_d |_d S) a :param uri: uri of the form ``async+mongodb://[user:password]@host:port?...``, This uri is passed directly to :class:`~motor.motor_asyncio.AsyncIOMotorClient` :param database_name: The database to use for storing the rate limit collections. :param counter_collection_name: The collection name to use for individual counters used in fixed window strategies :param window_collection_name: The collection name to use for sliding & moving window storage :param wrap_exceptions: Whether to wrap storage exceptions in :exc:`limits.errors.StorageError` before raising it. :param options: all remaining keyword arguments are passed to the constructor of :class:`~motor.motor_asyncio.AsyncIOMotorClient` :raise ConfigurationError: when the :pypi:`motor` or :pypi:`pymongo` are not available rZmongodbrrrzpymongo.errors)rrFN)replacesuper__init__ dependencies dependencyproxy_dependencyr lib_errorsmoduleZAsyncIOMotorClientstorageasyncioZget_running_loopZ get_io_loop_MongoDBStorage__database_name#_MongoDBStorage__collection_mapping _MongoDBStorage__indices_created)selfrrrrrr_ __class__M/var/www/html/venv/lib/python3.10/site-packages/limits/aio/storage/mongodb.pyr%,s    zMongoDBStorage.__init__-type[Exception] | tuple[type[Exception], ...]cCs|jjSN)r)Z PyMongoErrorr0r4r4r5base_exceptions[szMongoDBStorage.base_exceptionscCs|j|jSr7)r+Z get_databaser-r8r4r4r5databaseaszMongoDBStorage.databasecsP|js#t|j|jdjddd|j|jdjdddIdHd|_dS)NrexpireAtr)ZexpireAfterSecondsrT)r/r,gatherr:r.Z create_indexr8r4r4r5create_indiceses zMongoDBStorage.create_indices int | Nonecsxtt|j|jdi|j|jdiIdH}t|j|jd|j|jdIdHtt|S)z^ Delete all rate limit keys in the rate limit collections (counters, windows) rrN) sumr,r<r:r.Zcount_documentsZdropr int)r0Znum_keysr4r4r5resetqs zMongoDBStorage.resetkeycsDt|j|jdd|i|j|jdd|iIdHdS)z> :param key: the key to clear rate limits for r_idrN)r,r<r:r.Zfind_one_and_delete)r0rBr4r4r5clearszMongoDBStorage.clearfloatcsH|j|jdd|iIdH}|r|dntjjtjjdS)z; :param key: the key to get the expiry for rrCNr;tzinfo) r:r.find_onedatetimenowr#timezoneutc timestampr0rBZcounterr4r4r5 get_expirys zMongoDBStorage.get_expiryr@csJ|j|jdj|dtjtjjiddgdIdH}|r#|dp$dS)zB :param key: the key to get the counter value for r$gte)rCr;count) projectionNr)r:r.rHrIrJrKrLrNr4r4r5gets zMongoDBStorage.getr"expiryamountc s|IdHtjtjjtj|d}|j|jdjd|iddddd gi|d d |gid idddd gi|dd id igddg|j j j j dIdH}t |dS)z 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 amount: the number to increment by NsecondsrrC$set$condz$lt $expireAt$$NOW$addz$countifZthenelse)rQr;TrQ)upsertrRreturn_document)r=rIrJrKrL timedeltar:r.find_one_and_updater(r*ReturnDocumentAFTERr@)r0rBrTrUZ expirationresponser4r4r5incrs>     zMongoDBStorage.incrcs&z |jIdHWdSYdS)z{ Check if storage is healthy by calling :meth:`motor.motor_asyncio.AsyncIOMotorClient.server_info` NTF)r+Z server_infor8r4r4r5checks zMongoDBStorage.checklimittuple[float, int]c st}|j|jddd|iidddddd d ||gid iiidd d idd idigjddIdH}rE|dd|ddfS|dfS)z returns the starting point and the number of entries in the moving window :param str key: rate limit key :param int expiry: expiry of entry :return: (start of window, number of acquired entries) rz$matchrCz$projectZfilteredEntriesz$filterz$entriesentryrPz$$entry)inputasZcondz$minz$filteredEntriesz$size)minrQr")lengthNrrnrQ)timer:r.Z aggregateZto_list)r0rBrirTrMresultr4r4r5get_moving_windows6   z MongoDBStorage.get_moving_windowcs|IdH||krdSt}z?d|g|d|didtjtjjtj|did}|j|jd j d |d ||d d ||iii|ddIdHWdS|j j j j y_YdSw)z :param key: rate limit key to acquire an entry in :param limit: amount of entries allowed :param expiry: expiry of the entry :param amount: the number of entries to acquire NFentriesr)z$eachz $positionz$slicer;rV)z$pushrXrrCzentries.z$notrPT)r`)r=rprIrJrKrLrbr:r.Z update_oner(r*errorsZDuplicateKeyError)r0rBrirTrUrMZupdatesr4r4r5 acquire_entrys<    zMongoDBStorage.acquire_entrycs|IdH|d}|j|jdjd|idddddd d gi|gid d d gid dd gidiiiddddd d gi|gid d d d gididddd d gi|giddd d gidd |gidd d|gidid dididdddddddd dd dd |gigigi|gigid giiiidddddd|gi|gidd |gid diiiddddd|gi|giiiddgig|jjjjddIdH}tt |dS)NrrCrX previousCountrY$lte $subtractrZr[$ifNull $currentCountr$previousCountr]z$gtr\) currentCountr;ZcurWeightedCountz$floorz $multiplyz$dividez$maxr~z$curWeightedCountZ _acquiredz$unsetT)rar`) r=r:r.rcr(r*rdrer r)r0rBrirTrU expiry_msrqr4r4r5acquire_sliding_window_entry?s         #&   q yz+MongoDBStorage.acquire_sliding_window_entrytuple[int, float, int, float]c s2|d}|j|jdjd|idddddd gi|gid d d gid d d gididdddd gi|gid d d d gididdddd gi|gidd|giddidig|jjjjgddIdH}r|drs|djt j j d nt }td |t }td |dr||nd }|d||d|fSdS)NrvrrCrXrYrxryrZr[rzr{rr|r]r\)rwr~r;)r~rwr;)rarRr;rFrwr~)rrr)r:r.rcr(r*rdrerSr#rIrKrLrMrpmax)r0rBrTrrqZ expires_atZ current_ttlZprev_ttlr4r4r5get_sliding_windowsp          *3z!MongoDBStorage.get_sliding_windowcs||IdHSr7)rD)r0rBrTr4r4r5clear_sliding_windowsz#MongoDBStorage.clear_sliding_windowcCs|jo |jdSdSr7)r+closer8r4r4r5__del__szMongoDBStorage.__del__)rrrF)rrrrrrrrrrrrr r!)r r6)r r!)r r>)rBrr r!)rBrr rE)rBrr r@)r")rBrrTr@rUr@r r@)r r)rBrrir@rTr@r rj) rBrrir@rTr@rUr@r r)rBrrTr@r r)rBrrTr@r r!)__name__ __module__ __qualname____doc__ZSTORAGE_SCHEMEZ DEPENDENCIESr%propertyr9r:r=rArDrOrSrgrhrrrurrrr __classcell__r4r4r2r5rs</      - + 1   Er) __future__rr,rIrpZdeprecated.sphinxrrZlimits.aio.storage.baserrrZ limits.typingrr r Z limits.utilr r r rr4r4r4r5s