U dg @sUdZddlmZddlmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd l m!Z!Gd dde Z"e"e#dZ$e$j%e$j&fZ'e e(e(fe)d<eeegeefZ*GdddZ+ej,e$j&ej-e$j.ej/e$j0ej1e$j0ej2e$j0iZ3e+eddddZ4e+ee e5efe+dddZ6e e!efe e5ee e(e ee e(e e5e e(e efdddZ7e e!efe e5ee(dddZ8e e!efe e5ee e(e e5fddd Z9ee!efe(d!d"d#Z:d$S)%z*Represent a deployment of MongoDB servers.)sample) AnyCallableDictListMappingMutableMapping NamedTupleOptionalTuplecastMinKey)ObjectId)common)ConfigurationError)ReadPreference _AggWritePref _ServerModeServerDescription) Selection) SERVER_TYPE)_Addressc@s>eZdZUeed<eed<eed<eed<eed<eed<dS) _TopologyTypeSingleReplicaSetNoPrimaryReplicaSetWithPrimaryShardedUnknown LoadBalancedN)__name__ __module__ __qualname__int__annotations__r&r&@/tmp/pip-unpacked-wheel-oblwsawz/pymongo/topology_description.pyr+s rSRV_POLLING_TOPOLOGIESc@seZdZeeeefeeeeee e ddddZ ddddZ dddd Z eed d d Zedd d dZddddZeeefdddZeedddZeedddZeeedddZeeedddZeee dddZeeedddZeeeddd Zeedd!d"Zeeedd#d$Zeeedd%d&Zeedd'd(Zeedd)d*Z ee!eed+d,d-Z"d8e eeee#eed.d/d0Z$e%j&fe'ed1d2d3Z(edd4d5Z)edd6d7Z*dS)9TopologyDescriptionN) topology_typeserver_descriptionsreplica_set_namemax_set_versionmax_election_idtopology_settingsreturncCs||_||_||_||_||_||_d|_|jtjkr>| |j }|sPd|_ n.t dd|Drjd|_ nt dd|D|_ dS)aRepresentation of a deployment of MongoDB servers. :Parameters: - `topology_type`: initial type - `server_descriptions`: dict of (address, ServerDescription) for all seeds - `replica_set_name`: replica set name or None - `max_set_version`: greatest setVersion seen from a primary, or None - `max_election_id`: greatest electionId seen from a primary, or None - `topology_settings`: a TopologySettings Ncss|]}|jdkVqdSNlogical_session_timeout_minutes.0sr&r&r' jsz/TopologyDescription.__init__..css|] }|jVqdSr2r3r5r&r&r'r8ms)_topology_type_replica_set_name_server_descriptions_max_set_version_max_election_id_topology_settings_incompatible_err TOPOLOGY_TYPEr _init_incompatible_errreadable_servers_ls_timeout_minutesanymin)selfr+r,r-r.r/r0rBr&r&r'__init__>s"  zTopologyDescription.__init__)r1cCs|jD]}|jsq |jdk o*|jtjk}|jdk o@|jtjk}|rnd|jd|jdp\d|jtjf|_ q |r d|jd|jdpd|jtjtj f|_ qq dS)z>Internal compatibility check for non-load balanced topologies.Nz]Server at %s:%d requires wire version %d, but this version of PyMongo only supports up to %d.rzgServer at %s:%d reports wire version %d, but this version of PyMongo requires at least %d (MongoDB %s).) r;valuesis_server_type_knownmin_wire_versionrZMAX_SUPPORTED_WIRE_VERSIONmax_wire_versionZMIN_SUPPORTED_WIRE_VERSIONaddressr?ZMIN_SUPPORTED_SERVER_VERSION)rFr7Zserver_too_newZserver_too_oldr&r&r'rAqs:        z*TopologyDescription._init_incompatible_errcCs|jrt|jdS)zRaise ConfigurationError if any server is incompatible. A server is incompatible if its wire protocol version range does not overlap with PyMongo's. N)r?rrFr&r&r'check_compatiblesz$TopologyDescription.check_compatible)rMr1cCs ||jkSr2)r;)rFrMr&r&r' has_serverszTopologyDescription.has_servercCs|j|}t||S)z;A copy of this description, with one server marked Unknown.)r; to_unknownupdated_topology_description)rFrMZ unknown_sdr&r&r' reset_serversz TopologyDescription.reset_servercCsD|jtjkrtj}n|j}dd|jD}t|||j|j|j|j S)zsz-TopologyDescription.reset..) r9r@rrr;r*r:r<r=r>)rFr+sdsr&r&r'resets zTopologyDescription.resetcCs |jS)z[Dict of (address, :class:`~pymongo.server_description.ServerDescription`). )r;copyrNr&r&r'r,sz'TopologyDescription.server_descriptionscCs|jS)zThe type of this topology.)r9rNr&r&r'r+sz!TopologyDescription.topology_typecCs tj|jS)zUThe topology type as a human readable string. .. versionadded:: 3.4 )r@_fieldsr9rNr&r&r'topology_type_namesz&TopologyDescription.topology_type_namecCs|jS)zThe replica set name.)r:rNr&r&r'r-sz$TopologyDescription.replica_set_namecCs|jS)z1Greatest setVersion seen from a primary, or None.)r<rNr&r&r'r.sz#TopologyDescription.max_set_versioncCs|jS)z1Greatest electionId seen from a primary, or None.)r=rNr&r&r'r/sz#TopologyDescription.max_election_idcCs|jS)z)Minimum logical session timeout, or None.)rCrNr&r&r'r4sz3TopologyDescription.logical_session_timeout_minutescCsdd|jDS)z)List of Servers of types besides Unknown.cSsg|]}|jr|qSr&rJr5r&r&r' sz5TopologyDescription.known_servers..r;rIrNr&r&r' known_serverssz!TopologyDescription.known_serverscCstdd|jDS)z7Whether there are any Servers of types besides Unknown.css|]}|jr|VqdSr2rZr5r&r&r'r8sz8TopologyDescription.has_known_servers..)rDr;rIrNr&r&r'has_known_serverssz%TopologyDescription.has_known_serverscCsdd|jDS)zList of readable Servers.cSsg|]}|jr|qSr&)Z is_readabler5r&r&r'r[sz8TopologyDescription.readable_servers..r\rNr&r&r'rBsz$TopologyDescription.readable_serverscCs"|j}|rtdd|jDSdS)z3Minimum of all servers' max wire versions, or None.css|] }|jVqdSr2)rLr5r&r&r'r8sz:TopologyDescription.common_wire_version..N)r]rErFZserversr&r&r'common_wire_versionsz'TopologyDescription.common_wire_versioncCs|jjSr2)r>heartbeat_frequencyrNr&r&r'rasz'TopologyDescription.heartbeat_frequencycCs|jjSr2)r>Z_srv_max_hostsrNr&r&r' srv_max_hostssz!TopologyDescription.srv_max_hosts) selectionr1cs>|sgStdd|jD|jjdfdd|jDS)Ncss|]}tt|jVqdSr2r floatZround_trip_timer5r&r&r'r8sz=TopologyDescription._apply_local_threshold..g@@cs$g|]}tt|jkr|qSr&rdr5Zfastest thresholdr&r'r[sz>TopologyDescription._apply_local_threshold..)rEr,r>Zlocal_threshold_ms)rFrcr&rfr'_apply_local_threshold s  z*TopologyDescription._apply_local_threshold)selectorrMcustom_selectorr1cCst|ddr4|j}|r4||jkr4td||j|ft|trH|||jtj krXgS|jtj tj fkrp|j S|r| |}|r|gSgSt|}|jtjkr||}|dk r|r|||j }||S)aList of servers matching the provided selector(s). :Parameters: - `selector`: a callable that takes a Selection as input and returns a Selection as output. For example, an instance of a read preference from :mod:`~pymongo.read_preferences`. - `address` (optional): A server address to select. - `custom_selector` (optional): A callable that augments server selection rules. Accepts a list of :class:`~pymongo.server_description.ServerDescription` objects and return a list of server descriptions that should be considered suitable for the desired operation. .. versionadded:: 3.4 rKrzF%s requires min wire version %d, but topology's min wire version is %dN)getattrr`rKr isinstancerZselection_hookr+r@rrr r]r,getrZfrom_topology_descriptionrZwith_server_descriptionsrh)rFrirMrjZ common_wv descriptionrcr&r&r'apply_selectors2        z"TopologyDescription.apply_selector)read_preferencer1cCstd|t||S)aDoes this topology have any readable servers available matching the given read preference? :Parameters: - `read_preference`: an instance of a read preference from :mod:`~pymongo.read_preferences`. Defaults to :attr:`~pymongo.read_preferences.ReadPreference.PRIMARY`. .. note:: When connected directly to a single server this method always returns ``True``. .. versionadded:: 3.4 rp)rZvalidate_read_preferencerDro)rFrpr&r&r'has_readable_serverMs z'TopologyDescription.has_readable_servercCs |tjS)zDoes this topology have a writable server available? .. note:: When connected directly to a single server this method always returns ``True``. .. versionadded:: 3.4 )rqrPRIMARYrNr&r&r'has_writable_server^sz'TopologyDescription.has_writable_servercCs0t|jddd}d|jj|jj|j|S)NcSs|jSr2)rM)sdr&r&r'jz.TopologyDescription.__repr__..)keyz-<{} id: {}, topology_type: {}, servers: {!r}>) sortedr;rIformat __class__r!r>Z _topology_idrYr_r&r&r'__repr__hszTopologyDescription.__repr__)NN)+r!r"r#r$rrrr strrrrGrArOboolrPrSrVr,propertyr+rYr-r.r/r4rr]r^rBr`rarbrrh_ServerSelectorrorrrrrqrsr{r&r&r&r'r*=sd  3/  6 r*)topology_descriptionserver_descriptionr1c Cs|j}|j}|j}|j}|j}|j}|}|||<|tjkr|dk rr||jkrrt d ||j} |j | d||<t tj|||||j S|tjkr|tjtjfkrt|j jdkrtj}q||n|tjtjfkrt|}|tjkr|tjtjfkr||n|tjkr|tjtjfkr2||nL|tjkrXt|||||\}}}}n&|tjtjtjfkrt|||\}}n|tj kr|tjtjfkr||t!|}nR|tjkrt|||||\}}}}n,|tjtjtjfkrt"|||}nt!|}t ||||||j S)a}Return an updated copy of a TopologyDescription. :Parameters: - `topology_description`: the current TopologyDescription - `server_description`: a new ServerDescription that resulted from a hello call Called after attempting (successfully or not) to call hello on the server at server_description.address. Does not modify topology_description. Nzeclient is configured to connect to a replica set named '{}' but this node belongs to a set named '{}')errorrH)#rMr+r-r.r/ server_typer,r@rrryrQr*r>rrZ StandaloneZ LoadBalancerlenZseedspopZRSGhost_SERVER_TYPE_TO_TOPOLOGY_TYPErMongosr RSPrimary_update_rs_from_primary RSSecondary RSArbiterRSOther!_update_rs_no_primary_from_memberr_check_has_primary#_update_rs_with_primary_from_member) rrrMr+set_namer.r/rrUrr&r&r'rRs              rR)rseedlistr1cCs|jtkst|}t|t|kr.|St|D]}||kr:||q:|jdkrt|t|}|jt |}|dkrt t |t |t |}ng}|D]}||krt |||<qt|j||j|j|j|jS)zReturn an updated copy of a TopologyDescription. :Parameters: - `topology_description`: the current TopologyDescription - `seedlist`: a list of new seeds new ServerDescription that resulted from a hello call r)r+r)AssertionErrorr,setkeyslistrrbrrrxrErr*r-r.r/r>)rrrUrMZ new_hostsZn_to_addr&r&r')_updated_topology_description_srv_pollings0   r)rUr-rr.r/r1c Cs|dkr|j}n&||jkr6||jt||||fS|jdksJ|jdkr|j|jf}||f}d|krd|kr||kr|||j<t||||fS|j}|jdk r|dks|j|kr|j}nl|j|jf}||f}tdd|D}tdd|D}||kr|||j<t||||fS|j}|j}| D]6} | j t j kr2| j|jkr2| || j<qjq2|j D]} | |krpt| || <qpt||j D]} || qt||||fS)agUpdate topology description from a primary's hello response. Pass in a dict of ServerDescriptions, current replica set name, the ServerDescription we are processing, and the TopologyDescription's max_set_version and max_election_id if any. Returns (new topology type, new replica_set_name, new max_set_version, new max_election_id). Ncss |]}|dkrtn|VqdSr2r r6ir&r&r'r8>sz*_update_rs_from_primary..css |]}|dkrtn|VqdSr2r rr&r&r'r8?s)r-rrMrrLZ set_versionZ election_idrQtuplerIrrr all_hostsrr) rUr-rr.r/Znew_election_tupleZmax_election_tupleZnew_election_safeZmax_election_safeserverZ new_addressaddrr&r&r'rsR            r)rUr-rr1cCsJ|dk s t||jkr$||jn|jrB|j|jkrB||jt|S)zRS with known primary. Process a response from a non-primary. Pass in a dict of ServerDescriptions, current replica set name, and the ServerDescription we are processing. Returns new topology type. N)rr-rrMmer)rUr-rr&r&r'rcs   rcCsztj}|dkr|j}n||jkr4||j||fS|jD]}||kr:t|||<q:|jrr|j|jkrr||j||fS)zRS without known primary. Update from a non-primary's response. Pass in a dict of ServerDescriptions, current replica set name, and the ServerDescription we are processing. Returns (new topology type, new replica_set_name). N)r@rr-rrMrrr)rUr-rr+rMr&r&r'rzs     r)rUr1cCs*|D]}|jtjkrtjSqtjS)zCurrent topology type is ReplicaSetWithPrimary. Is primary still known? Pass in a dict of ServerDescriptions. Returns new topology type. N)rIrrrr@rr)rUr7r&r&r'rs   rN);__doc__randomrtypingrrrrrrr r r r Z bson.min_keyrZ bson.objectidrZpymongorZpymongo.errorsrZpymongo.read_preferencesrrrZpymongo.server_descriptionrZpymongo.server_selectorsrZpymongo.server_typerZpymongo.typingsrrranger@rrr)r$r%rr*rrrrrrrrrRr|rrrrrr&r&r&r'sn 0         ;  j -  P