o 'h@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)rgetrrI/var/www/html/olx_land/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.eZdZdZdddZddd ZdddZdS)_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__rr#cCsZt|j}t}|jrt|jntd}tt||}t d}|||f|_ dS)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_tbcCs8|jr|j\}}}t|t|t|dSdSr)r0rresetrr)r1r9r:r;r3r6r7rrr__exit__Vs    z_TimeoutContext.__exit__N)r,rrr#)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.r1rFrGr,rErr csot_wrapperfs   zapply..csot_wrappercsltdur+|j}|dur+t||g|Ri|WdS1s&wY|g|Ri|SrrHrIrJrrrKps   N)r1rrFrrGrrr)inspectiscoroutinefunction functoolswrapsrrC)rErKrrJrapplybs   rPcmdMutableMapping[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)rQrSwcrrrapply_write_concern|s    r[ 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^rbr1rrrr2szMovingMinimum.__init__samplercCs&|dkr td||j|dS)Nrzduration cannot be negative ) ValueErrorrbappend)r1rerrr add_sampleszMovingMinimum.add_samplecCst|jtkr t|jSdS)z7Get the min, or 0.0 if there aren't enough samples yet.r)lenrbr`r+rdrrrrs zMovingMinimum.getcCs|jdSr)rbclearrdrrrr<r&zMovingMinimum.resetNr>)rerrr#rr) r?r@rArB__annotations__r2rhrr<rrrrras    ra)rrrk)r"rrr#)r*rrr)rErCrrC)rQrRrSrTrr#)*rB __future__rrNrLr' collectionsr contextlibr contextvarsrrtypingrrr r r r r rpymongo.write_concernrrrlrrrrrr!r%r)r-r.rCrPr[r^r`rarrrrs6   (