3 O6b&@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z mZddlmZddlmZmZmZddlmZddlmZmZmZmZddlmZddlm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3dd l4m5Z5dd l6m7Z7dd l8m9Z9dd l:m;Z<ddl=m>Z>m?Z?ddl@mAZAmBZBddlCmDZDddlEmFZFddlGmHZHddlImJZJddlKmLZLy$ddlMmMZMmNZNmOZOmPZPddZQWneRk rddZQYnXdZSdZTdZUejdkry ddlVZWWneRk rddlWZWYnXddZXy6eWjYeWjZdZ[eXe[d d!Z\eXe[d"d#Z]WdQRXWne^k rxd!Z\d#Z]YnXd$d%Z_nd&d'Z`d(d%Z_ed)ed\d,efgfgZaejjbd-r&ejcdd.d]krd1jdd2d3ejeDZfnejgZfed4ejgfd*effd5ejhfd,ejifgead6<nNejd7krned4ejgfd*ejgfd5ejhfd,ejjdfgead6<nejdkred4ejgfd*d1jdejgejiffd5ejhfd,d8jdejkd9d/fgead6<nejjbd:rejld^\ZfZmZned4effd*effd5enfd,emfgead6<nbejoejgejiejpZqed4ejgfd*d1jdd;d3eqdd.Dfd5ejhfd,eqd.fgead6<ejrjbd<rd1jdejrd=jdeesejtd>d=jdeesejcfead?<npejjbd:r d1jdejrd=jdeesejcd@d1jdejgejiffead?<n$d1jdejrd=jdeesejcfead?<dAjudBdCZvd_dDdEZwGdFdGdGexZydHdIZzdJdKZ{GdLdMdMexZ|GdNdOdOexZ}dPdQZ~dRdSZGdTdUdUe3ZGdVdWdWexZGdXdYdYexZGdZd[d[ZdS)`N)SSLErrorHAS_SNI IPADDR_SAFE)DEFAULT_CODEC_OPTIONS)imap itervalues_unicode)SON)authhelpers thread_util __version__)_validate_session_write_concern) MAX_BSON_SIZEMAX_IDLE_TIME_SECMAX_MESSAGE_SIZE MAX_POOL_SIZEMAX_WIRE_VERSIONMAX_WRITE_BATCH_SIZE MIN_POOL_SIZE ORDERED_TYPESWAIT_QUEUE_TIMEOUT) AutoReconnectCertificateErrorConnectionFailureConfigurationErrorInvalidOperationDocumentTooLargeNetworkTimeoutNotPrimaryErrorOperationFailure PyMongoError) HelloCompat) is_ip_address)IsMaster)time)ConnectionCheckOutFailedReasonConnectionClosedReason)commandreceive_message)ReadPreference)_add_to_command) SERVER_TYPE) SocketChecker)match_hostname)fcntlF_GETFDF_SETFD FD_CLOEXECcCst|t}t|t|tBdS)z8Set the close-on-exec flag on the given file descriptor.N)r/r0r1r2)fdflagsr5K/var/www/html/sandeepIITI/myenv/lib/python3.6/site-packages/pymongo/pool.py_set_non_inheritable_non_atomicGs r7cCsdS)z6Dummy function for platforms that don't provide fcntl.Nr5)dummyr5r5r6r7Osx win32c Cs6ytj||\}}t|Sttfk r0|SXdS)N)winreg QueryValueExintOSError ValueError)keynamedefaultvalue_r5r5r6_query]s rGz2SYSTEM\CurrentControlSet\Services\Tcpip\ParametersZ KeepAliveTimeimZKeepAliveIntervalicCsDtttd}tttd}|tks,|tkr@|jtjd||fdS)Ni)min_WINDOWS_TCP_IDLE_MS_MAX_TCP_KEEPIDLE_WINDOWS_TCP_INTERVAL_MS_MAX_TCP_KEEPINTVLZioctlsocketZSIO_KEEPALIVE_VALS)sockZidle_msZ interval_msr5r5r6_set_keepalive_timesss rPc CsZtt|rVtt|}y*|jtj|}||kr<|jtj||Wntjk rTYnXdS)N)hasattrrNgetattr getsockopt IPPROTO_TCP setsockopterror)rOZ tcp_optionZ max_valueZsockoptrDr5r5r6_set_tcp_option|s  rWcCs(t|dtt|dtt|dtdS)N TCP_KEEPIDLE TCP_KEEPINTVL TCP_KEEPCNT)rWrKrM_MAX_TCP_KEEPCNT)rOr5r5r6rPs  driverrCPyMongoversionlinux cCsg|] }|r|qSr5r5).0partr5r5r6 srftype architectureosdarwin-rHjavacCsg|] }|r|qSr5r5)rdrer5r5r6rfsPyPy.z (Python %s)platformz(%s)ZfooidnaFcCsz|\}}|dk r d|||f}n d||f}|r8||}t|tjrNt|n(t|trndt|krnt|nt|dS)z9Convert a socket.error to ConnectionFailure and raise it.Nz %s:%d: %sz%s: %sz timed out) isinstancerNtimeoutr _SSLErrorstrr)addressrVZ msg_prefixhostportmsgr5r5r6_raise_connection_failures    ryc@seZdZd:ZeeeddeddddddddddfddZe ddZ e ddZ e ddZ e ddZ e ddZe d d!Ze d"d#Ze d$d%Ze d&d'Ze d(d)Ze d*d+Ze d,d-Ze d.d/Ze d0d1Ze d2d3Ze d4d5Ze d6d7Ze d8d9ZdS); PoolOptions__max_pool_size__min_pool_size__max_idle_time_seconds__connect_timeout__socket_timeout__wait_queue_timeout__wait_queue_multiple __ssl_context__ssl_match_hostname__socket_keepalive__event_listeners __appname__driver __metadata__compression_settings __server_api__load_balancedNTcCs||_||_||_||_||_||_||_||_| |_| |_ | |_ | |_ | |_ ||_ ||_||_tjt|_| r~d| i|jd<| r| jrdtdd| jf|jdd<| jrdtdd| jf|jdd<| jrdtd| jf|jd<dS)NrCZ applicationz%s|%sr\r^ro)_PoolOptions__max_pool_size_PoolOptions__min_pool_size#_PoolOptions__max_idle_time_seconds_PoolOptions__connect_timeout_PoolOptions__socket_timeout _PoolOptions__wait_queue_timeout!_PoolOptions__wait_queue_multiple_PoolOptions__ssl_context _PoolOptions__ssl_match_hostname_PoolOptions__socket_keepalive_PoolOptions__event_listeners_PoolOptions__appname_PoolOptions__driver"_PoolOptions__compression_settings_PoolOptions__server_api_PoolOptions__load_balancedcopydeepcopy _METADATA_PoolOptions__metadatarCr^ro)self max_pool_size min_pool_sizemax_idle_time_secondsconnect_timeoutsocket_timeoutwait_queue_timeoutwait_queue_multiple ssl_contextssl_match_hostnamesocket_keepaliveevent_listenersappnamer\compression_settings server_api load_balancedr5r5r6__init__s:   zPoolOptions.__init__cCs`i}|jtkr|j|d<|jtkr,|j|d<|jtkrD|jd|d<|jtkr\|jd|d<|S)zqThe non-default options this pool was created with. Added for CMAP's :class:`PoolCreatedEvent`. Z maxPoolSizeZ minPoolSizeiZ maxIdleTimeMSZwaitQueueTimeoutMS)rrrrrrrr)roptsr5r5r6non_default_options4s      zPoolOptions.non_default_optionscCs|jS)aQThe maximum allowable number of concurrent connections to each connected server. Requests to a server will block if there are `maxPoolSize` outstanding connections to the requested server. Defaults to 100. Cannot be 0. When a server's pool has reached `max_pool_size`, operations for that server block waiting for a socket to be returned to the pool. If ``waitQueueTimeoutMS`` is set, a blocked operation will raise :exc:`~pymongo.errors.ConnectionFailure` after a timeout. By default ``waitQueueTimeoutMS`` is not set. )r)rr5r5r6rEs zPoolOptions.max_pool_sizecCs|jS)zThe minimum required number of concurrent connections that the pool will maintain to each connected server. Default is 0. )r)rr5r5r6rTszPoolOptions.min_pool_sizecCs|jS)zThe maximum number of seconds that a connection can remain idle in the pool before being removed and replaced. Defaults to `None` (no limit). )r)rr5r5r6r[sz!PoolOptions.max_idle_time_secondscCs|jS)zGHow long a connection can take to be opened before timing out. )r)rr5r5r6rcszPoolOptions.connect_timeoutcCs|jS)zKHow long a send or receive on a socket can take before timing out. )r)rr5r5r6riszPoolOptions.socket_timeoutcCs|jS)zhHow long a thread will wait for a socket from the pool if the pool has no free sockets. )r)rr5r5r6roszPoolOptions.wait_queue_timeoutcCs|jS)ztMultiplied by max_pool_size to give the number of threads allowed to wait for a socket at one time. )r)rr5r5r6rvszPoolOptions.wait_queue_multiplecCs|jS)z(An SSLContext instance or None. )r)rr5r5r6r}szPoolOptions.ssl_contextcCs|jS)zCCall ssl.match_hostname if cert_reqs is not ssl.CERT_NONE. )r)rr5r5r6rszPoolOptions.ssl_match_hostnamecCs|jS)zZWhether to send periodic messages to determine if a connection is closed. )r)rr5r5r6rszPoolOptions.socket_keepalivecCs|jS)z;An instance of pymongo.monitoring._EventListeners. )r)rr5r5r6rszPoolOptions.event_listenerscCs|jS)zJThe application name, for sending with hello in server handshake. )r)rr5r5r6rszPoolOptions.appnamecCs|jS)zFDriver name and version, for sending with hello in handshake. )r)rr5r5r6r\szPoolOptions.drivercCs|jS)N)r)rr5r5r6rsz PoolOptions.compression_settingscCs |jjS)zLA dict of metadata about the application, driver, os, and platform. )rr)rr5r5r6metadataszPoolOptions.metadatacCs|jS)z0A pymongo.server_api.ServerApi or None. )r)rr5r5r6rszPoolOptions.server_apicCs|jS)z?True if this Pool is configured in load balanced mode. )r)rr5r5r6rszPoolOptions.load_balanced)r{r|r}r~rrrrrrrrrrrrr)__name__ __module__ __qualname__ __slots__rrrrrpropertyrrrrrrrrrrrrrr\rrrrr5r5r5r6rzsD (                 rzcCs.|r*x$|jD]}|jdkr|jr|SqWdS)zDReturn one credential that needs mechanism negotiation, if any. DEFAULTN)valuesZ mechanismusername)all_credentialscredsr5r5r6_negotiate_credss rcCs,|r(t|dkr(tt|}tjj|SdS)zAReturn the _AuthContext to use for speculative auth, if any. rHN)lennextrr Z _AuthContextZfrom_credentials)rrr5r5r6_speculative_contexts  rc@s(eZdZddZddZeddZdS)_CancellationContextcCs d|_dS)NF) _cancelled)rr5r5r6rsz_CancellationContext.__init__cCs d|_dS)zCancel this context.TN)r)rr5r5r6cancelsz_CancellationContext.cancelcCs|jS)zWas cancel called?)r)rr5r5r6 cancelledsz_CancellationContext.cancelledN)rrrrrrrr5r5r5r6rsrc@s&eZdZdZddZddZddZdd Zd d ZdAd dZ ddZ ddZ de j edd dd d dd d d ddd dfddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Z d9d:Z!d;d<Z"d=d>Z#d?d@Z$d S)B SocketInfozStore a socket with some metadata. :Parameters: - `sock`: a raw socket object - `pool`: a Pool instance - `address`: the server's (host, port) - `id`: the id of this socket in it's pool cCstj||_||_||_||_t|_d|_t |_ d|_ d|_ t |_t|_t|_t|_d|_d|_d|_d|_|jj|_|j|_|jj|_d|_t|_ i|_!i|_"|j#|_$|j$j%|_&d|_'d|_(|j)st*|_(|j|_d|_+d|_,d|_-d|_.d|_/dS)NF)0weakrefrefpool_refrOruidsetauthsetclosed_timelast_checkin_timeperformed_handshake is_writablermax_wire_versionr max_bson_sizermax_message_sizermax_write_batch_sizesupports_sessionshello_ok is_mongosop_msg_enabledrr listenersenabled_for_cmaprcompression_contextr-socket_checkernegotiated_mechanismsauth_ctxgenpool_gen get_overall generationreadycancel_context handshaker more_to_come service_id pinned_txn pinned_cursoractive)rrOpoolrurr5r5r6rsH    zSocketInfo.__init__cCsd|_|j stdS)NT)rrAssertionError)rr5r5r6pin_txnszSocketInfo.pin_txncCsd|_|j stdS)NT)rrr)rr5r5r6 pin_cursorszSocketInfo.pin_cursorcCs(|j}|r|j|n |jtjdS)N)r return_socket close_socketr'STALE)rrr5r5r6unpins zSocketInfo.unpincCs4|jjs|jrttjdfgSttjdfdgSdS)NrHhelloOkT)rT)rrrr r"ZCMDZ LEGACY_CMD)rr5r5r6 hello_cmd!szSocketInfo.hello_cmdNcCs|jddd|S)N)_hello)rrr5r5r6hello'szSocketInfo.hellocCs&|j}|j }d}|rNd|_|jj|d<|jr<|jj|d<|jjrd|d<n@|dk r||d<t|d|d<d}|jjr|j j |jj||j d kr|dk r||d <t |}|r|j d |j|d <t|} | r| j|d <|jd|d|d} |jjotr| jdijd} | jd| |jjs4| jddt| |d} | j|_| j |_ | j|_| j|_| j|_| jdk |_| j|_| jtj k|_!|r|jr|jj"| j} | |_#| j d k|_$|r| j%|j&|<| r| j'| | j(r| |j)| j*<|jjr"| j+s t,d| j+|_+|j-j|j+|_.| S)NFTclient compressionZ loadBalancedZtopologyVersioniZmaxAwaitTimeMSz $clusterTimernZsaslSupportedMechsZspeculativeAuthenticateZadmin)publish_eventsexhaust_allowedZ processId serviceId) awaitablez`Driver attempted to initialize in load balancing mode, but the server does not support this mode)/rrrrrZ compressorsrr?rrO settimeoutrrsourcerrZspeculate_commandr(_MOCK_SERVICE_IDget setdefaultpopr$rrrrZlogical_session_timeout_minutesrrZ server_typer,ZMongosrZget_compression_contextrrZsasl_supported_mechsrparse_responseZspeculate_succeededr credentialsrrrr)rZ cluster_timeZtopology_versionZheartbeat_frequencyrcmdZperforming_handshakerrrdocZ process_idrctxr5r5r6r*sv                zSocketInfo._hellocCsH|jd}|j|_|j}|d}tj||j|jjsD|jdd|S)Nrr) r)rZunpack_responser _check_command_responserrrr)rreplyZ unpacked_docsZ response_docr5r5r6 _next_replyrs  zSocketInfo._next_replyFTcCs|j|| t| | } t|ts(t|}| rR|jdkrR| j rRtd| j|jf| dkpf| j pf| dksptd|jdkr| r| j r| j |d<n|jdkr| dk rtd|j || r| j |||||j|| ||r|jnd}| o| j }|jr|j|y@t|||||j||| ||||j|||j| | | |j|j|||dSttfk r`Yn.tk r}z|j|WYdd}~XnXdS) aNExecute a command or raise an error. :Parameters: - `dbname`: name of the database on which to run the command - `spec`: a command document as a dict, SON, or mapping object - `secondary_ok`: whether to set the secondaryOkay wire protocol bit - `read_preference`: a read preference - `codec_options`: a CodecOptions instance - `check`: raise OperationFailure if there are errors - `allowable_errors`: errors to ignore if `check` is True - `check_keys`: if True, check `spec` for invalid keys - `read_concern`: The read concern for this command. - `write_concern`: The write concern for this command. - `parse_write_concern_error`: Whether to parse the ``writeConcernError`` field in the command response. - `collation`: The collation for this command. - `session`: optional ClientSession instance. - `client`: optional MongoClient for gossipping $clusterTime. - `retryable_write`: True if this command is a retryable write. - `publish_events`: Should we publish events for this command? - `user_fields` (optional): Response fields that should be decoded using the TypeDecoders from codec_options, passed to bson._decode_all_selective. zDread concern level of %s is not valid with a max wire version of %d.Nz3Collation is unsupported for unacknowledged writes.rbZ writeConcernz5Must be connected to MongoDB 3.4+ to use a collation.)parse_write_concern_error collationZcompression_ctxZ use_op_msgunacknowledged user_fieldsr)validate_sessionrrqrr rZ ok_for_legacyrlevelZ acknowledgedZis_server_defaultdocumentadd_server_apiZ _apply_tosend_cluster_timerr_raise_if_not_writabler(rrurrr r BaseExceptionry)rZdbnamespecZ secondary_okZread_preferenceZ codec_optionscheckZallowable_errorsZ check_keysZ read_concernZ write_concernrrsessionrZretryable_writerrrrrrVr5r5r6r(}s\&          zSocketInfo.commandcCsh|jdk r&||jkr&td||jfy|jj|Wn,tk rb}z|j|WYdd}~XnXdS)z}Send a raw BSON message or raise ConnectionFailure. If a network exception is raised, the socket is closed. NzfBSON document too large (%d bytes) - the connected server supports BSON document sizes up to %d bytes.)rrrOsendallrry)rmessage max_doc_sizerVr5r5r6 send_messages  zSocketInfo.send_messagecCs@yt|||jStk r:}z|j|WYdd}~XnXdS)zzReceive a raw BSON message or raise ConnectionFailure. If any exception is raised, the socket is closed. N)r)rrry)r request_idrVr5r5r6r)szSocketInfo.receive_messagecCs"|r|j rtddddddS)z^Raise NotPrimaryError on unacknowledged write if this socket is not writable. z not primaryri{')okerrmsgcodeN)rr)rrr5r5r6rs z!SocketInfo._raise_if_not_writablecCs<|j| |j|||r8|j|}tj|j|jSdS)aSend OP_INSERT, etc., optionally returning response as a dict. Can raise ConnectionFailure or OperationFailure. :Parameters: - `request_id`: an int. - `msg`: bytes, an OP_INSERT, OP_UPDATE, or OP_DELETE message, perhaps with a getlasterror command appended. - `max_doc_size`: size in bytes of the largest document in `msg`. - `with_last_error`: True if a getlasterror command is appended. N)rrr)r Z_check_gle_responsecommand_responser)rr rxrZwith_last_errorr r5r5r6 legacy_writes    zSocketInfo.legacy_writecCs0|j|d|j|}|j}tj||j|S)zSend "insert" etc. command, returning response as a dict. Can raise ConnectionFailure or OperationFailure. :Parameters: - `request_id`: an int. - `msg`: bytes, the command message. r)rr)r$r r r)rr rxr resultr5r5r6 write_commands  zSocketInfo.write_commandcCs|s |jrhtt|}|jj}x*||D]}tj|j||jj|q*Wx||D]}|j|qVW|j sd|_ |j r|j j |j |jdS)a Update this socket's authentication. Log in or out to bring this socket's credentials up to date with those provided. Can raise ConnectionFailure or OperationFailure. :Parameters: - `all_credentials`: dict, maps auth source to MongoCredential. TN)rrrrr Zlogoutrdiscard authenticaterrrZpublish_connection_readyrur)rrcachedrrr5r5r6 check_auth!s   zSocketInfo.check_authcCs8tj|||jj||jj|d|jj|ddS)zLog in to the server and store these credentials in `authset`. Can raise ConnectionFailure or OperationFailure. :Parameters: - `credentials`: A MongoCredential. N)r r)raddrrr)rrr5r5r6r)=s  zSocketInfo.authenticatecCs.|r*|j|k rtd|j|jkr*tddS)zValidate this session before use with client. Raises error if this session is logged in as a different user or the client is not the one that created the session. z9Can only use session with the MongoClient that started itzBCannot use session after authenticating with different credentialsN)Z_clientrZ_authsetr)rrrr5r5r6rKs  zSocketInfo.validate_sessioncCs4|jr dS|j|r0|jr0|jj|j|j|dS)z$Close this connection with a reason.N)r _close_socketrrpublish_connection_closedrur)rreasonr5r5r6r[s  zSocketInfo.close_socketc CsH|jr dSd|_|jr |jjy|jjWntk rBYnXdS)zClose this connection.NT)rrrrOclose Exception)rr5r5r6r-ds zSocketInfo._close_socketcCs|jj|jS)z?Return True if we know socket has been closed, False otherwise.)r socket_closedrO)rr5r5r6r2rszSocketInfo.socket_closedcCs|jdkr|r|j||dS)z$Add cluster time for MongoDB >= 3.6.rN)rZ_send_cluster_time)rr(rrr5r5r6rvszSocketInfo.send_cluster_timecCs|jjrt||jjdS)zAdd server_api parameters.N)rrr+)rr(r5r5r6r{szSocketInfo.add_server_apicCs t|_dS)N)rr)rr5r5r6update_last_checkin_timesz#SocketInfo.update_last_checkin_timecCs ||_dS)N)r)rrr5r5r6update_is_writableszSocketInfo.update_is_writablecCs t|jS)z9Seconds since this socket was last checked into its pool.)rr)rr5r5r6idle_time_secondsszSocketInfo.idle_time_secondscCs@|jr d}ntj}|j|t|tttfr:t|j |ndS)N) rr'ERRORrrqIOErrorr@rsryru)rrVr/r5r5r6rys z$SocketInfo._raise_connection_failurecCs |j|jkS)N)rO)rotherr5r5r6__eq__szSocketInfo.__eq__cCs ||k S)Nr5)rr8r5r5r6__ne__szSocketInfo.__ne__cCs t|jS)N)hashrO)rr5r5r6__hash__szSocketInfo.__hash__cCs"dt|j|jrdpdt|fS)NzSocketInfo(%s)%s at %sz CLOSED)reprrOrr)rr5r5r6__repr__s zSocketInfo.__repr__)N)%rrr__doc__rrrrrrrr r*ZPRIMARYrr(rr)rr%r'r+r)rrr-r2rrr3r4r5ryr9r:r<r?r5r5r5r6rsX/ H  N   rc!Cs|\}}|jdrlttds$tdtjtj}t|jy|j||Stjk rj|j YnXtj }tj r|dkrtj }d}xtj |||tjD]}|\}} } } } ytj|| ttddB| }Wn$tjk rtj|| | }YnXt|jyP|jtjtjd|j|j|jtjtj|j|jrFt||j| |Stjk r} z| }|j WYdd} ~ XqXqW|dk r|n tjd dS) zGiven (host, port) and PoolOptions, connect and return a socket object. Can raise socket.error. This is a modified version of create_connection from CPython >= 2.7. z.sockAF_UNIXz-UNIX-sockets are not supported on this system localhostN SOCK_CLOEXECrrHzgetaddrinfo failed)endswithrQrNrrAr7filenoconnectrVr0AF_INEThas_ipv6 AF_UNSPEC getaddrinfo SOCK_STREAMrRrUrT TCP_NODELAYrr SOL_SOCKET SO_KEEPALIVErrP)ruoptionsrvrwrOfamilyerrresafsocktypeprotor8saer5r5r6_create_connectionsN          rXcCst||}|j}|dk r|d}y0trDt| s4trD|j||d}n |j|}WnXtk rn|jYn<tt t fk r}z|jt ||dWYdd}~XnX|j rt |dd r|jryt|j|dWntk r|jYnX|j|j|S)zGiven (host, port) and PoolOptions, return a configured socket. Can raise socket.error, ConnectionFailure, or CertificateError. Sets socket's SSL and timeout options. Nr)server_hostnamezSSL handshake failed: check_hostnameF)hostname)rXr _HAVE_SNIr# _IPADDR_SAFE wrap_socketrr0r7r@rsry verify_moderRrr. getpeercertrr)rurOrOrrvexcr5r5r6_configured_sockets0  rbc@seZdZdZdS)_PoolClosedErrorzZInternal error raised when a thread tries to get a connection from a closed pool. N)rrrr@r5r5r5r6rc+srcc@s4eZdZddZddZddZddZd d Zd S) _PoolGenerationcCstjt|_d|_dS)Nr) collections defaultdictr? _generations _generation)rr5r5r6r3s z_PoolGeneration.__init__cCs|dkr|jS|j|S)z,Get the generation for the given service_id.N)rhrg)rrr5r5r6r9sz_PoolGeneration.getcCs|jS)z"Get the Pool's overall generation.)rh)rr5r5r6r?sz_PoolGeneration.get_overallcCsP|jd7_|dkr:x4|jD]}|j|d7<qWn|j|d7<dS)z2Increment the generation for the given service_id.rHN)rhrg)rrr5r5r6incCs  z_PoolGeneration.inccCs||j|kS)z?Return if the given generation for a given service_id is stale.)r)rrrr5r5r6staleLsz_PoolGeneration.staleN)rrrrrrrirjr5r5r5r6rd2s  rdc@seZdZdZdZdZdS) PoolStaterHr`raN)rrrZPAUSEDZREADYZCLOSEDr5r5r5r6rkQsrkc@seZdZdddZd ddZddZd!d d Zd d Zd dZddZ d"ddZ e j d#ddZ ddZddZddZddZddZdS)$PoolTcCsd|_tj|_tj|_d|_d|_d|_ d|_ t |_ t j|_||_||_||_|jop|jjdk op|jjj|_|jjdks|jjdkrd}n|jj|jj}tj|jj||_|jr|jjj|j|jjt|_d|_d|_dS)z :Parameters: - `address`: a (hostname, port) tuple - `options`: a PoolOptions instance - `handshake`: whether to call hello for each new SocketInfo rHrFN) _check_interval_secondsredequesockets threadingLocklockactive_socketsnext_connection_idrrrdrrigetpidpidrurrrrrrr Zcreate_semaphore_socket_semaphoreZpublish_pool_createdrr_Pool__pinned_socketsncursorsntxns)rrurOrZ max_waitersr5r5r6rZs6        z Pool.__init__Nc Cs(|j|jrdS|jj|tj}|j|kr<||_d|_|dkrX|jt j }|_nJt j }t j }x.|jD]$}|j |kr|j |qp|j |qpW|}||_|rd|_WdQRX|j j}|rx|D]}|jtjqW|jr|j|jn4|jr|j|j|dx|D]}|jtjqWdS)NrT)r)rrrrririrurvrsrorernrappendrrrr' POOL_CLOSEDrZpublish_pool_closedruZpublish_pool_clearedr) rr0rZnewpidror(Zkeep sock_inforr5r5r6_resets>       z Pool._resetc Cs8||_|j"x|jD]}|j|jqWWdQRXdS)zXUpdates the is_writable attribute on all sockets currently in the Pool. N)rrrror4)rrrNr5r5r6r4s zPool.update_is_writablecCs|jd|ddS)NF)r0r)r~)rrr5r5r6resetsz Pool.resetcCs|jdddS)NT)r0)r~)rr5r5r6r0sz Pool.closecCs|jj||S)N)rrj)rrrr5r5r6stale_generationszPool.stale_generationcCs|jjdk rV|j>x6|jrJ|jdj|jjkrJ|jj}|jtjqWWdQRXx|j t |j|j |jj krzPWdQRX|j j dsPzH|j|}|j.|jj|kr|jtjP|jj|WdQRXWd|j jXqXWdS)zRemoves stale sockets then adds new ones if pool is too small and has not been reset. The `reference_generation` argument specifies the `generation` at the point in time this operation was requested on the pool. NrHF)rrrrror5rrr'IDLErrsrrwacquirerFrrr appendleftrelease)rZreference_generationrr}r5r5r6remove_stale_socketss*       zPool.remove_stale_socketsc Cs|j|j}|jd7_WdQRX|jj}|jrB|j|j|yt|j|j}WnXtk r}z<|jr~|j |j|t j t |t ttfrt|j|WYdd}~XnXt|||j|}y&|jr|j||j|_|j|Wn$tk r|jt j YnX|S)zConnect to Mongo and return a new SocketInfo. Can raise ConnectionFailure or CertificateError. Note that the pool does not keep a reference to the socket -- you must call return_socket() when you're done with it. rHN)rrrtrrrZpublish_connection_createdrurbrr.r'r6rqr7r@rsryrrrrr+r)rrZconn_idrrOrVr}r5r5r6rFs2   z Pool.connectc cs|jj}|jr|j|j|j|}|jr:|j|j|jy |VWnN|jpV|j }|rvt j \}}}|j ||| r|j r|j|YnX|jr|j |jj||jd7_WdQRXnF|j r|j |jj||jd7_WdQRXn|j r|j|dS)a$Get a socket from the pool. Use with a "with" statement. Returns a :class:`SocketInfo` object wrapping a connected :class:`socket.socket`. This method should always be used in a with-statement:: with pool.get_socket(credentials) as socket_info: socket_info.send_message(msg) data = socket_info.receive_message(op_code, request_id) The socket is logged in or out as needed to match ``all_credentials`` using the correct authentication mechanism for the server's wire protocol version. Can raise ConnectionFailure or OperationFailure. :Parameters: - `all_credentials`: dict, maps auth source to MongoCredential. - `handler` (optional): A _MongoClientErrorHandler. rHN)rrrZ$publish_connection_check_out_startedru _get_socketZpublish_connection_checked_outrrrsysexc_infohandlerrrrrxr,rzry) rrhandlerrr}pinnedexc_typeexc_valrFr5r5r6 get_sockets6          zPool.get_socketcCsp|jtjkr|j|jr>|jr6|jjj|j t j t d|j jd|jjsX|jd}d}y|j|jd7_d}WdQRXxZ|dkry |j|jj}WdQRXWntk r|j|}YqX|j|rd}qW|j|Wnvtk rd|r|jtj|j j|rB|j|jd8_WdQRX|jr^|jjj|j t jYnXd|_|S)z8Get or create a SocketInfo. Can raise ConnectionFailure.z?Attempted to check out a connection from closed connection poolTNFrH) rvrirurrrrr#publish_connection_check_out_failedrur&r|rcrwrr_raise_wait_queue_timeoutrrrsropopleft IndexErrorrF _perishedr+rrr'r6rZ CONN_ERRORr)rrr}Z incrementedr5r5r6rRsN       zPool._get_socketcCs>|j}|j}d|_d|_d|_|jj||jj}|jrH|j|j |j |j t j kr`|jn|jrt|jtjnr|jr|jr|j|j |j tjnP|jD|j|j|jr|jtjn |j|j|j|jj|WdQRX|jj |j>|r|j!d8_!n|r"|j"d8_"|j#d8_#WdQRXdS)zReturn the socket to the pool, or if it's closed discard it. :Parameters: - `sock_info`: The socket to check into the pool. FNrH)$rrrrxr(rrrZpublish_connection_checked_inrurrvrirurrrr'r|r.r6rrrrrrr3r4rrorrwrrzryrs)rr}Ztxncursorrr5r5r6rs@     zPool.return_socketcCs|j}|jjdk r0||jjkr0|jtjdS|jdk rfd|jksN||jkrf|jrf|jtjdS|j |j |j r|jtj dSdS)aReturn True and close the connection if it is "perished". This side-effecty function checks if this socket has been idle for for longer than the max idle time, or if the socket has been closed by some external network error, or if the socket's generation is outdated. Checking sockets lets us avoid seeing *some* :class:`~pymongo.errors.AutoReconnect` exceptions on server hiccups, etc. We only check if the socket was closed by an external error if it has been > 1 second since the socket was checked into the pool, to keep performance reasonable - we can't avoid AutoReconnects completely anyway. NTrF) r5rrrr'rrmr2r6rrrr)rr}r5r5r5r6rs        zPool._perishedcCsv|jj}|jr|j|jtj|jjrZ|j|j |j }t d|jj |j |j ||jj ft d|jj |jj fdS)NzTimeout waiting for connection from the connection pool. maxPoolSize: %s, connections in use by cursors: %s, connections in use by transactions: %s, connections in use by other operations: %s, wait_queue_timeout: %szgTimed out while checking out a connection from connection pool. maxPoolSize: %s, wait_queue_timeout: %s)rrrrrur&TIMEOUTrrsryrzrrr)rrZ other_opsr5r5r6rs zPool._raise_wait_queue_timeoutcCsx|jD]}|jdqWdS)N)ror)rr}r5r5r6__del__s z Pool.__del__)T)N)N)N)N)rrrrr~r4rr0rrrF contextlibcontextmanagerrrrrrrr5r5r5r6rlYs 7 ) # ) 9:."rl)rCr])rarbr)N)rrrirorNrrprerZpymongo.ssl_supportrrsrr\rr]ZbsonrZbson.py3compatrrrZbson.sonr Zpymongor r r r Zpymongo.client_sessionrZpymongo.commonrrrrrrrrrZpymongo.errorsrrrrrrrrr r!Zpymongo.hello_compatr"Zpymongo._ipaddressr#Zpymongo.ismasterr$Zpymongo.monotonicr%rZpymongo.monitoringr&r'Zpymongo.networkr(r)Zpymongo.read_preferencesr*Zpymongo.server_apir+Zpymongo.server_typer,Zpymongo.socket_checkerr-Zpymongo.ssl_match_hostnamer.r/r0r1r2r7 ImportErrorrKrMr[_winregr=rGOpenKeyHKEY_LOCAL_MACHINErBrJrLr@rPrWr startswith version_infojoinlinux_distribution_namesystemmachinermac_ver win32_verjava_ver_verZ_arch system_aliasr^Z_aliasedpython_implementationrtpypy_version_infoencoderryobjectrzrrrrrXrbrcrdrkrlr5r5r5r6s   , 0                           $   $  @  _E0