o Sûbã@s^UdZddlZddlZddlmZmZddlmZmZm Z m Z mZmZm Z ddlmZedddZee eed<edd dZeeed<ed edƒdZeeed <de efd d„Zdefdd„Zdefdd„Zdeddfdd„Zde efdd„Zdedefdd„ZGdd„deƒZededefdZd edefd!d"„Zd#e d$e eddfd%d&„ZdS)'zInternal helpers for CSOT.éN)Ú ContextVarÚToken)ÚAnyÚCallableÚMutableMappingÚOptionalÚTupleÚTypeVarÚcast)ÚWriteConcernÚTIMEOUT)ÚdefaultÚRTTçÚDEADLINEÚinfÚreturncCs t d¡S©N)rÚget©rrú4/tmp/pip-target-onvjaxws/lib/python/pymongo/_csot.pyÚget_timeouts rcCót ¡Sr)rrrrrrÚget_rtt!órcCrr)rrrrrrÚget_deadline%rrÚrttcCst |¡dSr)rÚset)rrrrÚset_rtt)srcCstƒsdSt ¡t ¡Sr)rrrÚtimeÚ monotonicrrrrÚ remaining-sr!Úmax_timeoutcCstƒ}|dur |St||ƒS)z4Return the remaining timeout clamped to a max value.N)r!Úmin)r"ÚtimeoutrrrÚclamp_remaining3s r%c@s6eZdZdZdZdeefdd„Zdd„Zdd „Z d S)Ú_TimeoutContextzInternal timeout context manager. Use :func:`pymongo.timeout` instead:: with pymongo.timeout(0.5): client.test.test.insert_one({}) ©Ú_timeoutÚ_tokensr$cCs||_d|_dSrr')Úselfr$rrrÚ__init__Fs z_TimeoutContext.__init__cCsZt |j¡}t ¡}|jrt ¡|jntdƒ}t t||ƒ¡}t d¡}|||f|_ |S)Nrr)rrr(rrrr Úfloatr#rr))r*Ú timeout_tokenZ prev_deadlineZ next_deadlineÚdeadline_tokenÚ rtt_tokenrrrÚ __enter__Js z_TimeoutContext.__enter__cCs8|jr|j\}}}t |¡t |¡t |¡dSdSr)r)rÚresetrr)r*Úexc_typeÚexc_valÚexc_tbr-r.r/rrrÚ__exit__Ss üz_TimeoutContext.__exit__N) Ú__name__Ú __module__Ú__qualname__Ú__doc__Ú __slots__rr,r+r0r5rrrrr&;s r&ÚF.)ÚboundÚfunccs t ˆ¡‡fdd„ƒ}tt|ƒS)z/Apply the client's timeoutMS to this operation.csltƒdur+|j}|dur+t|ƒˆ|g|¢Ri|¤ŽWdƒS1s&wYˆ|g|¢Ri|¤ŽSr)rr(r&)r*ÚargsÚkwargsr$©r=rrÚcsot_wrapperbs ÿzapply..csot_wrapper)Ú functoolsÚwrapsr r;)r=rArr@rÚapply_s rDÚcmdÚ write_concerncCs>|r|jrdS|j}tƒdur| dd¡|r||d<dSdS)z+Apply the given write concern to a command.NZwtimeoutZwriteConcern)Zis_server_defaultZdocumentrÚpop)rErFÚwcrrrÚapply_write_concernns ÿrI) r9rBrÚcontextvarsrrÚtypingrrrrrr r Zpymongo.write_concernrrr,Ú__annotations__rrrrrrr!r%Úobjectr&r;rDrIrrrrÚs&$!