o 8Bi@srUdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZmZmZmZmZmZmZmZerEddlmZe d dd Zd ed <e d d d Zded <e dedd Zded<d:ddZd;ddZdd#d$Z#Gd%d&d&e Z$ed'ed(efd)Z%d?d+d,Z&d@d1d2Z'd3Z(d4ed5<d6Z)d4ed7<Gd8d9d9Z*dS)AzInternal helpers for CSOT.) annotationsN)deque)AbstractContextManager) ContextVarToken) TYPE_CHECKINGAnyCallableDequeMutableMappingOptionalTypeVarcast) WriteConcernTIMEOUT)defaultzContextVar[Optional[float]]RTTzContextVar[float]DEADLINEinfreturnNonecCs&tdtdttddS)Nrr)rsetrrfloatrrM/var/www/html/Trade-python/venv/lib/python3.10/site-packages/pymongo/_csot.py reset_all#s  rOptional[float]cCs tdSN)rgetrrrr get_timeout)s r rcCtSr)rrrrrrget_rtt-r"cCr!r)rrrrrr get_deadline1r#r$rttcCst|dSr)rr)r%rrrset_rtt5r&cCstsdSttSr)r rrtime monotonicrrrr remaining9sr* max_timeoutcCst}|dur |St||S)z4Return the remaining timeout clamped to a max value.N)r*min)r+timeoutrrrclamp_remaining?s 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__Ps z_TimeoutContext.__init__rrcCsZt|j}t}|jrt|jntd}tt||}t d}|||f|_ dS)Nrr) rrr0rrr(r)rr,rr1)r2 timeout_token prev_deadline next_deadlinedeadline_token rtt_tokenrrr __enter__Ts  z_TimeoutContext.__enter__exc_typerexc_valexc_tbcCs8|jr|j\}}}t|t|t|dSdSr)r1rresetrr)r2r:r;r<r4r7r8rrr__exit__\s    z_TimeoutContext.__exit__N)r-rrr)r:rr;rr<rrr)__name__ __module__ __qualname____doc__r3r9r>rrrrr/Gs   r/F.)boundfunccsFtrtd fdd }n td fdd }tt|S) zbApply the client's timeoutMS to this operation. Can wrap both asynchronous and synchronous methodsr2rargskwargsrcsztdur/|j}|dur/t||g|Ri|IdHWdS1s*wY|g|Ri|IdHSrr r0r/r2rGrHr-rFrr csot_wrapperls   zapply..csot_wrappercsltdur+|j}|dur+t||g|Ri|WdS1s&wY|g|Ri|SrrIrJrKrrrLvs   N)r2rrGrrHrrr)inspectiscoroutinefunction functoolswrapsrrD)rFrLrrKrapplyhs   rQcmdMutableMapping[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_defaultdocumentr pop)rRrTwcrrrapply_write_concerns    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]samplesrrcCsttd|_dS)N)maxlen)rr_rcr2rrrr3szMovingMinimum.__init__samplercCs&|dkr td||j|dS)Nrzduration cannot be negative ) ValueErrorrcappend)r2rfrrr add_sampleszMovingMinimum.add_samplecCst|jtkr t|jSdS)z7Get the min, or 0.0 if there aren't enough samples yet.r)lenrcrar,rerrrrs zMovingMinimum.getcCs|jdSr)rcclearrerrrr=r'zMovingMinimum.resetNr?)rfrrrrr) r@rArBrC__annotations__r3rirr=rrrrrbs    rbr?)rrrl)r%rrr)r+rrr)rFrDrrD)rRrSrTrUrr)+rC __future__rrOrMr( collectionsr contextlibr contextvarsrrtypingrrr r r r r rpymongo.write_concernrrrmrrrrr r"r$r&r*r.r/rDrQr\r_rarbrrrrs8   (