a a*jG@sddlmZddlZddlZddlmZmZddlmZm Z ddl m Z m Z m Z mZmZmZmZddlmZdd lmZmZmZGd d d eeeeZed d e dddGdddeZdS)) annotationsN)ABCabstractmethod) versionaddedversionchanged) MongoClientMongoCollection MongoDatabaseOptionalTypeUnioncast)get_dependency)MovingWindowSupportSlidingWindowCounterSupportStoragec sreZdZdZdgZdBdddddd d d fd d ZeddddZeddddZeddddZ eddddZ e dd ddddZ edddd Z d dd!d"Zd#dd$d%Zdd d&d'd(Zdd)d&d*d+Zdd,d&d-d.ZdCdd,dd,d,d0d1d2Zddd3d4Zdd,d,d5d6d7d8ZdDdd,d,d,dd9d:d;Zdd,dd?ZdEdd,d,d,dd9d@dAZZS)FMongoDBStorageBasezV Rate limit storage with MongoDB as backend. Depends on :pypi:`pymongo`. pymongolimitscounterswindowsFstrboolUnion[int, str, bool]None)uri database_namecounter_collection_namewindow_collection_namewrap_exceptionsoptionsreturnc s\tj|fd|i|||_||d|_|jdj|_td\|_}||_ ||_ d|_ dS)ai :param uri: uri of the form ``mongodb://[user:password]@host:port?...``, This uri is passed directly to :class:`~pymongo.mongo_client.MongoClient` :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:`~pymongo.mongo_client.MongoClient` :raise ConfigurationError: when the :pypi:`pymongo` library is not available r!)rrrzpymongo.errorsN) super__init___database_name_collection_mapping dependenciesmodulelibr lib_errors _storage_uri_storage_options_storage)selfrrrr r!r"_ __class__R/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/storage/mongodb.pyr%"szMongoDBStorageBase.__init__r)r#cCs0|jdur*|j|jfi|j|_||jSN)r._init_mongo_clientr,r-(_MongoDBStorageBase__initialize_databaser/r3r3r4storageGs zMongoDBStorageBase.storager cCs |j|jSr5)r9r&r8r3r3r4 _databasePszMongoDBStorageBase._databasercCs|j|jdS)Nrr:r'r8r3r3r4rTszMongoDBStorageBase.counterscCs|j|jdS)Nrr;r8r3r3r4rXszMongoDBStorageBase.windows Optional[str]rr"r#cKs tdSr5)NotImplementedErrorr/rr"r3r3r4r6\sz%MongoDBStorageBase._init_mongo_clientz3Union[Type[Exception], tuple[Type[Exception], ...]]cCs|jjSr5)r+Z PyMongoErrorr8r3r3r4base_exceptionsbsz"MongoDBStorageBase.base_exceptionscCs$|jjddd|jjddddS)NexpireAtr)ZexpireAfterSeconds)rZ create_indexrr8r3r3r4Z__initialize_databasehsz(MongoDBStorageBase.__initialize_databasez Optional[int]cCs4|ji|ji}|j|jt|S)z^ Delete all rate limit keys in the rate limit collections (counters, windows) )rZcount_documentsrZdropint)r/Znum_keysr3r3r4resetls  zMongoDBStorageBase.reset)keyr#cCs$|jd|i|jd|idS)z> :param key: the key to clear rate limits for _idN)rZfind_one_and_deleter)r/rDr3r3r4clearvszMongoDBStorageBase.clearfloatcCs6|jd|i}|r|dntjjtjjdS)z; :param key: the key to get the expiry for rErAtzinfo)rfind_onedatetimenowreplacetimezoneutc timestampr/rDcounterr3r3r4 get_expiry}s zMongoDBStorageBase.get_expiryrBcCs8|jj|dtjtjjiddgd}|r4|dp6dS)zB :param key: the key to get the counter value for $gte)rErAcount) projectionr)rrJrKrLrNrOrQr3r3r4getszMongoDBStorageBase.getr)rDexpiryelastic_expiryamountr#c Cstjtjjtj|d}t|jjd|idddddgi|dd |gid iddddgi||r`|ndd id igd d g|jj j dd S)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 secondsrE$set$condz$ltz $expireAt$$NOW$addz$countifZthenelse)rUrATrU)upsertrVreturn_document) rKrLrNrO timedeltarBrfind_one_and_updater*ReturnDocumentAFTER)r/rDrXrYrZZ expirationr3r3r4incrs:     zMongoDBStorageBase.incrcCs$z|jWdSYdS0dS)zm Check if storage is healthy by calling :meth:`pymongo.mongo_client.MongoClient.server_info` TFN)r9Z server_infor8r3r3r4checks  zMongoDBStorageBase.checkztuple[float, int])rDlimitrXr#c Cst}t|jdd|iiddddddd ||gid iiid did d ddiddidig}|r||dd|ddfS|dfS)z returns the starting point and the number of entries in the moving window :param key: rate limit key :param expiry: expiry of entry :return: (start of window, number of acquired entries) z$matchrEz$projectentriesz$filterz$entriesentryrTz$$entry)inputasZcondz$unwindz$groupz$_idz$minz$sumr)rEminrUrrqrU)timelistrZ aggregate)r/rDrlrXrPresultr3r3r4get_moving_windows6   z$MongoDBStorageBase.get_moving_window)rDrlrXrZr#cCs||kr dSt}znd|g|d|didtjtjjtj|did}|jjd|d ||d d ||iii|d d Wd S|jj j yYdS0dS)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 Frmr)z$eachz $positionz$slicerAr[)z$pushr]rEzentries.z$notrTT)rdN) rrrKrLrNrOrfrZ update_oner*errorsZDuplicateKeyError)r/rDrlrXrZrPZupdatesr3r3r4 acquire_entrys6  z MongoDBStorageBase.acquire_entryztuple[int, float, int, float])rDrXr#c Cs"|d}|jjd|iddddddgi|gid d d gid d d gid idddddgi|gid d d d gid idddddgi|gidd|gidd idig|jjjgdd}r|dr|djtjj d nt }t d |t }t d |dr||nd }|d||d|fSdS)NrEr]r^$lte $subtract $expiresAtr_$ifNull $currentCountr$previousCountrar`) previousCount currentCount expiresAt)rrr)rerVrrHrr)rrr) rrgr*rhrirWrMrKrNrOrPrrmax)r/rDrX expiry_msrtZ expires_atZ current_ttlZprev_ttlr3r3r4get_sliding_window$sj        , 3z%MongoDBStorageBase.get_sliding_windowcCsl|d}|jjd|iddddddd gi|gid d d gid d d gidiiidddddd gi|gid d d d gididdddd gi|gidddd gidd|gidd d|gididdididddddd ddd dddd |gigigi|gigid giiiidddddd|gi|gidd |gid diiiddddd|gi|giiiddgig|jjjdd}tt|dS)NrxrEr]rr^ryrzr{r_r|r}rr~raz$gtr`r)rrZcurWeightedCountz$floorz $multiplyz$dividez$maxrz$curWeightedCountZ _acquiredz$unsetT)rerd)rrgr*rhrir r)r/rDrlrXrZrrtr3r3r4acquire_sliding_window_entryis         #&  qvz/MongoDBStorageBase.acquire_sliding_window_entry)rrrF)Fr)r)r)__name__ __module__ __qualname____doc__Z DEPENDENCIESr%propertyr9r:rrrr6r@r7rCrFrSrWrjrkrurwrr __classcell__r3r3r1r4rs@ %  - *,Frz2.1)versionz3.14.0zEAdded option to select custom collection names for windows & counters)rreasonc@s&eZdZddgZddddddZd S) MongoDBStorageZmongodbz mongodb+srvr<rrr=cKstt|jj|fi|Sr5)r rr*r?r3r3r4r6sz!MongoDBStorage._init_mongo_clientN)rrrZSTORAGE_SCHEMEr6r3r3r3r4rsr) __future__rrKrrabcrrZdeprecated.sphinxrrZ limits.typingrrr r r r r utilrbaserrrrrr3r3r3r4s& $  R