o {g@shUdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZmZmZmZmZmZmZmZerEddlmZe d dd Zd ed <e d d d Zded <e dedd Zded<d8ddZd9ddZd9ddZd:ddZ d8ddZ!d;d!d"Z"Gd#d$d$e Z#ed%ed&efd'Z$dzInternal helpers for CSOT.) annotationsN)deque)AbstractContextManager) ContextVarToken) TYPE_CHECKINGAnyCallableDequeMutableMappingOptionalTypeVarcast) WriteConcernTIMEOUT)defaultzContextVar[Optional[float]]RTTzContextVar[float]DEADLINEinfreturnOptional[float]cCs tdSN)rgetrrj/var/www/bot.gig.net.ua/public_html/telegram/P1/HellBot/venv/lib/python3.10/site-packages/pymongo/_csot.py get_timeout#s rfloatcCtSr)rrrrrrget_rtt'rcCrr)rrrrrr get_deadline+r r!rttNonecCst|dSr)rset)r"rrrset_rtt/r%cCstsdSttSr)rrrtime monotonicrrrr remaining3sr) max_timeoutcCst}|dur |St||S)z4Return the remaining timeout clamped to a max value.N)r)min)r*timeoutrrrclamp_remaining9s r-c@s.eZdZdZdddZdddZdddZdS)_TimeoutContextzInternal timeout context manager. Use :func:`pymongo.timeout` instead:: with pymongo.timeout(0.5): client.test.test.insert_one({}) r,rcCs||_d|_dSr)_timeout_tokens)selfr,rrr__init__Js z_TimeoutContext.__init__rcCsZt|j}t}|jrt|jntd}tt||}t d}|||f|_ |S)Nrr) rr$r/rrr'r(rr+rr0)r1 timeout_token prev_deadline next_deadlinedeadline_token rtt_tokenrrr __enter__Ns   z_TimeoutContext.__enter__exc_typerexc_valexc_tbr#cCs8|jr|j\}}}t|t|t|dSdSr)r0rresetrr)r1r9r:r;r3r6r7rrr__exit__Ws    z_TimeoutContext.__exit__N)r,r)rr.)r9rr:rr;rrr#)__name__ __module__ __qualname____doc__r2r8r=rrrrr.As    r.F.)boundfunccsFtrtd fdd }n td fdd }tt|S) zbApply the client's timeoutMS to this operation. Can wrap both asynchronous and synchronous methodsr1rargskwargsrcsztdur/|j}|dur/t||g|Ri|IdHWdS1s*wY|g|Ri|IdHSrrr/r.r1rErFr,rDrr csot_wrappergs   zapply..csot_wrappercsltdur+|j}|dur+t||g|Ri|WdS1s&wY|g|Ri|SrrGrHrIrrrJqs   N)r1rrErrFrrr)inspectiscoroutinefunction functoolswrapsrrB)rDrJrrIrapplycs   rOcmdMutableMapping[str, Any] write_concernOptional[WriteConcern]cCs>|r|jrdS|j}tdur|dd|r||d<dSdS)z+Apply the given write concern to a command.Nwtimeout writeConcern)is_server_defaultdocumentrpop)rPrRwcrrrapply_write_concern}s    rZ int_MAX_RTT_SAMPLES_MIN_RTT_SAMPLESc@sBeZdZUdZded<dddZdd d Zdd d ZdddZdS) MovingMinimumz4Tracks a minimum RTT within the last 10 RTT samples.z Deque[float]samplesrr#cCsttd|_dS)N)maxlen)rr]rar1rrrr2szMovingMinimum.__init__samplercCs&|dkr td||j|dS)Nrzduration cannot be negative ) ValueErrorraappend)r1rdrrr add_sampleszMovingMinimum.add_samplecCst|jtkr t|jSdS)z7Get the min, or 0.0 if there aren't enough samples yet.r)lenrar_r+rcrrrrs zMovingMinimum.getcCs|jdSr)raclearrcrrrr<r&zMovingMinimum.resetN)rr#)rdrrr#rr) r>r?r@rA__annotations__r2rgrr<rrrrr`s    r`)rrrj)r"rrr#)r*rrr)rDrBrrB)rPrQrRrSrr#)*rA __future__rrMrKr' collectionsr contextlibr contextvarsrrtypingrrr r r r r rpymongo.write_concernrrrkrrrrrr!r%r)r-r.rBrOrZr]r_r`rrrrs6   (