U ~fhTi @sUdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd l m!Z!Gddde Z"e"e#dZ$e$j%e$j&fZ'de(d<eeegeefZ)GdddZ*ej+e$j&ej,e$j-ej.e$j/ej0e$j/ej1e$j/iZ2ddddddZ3ddddddZ4ddddd d!d"d#d$Z5dddd%d&d'd(Z6dddd)d&d*d+Z7d,d%d-d.d/Z8d0S)1zRepresent a deployment of MongoDB servers. .. seealso:: This module is compatible with both the synchronous and asynchronous PyMongo APIs. ) annotations)sample)AnyCallableListMappingMutableMapping NamedTupleOptionalcastMinKey)ObjectId)common)ConfigurationError)ReadPreference _AggWritePref _ServerModeServerDescription) Selection) SERVER_TYPE)_Addressc@s>eZdZUded<ded<ded<ded<ded<ded<dS) _TopologyTypeintSingleReplicaSetNoPrimaryReplicaSetWithPrimaryShardedUnknown LoadBalancedN)__name__ __module__ __qualname____annotations__r%r%@/tmp/pip-unpacked-wheel-36gvocj8/pymongo/topology_description.pyr-s rztuple[int, int]SRV_POLLING_TOPOLOGIESc@seZdZddddddddd d Zdd d d Zdd ddZdddddZdddddZdd ddZdd ddZ e dd ddZ e dd ddZ e dd d d!Z e dd d"d#Ze dd d$d%Ze dd d&d'Ze d(d d)d*Ze dd d+d,Ze d(d d-d.Ze dd d/d0Ze dd d1d2Ze dd d3d4Zd5d(d6d7d8ZdGdd:d;d(d<d=d>Zejfd?dd@dAdBZdd dCdDZdd dEdFZd9S)HTopologyDescriptionrz!dict[_Address, ServerDescription] Optional[str] Optional[int]Optional[ObjectId]rNone) 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. :param topology_type: initial type :param server_descriptions: dict of (address, ServerDescription) for all seeds :param replica_set_name: replica set name or None :param max_set_version: greatest setVersion seen from a primary, or None :param max_election_id: greatest electionId seen from a primary, or None :param topology_settings: a TopologySettings Ncss|]}|jdkVqdSNlogical_session_timeout_minutes.0sr%r%r& ksz/TopologyDescription.__init__..css|] }|jVqdSr5r6r8r%r%r&r;ns)_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/r0r1r2r3rEr%r%r&__init__@s"  zTopologyDescription.__init__)r4cCs|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_VERSIONaddressrBZMIN_SUPPORTED_SERVER_VERSION)rIr:Zserver_too_newZserver_too_oldr%r%r&rDrs:        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)rBrrIr%r%r&check_compatiblesz$TopologyDescription.check_compatiblerbool)rPr4cCs ||jkSr5)r>)rIrPr%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)rIrPZ unknown_sdr%r%r& reset_serversz TopologyDescription.reset_servercCsD|jtjkrtj}n|j}dd|jD}t|||j|j|j|j S)zsz-TopologyDescription.reset..) r<rCrrr>r)r=r?r@rA)rIr.sdsr%r%r&resets zTopologyDescription.resetcCs |jS)z[dict of (address, :class:`~pymongo.server_description.ServerDescription`). )r>copyrQr%r%r&r/sz'TopologyDescription.server_descriptionscCs|jS)zThe type of this topology.)r<rQr%r%r&r.sz!TopologyDescription.topology_typestrcCs tj|jS)zUThe topology type as a human readable string. .. versionadded:: 3.4 )rC_fieldsr<rQr%r%r&topology_type_namesz&TopologyDescription.topology_type_namecCs|jS)zThe replica set name.)r=rQr%r%r&r0sz$TopologyDescription.replica_set_namecCs|jS)z1Greatest setVersion seen from a primary, or None.)r?rQr%r%r&r1sz#TopologyDescription.max_set_versioncCs|jS)z1Greatest electionId seen from a primary, or None.)r@rQr%r%r&r2sz#TopologyDescription.max_election_idcCs|jS)z)Minimum logical session timeout, or None.)rFrQr%r%r&r7sz3TopologyDescription.logical_session_timeout_minuteszlist[ServerDescription]cCsdd|jDS)z)List of Servers of types besides Unknown.cSsg|]}|jr|qSr%rMr8r%r%r& sz5TopologyDescription.known_servers..r>rLrQr%r%r& known_serverssz!TopologyDescription.known_serverscCstdd|jDS)z7Whether there are any Servers of types besides Unknown.css|]}|jr|VqdSr5r_r8r%r%r&r;sz8TopologyDescription.has_known_servers..)rGr>rLrQr%r%r&has_known_serverssz%TopologyDescription.has_known_serverscCsdd|jDS)zList of readable Servers.cSsg|]}|jr|qSr%)Z is_readabler8r%r%r&r`sz8TopologyDescription.readable_servers..rarQr%r%r&rEsz$TopologyDescription.readable_serverscCs"|j}|rtdd|jDSdS)z3Minimum of all servers' max wire versions, or None.css|] }|jVqdSr5)rOr8r%r%r&r;sz:TopologyDescription.common_wire_version..N)rbrHrIZserversr%r%r&common_wire_versionsz'TopologyDescription.common_wire_versioncCs|jjSr5)rAheartbeat_frequencyrQr%r%r&rfsz'TopologyDescription.heartbeat_frequencycCs|jjSr5)rAZ_srv_max_hostsrQr%r%r& srv_max_hostssz!TopologyDescription.srv_max_hostszOptional[Selection]) selectionr4csz|sgSg}|jD]<}|jdkrBd|jd|d|j}t|||jqt||jjdfdd|jDS)Nzround_trip_time for server z is unexpectedly None: z , servers: g@@cs$g|]}tt|jkr|qSr%)r floatround_trip_timer8Zfastest thresholdr%r&r`sz>TopologyDescription._apply_local_threshold..)r/rjrPrappendrHrAZlocal_threshold_ms)rIrhZround_trip_timesserverZconfig_err_msgr%rkr&_apply_local_threshold s    z*TopologyDescription._apply_local_thresholdNzOptional[_Address]zOptional[_ServerSelector])selectorrPcustom_selectorr4cCst|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). :param 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`. :param address: A server address to select. :param custom_selector: 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 rNrzF%s requires min wire version %d, but topology's min wire version is %dN)getattrrerNr isinstancerZselection_hookr.rCrrr rbr/getrZfrom_topology_descriptionrZwith_server_descriptionsro)rIrprPrqZ common_wv descriptionrhr%r%r&apply_selectors2        z"TopologyDescription.apply_selectorr)read_preferencer4cCstd|t||S)aDoes this topology have any readable servers available matching the given read preference? :param 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 rw)rZvalidate_read_preferencerGrv)rIrwr%r%r&has_readable_serverSs 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 )rxrPRIMARYrQr%r%r&has_writable_servercsz'TopologyDescription.has_writable_servercCs0t|jddd}d|jj|jj|j|S)NcSs|jSr5)rP)sdr%r%r&oz.TopologyDescription.__repr__..)keyz-<{} id: {}, topology_type: {}, servers: {!r}>) sortedr>rLformat __class__r!rAZ _topology_idr^rdr%r%r&__repr__mszTopologyDescription.__repr__)NN)r!r"r#rJrDrRrTrWrZr/propertyr.r^r0r1r2r7rbrcrErerfrgrorvrryrxrzrr%r%r%r&r)?sL2/ 5 r)r)topology_descriptionserver_descriptionr4c 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)anReturn an updated copy of a TopologyDescription. :param topology_description: the current TopologyDescription :param 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 '{}')errorrK)#rPr.r0r1r2 server_typer/rCrrrrUr)rArrZ 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) rrrPr.set_namer1r2rrYrr%r%r&rVs              rVzlist[tuple[str, Any]])rseedlistr4cCs|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. :param topology_description: the current TopologyDescription :param seedlist: a list of new seeds new ServerDescription that resulted from a hello call r)r.r(AssertionErrorr/setkeyslistrrgrrrrHrr)r0r1r2rA)rrrYrPZ new_hostsZn_to_addr%r%r&)_updated_topology_description_srv_pollings0   rz+MutableMapping[_Address, ServerDescription]r*r+r,z.css |]}|dkrtn|VqdSr5r rr%r%r&r;Bs)r0rrPrrOZ set_versionZ election_idrUtuplerLrrr all_hostsrr) rYr0rr1r2Znew_election_tupleZmax_election_tupleZnew_election_safeZmax_election_safernZ new_addressaddrr%r%r&rsR            rr)rYr0rr4cCsJ|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)rr0rrPmer)rYr0rr%r%r&res   rztuple[int, Optional[str]]cCsztj}|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)rCrr0rrPrrr)rYr0rr.rPr%r%r&r|s     rz$Mapping[_Address, ServerDescription])rYr4cCs*|D]}|jtjkrtjSqtjS)zCurrent topology type is ReplicaSetWithPrimary. Is primary still known? Pass in a dict of ServerDescriptions. Returns new topology type. N)rLrrrrCrr)rYr:r%r%r&rs   rN)9__doc__ __future__rrandomrtypingrrrrrr r r Z bson.min_keyr Z bson.objectidrZpymongorZpymongo.errorsrZpymongo.read_preferencesrrrZpymongo.server_descriptionrZpymongo.server_selectorsrZpymongo.server_typerZpymongo.typingsrrrangerCrrr(r$_ServerSelectorr)rrrrrrrrrVrrrrrr%r%r%r&sH  (         > h+N