o hq7@sddZddlmZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZmZmZmZdd lmZmZmZmZdd l m!Z!m"Z"erdd l#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8dZ9ddddiZ:GdddZ;dS)z2Communicate with one MongoDB server in a topology.) annotationsN)datetime) TYPE_CHECKINGAnyAsyncContextManagerCallableOptionalUnion)_decode_all_selective)_handle_reauth)NotPrimaryErrorOperationFailure)_check_command_response)_COMMAND_LOGGER _SDAM_LOGGER_CommandStatusMessage _debug_log_SDAMStatusMessage)_convert_exception_GetMore_OpMsg_Query)PinnedResponseResponse)Queue) ReferenceType)ObjectId)AsyncMongoClient_MongoClientErrorHandler)Monitor)AsyncConnectionPool)_EventListeners) _ServerMode)ServerDescription) _DocumentOutFcursor) firstBatch nextBatchc@seZdZ   d@dAddZdBddZdCdDddZdBddZdBddZ dEdFd#d$Ze dGd,d-Z dCdHd1d2Z e dId3d4Z e jdJd5d4Z e dKd6d7ZdLd;d<ZdMd>d?ZdS)NServerNserver_descriptionr$poolr!monitorr topology_idOptional[ObjectId] listenersOptional[_EventListeners]eventsOptional[ReferenceType[Queue]]returnNonecCsJ||_||_||_||_|duo|j|_||_d|_|jr#||_dSdS)zRepresent one MongoDB server.N) _description_pool_monitor _topology_idZenabled_for_server_publish _listener_events)selfr+r,r-r.r0r2r>N/var/www/html/venv/lib/python3.10/site-packages/pymongo/asynchronous/server.py__init__>s  zServer.__init__cs|jjjs |jdSdS)z[Start monitoring, or restart after a fork. Multiple calls have no effect. N)r7optsZ load_balancedr8openr=r>r>r?rBRs z Server.open service_idcs|j|IdHdS)zClear the connection pool.N)r,reset)r=rDr>r>r?rEZsz Server.resetcs|jr!|jdus J|jdusJ|j|jj|jj|jfft t j r:t tt j|j|jjd|jjdd|jIdH|jIdHdS)zXClear the connection pool and stop the monitor. Reconnect with open(). Nrr')messageZ topologyId serverHost serverPort)r:r;r<putZpublish_server_closedr6addressr9r isEnabledForloggingDEBUGrrZ STOP_SERVERr8closer7rCr>r>r?rN^s&    z Server.closecCs|jdS)zCheck the server's state soon.N)r8 request_checkrCr>r>r?rOxszServer.request_checkF operationUnion[_Query, _GetMore]connr apply_timeoutbooltuple[dict[str, Any], str]csR|||\}}|jjr |jjjs |jj|j||jIdH}||||fSN)Z as_commandclient _encrypterZ_bypass_auto_encryptionZencryptdb codec_optionsZupdate_command)r=rPrRrScmdrYr>r>r?operation_to_command|s  zServer.operation_to_commandread_preferencer# unpack_res!Callable[..., list[_DocumentOut]]rWrrcs>|dusJ|j}t}||} |jo|jj} |||| IdH\} } | r+d} n|||| }||\} }}t t j rat t tj|jj| tt| | | | |j|j|jd|jd|jd |rd| vrk| | d<|dusqJ|j| | | |j|j|jdzK| r|dIdH}n|||IdH|| IdH}| rt}d}nd}d}|||j|j||d }| r|d}|j||j IdHt!||j"Wnkt#y5}z^t|}t$|t%t&fr|j'}nt(|}t t j rt t tj)|jj||tt| | | | |j|j|jd|jd|jt$|t&d |r0|dusJ|j*|||j+| |j|j|j| d d}~wwt|}| rD|d}n.|j+d krT|rQ|dni}n|j|,d dd}|j+dkrl||dd<n||dd<t t j rt t tj-|jj||tt| | | | |j|j|jd|jd|jd|r|dusJ|j.|||j+| |j|j|j| d |j}|r|j/r| r|j/0|1IdH}t2||j|}|3|j s|j4r|5t$|t6r|j} n t7|j4o|j} |jr|j8| t9||j:j||| | || d}|St;||j:j|| | |d}|S)aHRun a _Query or _GetMore operation and return a Response object. This method is used only to run _Query/_GetMore operations from cursors. Can raise ConnectionFailure, OperationFailure, etc. :param conn: An AsyncConnection instance. :param operation: A _Query or _GetMore object. :param read_preference: The read preference to use. :param listeners: Instance of _EventListeners or None. :param unpack_res: A callable that decodes the wire protocol response. :param client: An AsyncMongoClient instance. Nrr') rFclientIdcommand commandName databaseName requestId operationIddriverConnectionIdserverConnectionIdrGrH serviceIdz$db)rDFT)legacy_response user_fields)rFr` durationMSfailurerbrcrdrerfrgrGrHrhZisServerSideError)rDZ database_nameZexplain)idns)r&okfindr&r(r)) rFr`rkreplyrbrcrdrerfrgrGrHrh)datarJrRduration request_id from_commanddocs more_to_come)rrrJrsrtrurv)r>r? run_operationsR           %         zServer.run_operationhandler"Optional[_MongoClientErrorHandler]$AsyncContextManager[AsyncConnection]cs|j|SrV)r,checkout)r=rr>r>r?r^s zServer.checkoutcC|jSrV)r6rCr>r>r? descriptionczServer.descriptioncCs|j|jjks J||_dSrV)rJr6)r=r+r>r>r?rgs cCrrV)r7rCr>r>r?r,lrz Server.poolrF,Union[tuple[int, Any], tuple[int, Any, int]]tuple[int, Any, int]cCs"t|dkr|S|\}}||dfS)zReturn request_id, data, max_doc_size. :param message: (request_id, data, max_doc_size) or (request_id, data) r)len)r=rFrtrrr>r>r?ryps  zServer._split_messagestrcCsd|jjd|jdS)N< >) __class____name__r6rCr>r>r?__repr__~szServer.__repr__)NNN)r+r$r,r!r-rr.r/r0r1r2r3r4r5)r4r5rV)rDr/r4r5)F)rPrQrRr rSrTr4rU)rRr rPrQr]r#r0r1r^r_rWrr4r)rrr4r)r4r$)r+r$r4r5)r4r!)rFrr4r)r4r)r __module__ __qualname__r@rBrErNrOr\r rrpropertyrsetterr,ryrr>r>r>r?r*=s0      V     r*)<__doc__ __future__rrLrtypingrrrrrr Zbsonr Zpymongo.asynchronous.helpersr Zpymongo.errorsr r Zpymongo.helpers_sharedrZpymongo.loggerrrrrrZpymongo.messagerrrrZpymongo.responserrqueuerweakrefrZ bson.objectidrZ!pymongo.asynchronous.mongo_clientrrZpymongo.asynchronous.monitorrZpymongo.asynchronous.poolr r!Zpymongo.monitoringr"Zpymongo.read_preferencesr#Zpymongo.server_descriptionr$Zpymongo.typingsr%Z_IS_SYNCr}r*r>r>r>r?s4