^gnK ddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z ddl m Z mZmZddlmZe dZedZed ed d Gdde e e Zy)) annotationsN) versionaddedversionchanged)MovingWindowSupportSlidingWindowCounterSupportStorage) ParamSpecTypeVarcast)get_dependencyPRz2.1)versionz3.14.0zEAdded option to select custom collection names for windows & counters)rreasonc6eZdZdZddgZ ddgZ d dfd Ze ddZedZ dd Z dd Z dd Z dd Z dd Z d ddZddZ ddZ d d!dZ d d!dZ d"dZxZS)#MongoDBStoragezS Rate limit storage with MongoDB as backend. Depends on :pypi:`motor` async+mongodbzasync+mongodb+srvmotor.motor_asynciopymongoc |jddd}t||fd|i||jd|_|jd|_t d\|_}|jjj|fi||_ tj|j_ ||_||d|_d |_y ) 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 rmongodbwrap_exceptionsrrzpymongo.errors)counterswindowsFN)replacesuper__init__ dependencies dependencyproxy_dependencyr lib_errorsmoduleAsyncIOMotorClientstorageasyncioget_running_loop get_io_loop_MongoDBStorage__database_name#_MongoDBStorage__collection_mapping _MongoDBStorage__indices_created) selfuri database_namecounter_collection_namewindow_collection_nameroptions_ __class__s a/var/www/html/suriana-translation/venv/lib/python3.12/site-packages/limits/aio/storage/mongodb.pyrzMongoDBStorage.__init__,s4kk/9a8 IoII++,AB $ 1 1) <+,<=@t--@@PP $+#;#;  ,/-% !"'c.|jjSN)r" PyMongoErrorr,s r4base_exceptionszMongoDBStorage.base_exceptions[s+++r5cL|jj|jSr7)r% get_databaser)r9s r4databasezMongoDBStorage.databaseas||(()=)=>>r5cK|jsttj|j|jdj dd|j|jdj ddd{d|_y7 w)NrexpireAtr)expireAfterSecondsrT)r+r&gatherr=r* create_indexr9s r4create_indiceszMongoDBStorage.create_indiceses%%.. d77 CDQQ1R d77 BCPP1Q   "& sAB ? B cKttj|j|jdj i|j|jdj id{}tj|j|jdj |j|jdj d{tt|S77w)z^ Delete all rate limit keys in the rate limit collections (counters, windows) rrN) sumr&rAr=r*count_documentsdropr int)r,num_keyss r4resetzMongoDBStorage.resetqs.. d77 CDTT d77 BCSSTVW   nn MM$33J? @ E E G MM$33I> ? D D F   C""  s%A1C>3C:4A1C>%C<&C><C>cKtj|j|jdj d|i|j|jdj d|id{y7w)z> :param key: the key to clear rate limits for r_idrN)r&rAr=r*find_one_and_delete)r,keys r4clearzMongoDBStorage.clearsrnn MM$33J? @ T T   MM$33I> ? S S      sA0A:2A83A:c.K|j|jdjd|id{}|r|dntjj j tj jjS7_w)z; :param key: the key to get the expiry for rrLNr?tzinfo) r=r*find_onedatetimenowrtimezoneutc timestampr,rNcounters r4 get_expiryzMongoDBStorage.get_expirys~ d&?&? &KLUU CL  %,WZ 1B1B1F1F1H WH--11W 2 Y[  s1BBA BcK|j|jdj|dtjj tj j iddgd{}|xr|dxsdS7w)zB :param key: the key to get the counter value for r$gte)rLr?count) projectionNr)r=r*rSrTrUrVrWrYs r4getzMongoDBStorage.gets d&?&? &KLUU#X%6%6%:%:8;L;L;P;P%QR  y V  +77+0q0 sA,B.B/Bc K|jd{tjjtjjtj |z}|j |jdjd|iddddd gi|d d |gid idddd gi||r|ndd id igddg|jjjjd{}t|dS77w)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 NsecondsrrL$set$condz$ltz $expireAt$$NOW$addz$countifthenelse)r^r?Tr^)upsertr_return_document)rCrTrUrVrW timedeltar=r*find_one_and_updater!r#ReturnDocumentAFTERrH)r,rNexpiryelastic_expiryamount expirationresponses r4incrzMongoDBStorage.incrs3!!###&&**8+<+<+@+@AHDVDVE    % %j 1   CL#',{G.D&E(.)/(F1C(D&"$',{G.D&E(27E;&% (y 1188GGMM1   :8G$%%G $  s"DC<CD)C>*D>DcdK |jjd{y7#YyxYww)z{ Check if storage is healthy by calling :meth:`motor.motor_asyncio.AsyncIOMotorClient.server_info` NTF)r% server_infor9s r4checkzMongoDBStorage.checks2  ,,**, , , - s 0)')0)-0c 2Ktj}|j|jdjdd|iidddddd d ||z gid iiid did dddiddidigj dd{x}r|dd|ddfS|dfS7w)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$matchrLz$projectentriesz$filterz$entriesentryr]z$$entry)inputascondz$unwindz$groupz$_idz$minz$sumr)rLminr^)lengthNrrr^)timer=r* aggregateto_list)r,rNlimitrrrXresults r4get_moving_windowz MongoDBStorage.get_moving_windowsIIK -- 9 9) DE Ys|,"% )-7*1-3iVAS5T,U,"(%  + #)$*J#7&,a[#0WAW 3  6 8!9U#VAYw%77 7!|9 sA8B:B;Bc (K|jd{||kDrytj} d|g|zd|didtjjtjj tj |zid}|j|jd jd |d ||z d d ||z iii|dd{y77#|jjjj$rYywxYww)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 NFr|r)z$eachz $positionz$slicer?rb)z$pushrdrrLzentries.z$notr]T)rl)rCrrTrUrVrWrnr=r* update_oner!r#errorsDuplicateKeyError)r,rNrrrrtrXupdatess r4 acquire_entryzMongoDBStorage.acquire_entrys<!!### E>IIK   "+v!5%&"'  ))--h.?.?.C.CD",,V<= &-- 9 9) DEPP3uv~./&69vCU:V1W Q  G $4 $$++22DD  sEDCDBCCCDC-D DDDcpK|jd{|dz}|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|zgidid dididdddddddd dd dd |gigigi|gigid giiiidddddd|gi|gidd |gid diiiddddd|gi|giiiddgig|jj j jdd{}tt|dS77w)NrrLrd previousCountre$lte $subtract $expiresAtrf$ifNull $currentCountr$previousCountrhz$gtrg) currentCount expiresAtcurWeightedCountz$floorz $multiplyz$dividez$maxrz$curWeightedCount _acquiredz$unsetT)rmrl) rCr=r*ror!r#rprqr bool)r,rNrrrrt expiry_msrs r4acquire_sliding_window_entryz+MongoDBStorage.acquire_sliding_window_entryDs!!###TM }}  % %i 0   CL'#$*)4|W6M(N(1-&'" *3_a4H(I)25Eq4I(J & * "#$*)4|W6M(N(1-&'" )*)2_a4H(I & )$$*)4|W6M(N(1-&'"%,/4|Q6G.H17, 9R0S17'1y=9Q0R.&)")5&&!F*$ &(3,<098><=@KDPHNLSLUQJEFNB => A:56 5>#<2-.6*%&4%47)"' -"$L&#$*)/2Ev1N(O(--&'" *0/61J(K(7 & ) "#"f/BF.K%Le$T&./0_p b!1188GGMMi u w rD&-..w $w s"D6D1DD6D4D64D6c K|dz}|j|jdjd|idddddd gi|gid d d gid d d gididdddd gi|gid d d d gididdddd gi|gidd|giddidig|jjj j gdd{x}r|jdr;|djtjjjntj}td |tjz }td |dr||z nd }|d||d|fSy7w)NrrrLrdrerrrrfrrrrrhrg)rrr)rrr)rmr_rrQrr)rrr)r=r*ror!r#rprqr`rrTrVrWrXrmax)r,rNrrrr expires_at current_ttlprev_ttls r4get_sliding_windowz!MongoDBStorage.get_sliding_windowsTM ==  % %i 0   CL#$*)4|W6M(N(1-&'" *3_a4H(I)25Eq4I(J & *$$*)4|W6M(N(1-&'" )*)2_a4H(I & )$$*)4|W6M(N(1-&'" *0, 1J(K(4 & &3%') T!1188GGMME[ . 0 0 60 f::k* $,,H4E4E4I4I,JTTVYY[  adiik!9:K1f_6MkF2STUH'~&   0 sB%E'E(B+E)limitsrrF)r-strr.rr/rr0rrrr1zfloat | str | boolreturnNone)rz-type[Exception] | tuple[type[Exception], ...])rr)rz int | None)rNrrr)rNrrfloat)rNrrrH)Fr) rNrrrrHrsrrtrHrrH)rr)rNrrrHrrrHrztuple[float, int])r) rNrrrHrrrHrtrHrr)rNrrrrHrztuple[int, float, int, float])__name__ __module__ __qualname____doc__STORAGE_SCHEME DEPENDENCIESrpropertyr:r=rCrJrOr[r`rwrzrrrr __classcell__)r3s@r4rrs  &':;N*95L &'1&/ % -' -'-'"% -' !$ -'  -'&-' -'^, 6,, ?? &#&     1RS/&/& #/&59/&KN/& /&b ))"),/) )X@A..".,/.9<. .b@A~/~/"~/,/~/9<~/ ~/@CC #C &Cr5r) __future__rr&rTrdeprecated.sphinxrrlimits.aio.storage.baserrr limits.typingr r r limits.utilr r rrr5r4rs{" :  ' cN CLe  RiW13Ni  ir5