o sDiSt @sUdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZmZdd lmZmZmZmZdd lmZdd lmZdd l m!Z!dd l"m#Z#Gddde Z$e$e%dZ&e&j'e&j(fZ)de*d<eeegeefZ+GdddZ,e!j-e&j(e!j.e&j/e!j0e&j1e!j2e&j1e!j3e&j1iZ4d4ddZ5d5ddZ6d6d(d)Z7d7d+d,Z8d8d.d/Z9d9d1d2Z:d3S):zRepresent 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 PyMongoError)PrimaryReadPreference _AggWritePref _ServerModeServerDescription) Selection) SERVER_TYPE)_Addressc@s>eZdZUded<ded<ded<ded<ded<ded<dS) _TopologyTypeintSingleReplicaSetNoPrimaryReplicaSetWithPrimaryShardedUnknown LoadBalancedN)__name__ __module__ __qualname____annotations__r'r'Z/home/alumno/antzosa8961/venv/lib/python3.10/site-packages/pymongo/topology_description.pyr-s  rztuple[int, int]SRV_POLLING_TOPOLOGIESc@sVeZdZdSddZdTddZdTddZdUddZdVddZdWddZdXddZ e dYd d!Z e dZd#d$Z e d[d%d&Z e d\d'd(Ze d]d)d*Ze d\d+d,Ze d^d.d/Ze d_d0d1Ze d^d2d3Ze d^d4d5Ze d\d6d7Ze dYd8d9Ze dYd:d;Zd`d>d?Z @dadbdCdDZ @ @ @dcdddIdJZejfdedMdNZd_dOdPZdZdQdRZd@S)fTopologyDescription topology_typerserver_descriptions!dict[_Address, ServerDescription]replica_set_name Optional[str]max_set_version Optional[int]max_election_idOptional[ObjectId]topology_settingsrreturnNonecCs||_||_||_||_||_t|j|_||_d|_ |jt j kr'| |j }|s1d|_dStdd|Dr?d|_dStdd|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|]}|jduVqdSNlogical_session_timeout_minutes.0sr'r'r( lz/TopologyDescription.__init__..cs|]}|jVqdSr8r9r;r'r'r(r>os )_topology_type_replica_set_name_server_descriptions_max_set_version_max_election_idlistvalues_candidate_servers_topology_settings_incompatible_err TOPOLOGY_TYPEr"_init_incompatible_errreadable_servers_ls_timeout_minutesanymin)selfr,r-r/r1r3r5rMr'r'r(__init__@s$    zTopologyDescription.__init__cCs|jD]L}|js q|jduo|jtjk}|jduo |jtjk}|r7d|jd|jdp.d|jtjf|_ q|rQd|jd|jdpDd|jtjtj f|_ dSqdS)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).) rCrGis_server_type_knownmin_wire_versionrMAX_SUPPORTED_WIRE_VERSIONmax_wire_versionMIN_SUPPORTED_WIRE_VERSIONaddressrJMIN_SUPPORTED_SERVER_VERSION)rQr=server_too_newserver_too_oldr'r'r(rLss>        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)rJrrQr'r'r(check_compatibles z$TopologyDescription.check_compatiblerYrboolcCs ||jvSr8)rC)rQrYr'r'r( has_servers zTopologyDescription.has_servercCs|j|}t||S)z;A copy of this description, with one server marked Unknown.)rC to_unknownupdated_topology_description)rQrY unknown_sdr'r'r( reset_servers z TopologyDescription.reset_servercCsD|jtjkr tj}n|j}dd|jD}t|||j|j|j|j S)zz-TopologyDescription.reset..) rArKrrrCr+rBrDrErI)rQr,sdsr'r'r(resets zTopologyDescription.resetcCs |jS)z[dict of (address, :class:`~pymongo.server_description.ServerDescription`). )rCcopyr]r'r'r(r-s z'TopologyDescription.server_descriptionscC|jS)zThe type of this topology.)rAr]r'r'r(r,z!TopologyDescription.topology_typestrcCs tj|jS)zUThe topology type as a human readable string. .. versionadded:: 3.4 )rK_fieldsrAr]r'r'r(topology_type_names z&TopologyDescription.topology_type_namecCrj)zThe replica set name.)rBr]r'r'r(r/rkz$TopologyDescription.replica_set_namecCrj)z1Greatest setVersion seen from a primary, or None.)rDr]r'r'r(r1rkz#TopologyDescription.max_set_versioncCrj)z1Greatest electionId seen from a primary, or None.)rEr]r'r'r(r3rkz#TopologyDescription.max_election_idcCrj)z)Minimum logical session timeout, or None.)rNr]r'r'r(r:rkz3TopologyDescription.logical_session_timeout_minuteslist[ServerDescription]cCdd|jDS)z)List of Servers of types besides Unknown.cSg|]}|jr|qSr'rTr;r'r'r( rfz5TopologyDescription.known_servers..rCrGr]r'r'r( known_serversz!TopologyDescription.known_serverscCstdd|jDS)z7Whether there are any Servers of types besides Unknown.css|]}|jr|VqdSr8rrr;r'r'r(r>r?z8TopologyDescription.has_known_servers..)rOrCrGr]r'r'r(has_known_serverssz%TopologyDescription.has_known_serverscCrp)zList of readable Servers.cSrqr') is_readabler;r'r'r(rsrfz8TopologyDescription.readable_servers..rtr]r'r'r(rMrvz$TopologyDescription.readable_serverscCrj)z0List of Servers excluding deprioritized servers.)rHr]r'r'r(candidate_serversrkz%TopologyDescription.candidate_serverscCs"|j}|rtdd|jDSdS)z3Minimum of all servers' max wire versions, or None.csr@r8)rWr;r'r'r(r>sz:TopologyDescription.common_wire_version..N)rurPrQserversr'r'r(common_wire_versionsz'TopologyDescription.common_wire_versioncC|jjSr8)rIheartbeat_frequencyr]r'r'r(r~ z'TopologyDescription.heartbeat_frequencycCr}r8)rI_srv_max_hostsr]r'r'r( srv_max_hostsrz!TopologyDescription.srv_max_hosts selectionOptional[Selection]csz|sgSg}|jD]}|jdur!d|jd|d|j}t|||jq t||jjdfdd|jDS)Nzround_trip_time for server z is unexpectedly None: z , servers: g@@cs$g|]}tt|jkr|qSr')r floatround_trip_timer;fastest thresholdr'r(rss z>TopologyDescription._apply_local_threshold..)r-rrYrappendrPrIlocal_threshold_ms)rQrround_trip_timesserverconfig_err_msgr'rr(_apply_local_thresholds    z*TopologyDescription._apply_local_thresholdNdeprioritized_servers!Optional[list[ServerDescription]]csB|s|j|_dSdd|Dfdd|jD}|p|j|_dS)zBFilter out deprioritized servers from a list of server candidates.cSsh|]}|jqSr'rY)r<sdr'r'r( +sz6TopologyDescription._filter_servers..csg|] }|jvr|qSr'r)r<rdeprioritized_addressesr'r(rs,s  z7TopologyDescription._filter_servers..N)rurH)rQrfilteredr'rr(_filter_servers$s  z#TopologyDescription._filter_serversselectorOptional[_Address]custom_selectorOptional[_ServerSelector]c Cst|ddr|j}|r||jkrtd||j|ft|tr$|||jtj kr,gS|jtj tj fvr8|j S|rK| |}|rI|jrI|gSgS|||jtjkrt|tur|jD]}|jtjkrt|g}|rp||}|Sq_|r|D]}|jtjkr|g}|r||}|SqygSt|} |jtjkr|| } | s|r|dt|} || } |dur| 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 rUrzF%s requires min wire version %d, but topology's min wire version is %dN)getattrr|rUr isinstancerselection_hookr,rKr!rr"rur-getrTrrtyperrH server_typer RSPrimaryrfrom_topology_descriptionr with_server_descriptionsr) rQrrYrr common_wv descriptionrrgrr'r'r(apply_selector3s^               z"TopologyDescription.apply_selectorread_preferencercCstd|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 r)rvalidate_read_preferencerOr)rQrr'r'r(has_readable_servers 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 )rrPRIMARYr]r'r'r(has_writable_servers z'TopologyDescription.has_writable_servercCs0t|jddd}d|jj|jj|j|S)NcSrjr8r)rr'r'r(sz.TopologyDescription.__repr__..)keyz-<{} id: {}, topology_type: {}, servers: {!r}>) sortedrCrGformat __class__r#rI _topology_idrnrzr'r'r(__repr__szTopologyDescription.__repr__)r,rr-r.r/r0r1r2r3r4r5rr6r7)r6r7)rYrr6r_)rYrr6r+)r6r+)r6r.)r6r)r6rl)r6r0)r6r2)r6r4)r6ro)r6r_)rrr6ror8)rrr6r7)NNN) rrrYrrrrrr6ro)rrr6r_) r#r$r%rRrLr^r`rdrhr-propertyr,rnr/r1r3r:rurwrMryr|r~rrrrrrrrrr'r'r'r(r+?sX  3 /                   O  r+topology_descriptionserver_descriptionrr6c Cs|j}|j}|j}|j}|j}|j}|}|||<|tjkrD|dur9||jkr9t d ||j} |j | d||<t tj|||||j S|tjkro|tjtjfvrct|j jdkr]tj}n||n |tjtjfvrot|}|tjkr|tjtjfvr||nv|tjkr|tjtjfvr||nc|tjkrt|||||\}}}}nQ|tjtjtjfvrt|||\}}n>|tj kr|tjtjfvr||t!|}n'|tjkrt|||||\}}}}n|tjtjtjfvrt"|||}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 '{}')errorrS)#rYr,r/r1r3rr-rKrrrrar+rIr!r Standalone LoadBalancerlenseedspopRSGhost_SERVER_TYPE_TO_TOPOLOGY_TYPEr Mongosrr_update_rs_from_primary RSSecondary RSArbiterRSOther!_update_rs_no_primary_from_memberr_check_has_primary#_update_rs_with_primary_from_member) rrrYr,set_namer1r3rrgrr'r'r(rbs               rbseedlistlist[tuple[str, Any]]cCs|jtvsJ|}t|t|kr|St|D] }||vr(||q|jdkrRt|t|}|jt|}|dkrPt t |t |t|}ng}|D] }||vr`t |||<qTt |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*r-setkeysrFrrrrrrPrr+r/r1r3rI)rrrgrY new_hostsn_to_addr'r'r()_updated_topology_description_srv_pollings4    rrg+MutableMapping[_Address, ServerDescription]r/r0r1r2r3r4|j|jf}||f}t dd|D}t dd|D}||kr|td|d|||j<t||||fS|j}|j}| D]} | j t j ur| j|jkr| td|| j<nq|jD] } | |vrt| || <qt||jD]} || 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). Nzt z*_update_rs_from_primary..csrr8r rr'r'r(r>urz6primary marked stale due to discovery of newer primary)r/rrYrrW set_version election_idrartuplerGrrr all_hostsrr) rgr/rr1r3new_election_tuplemax_election_tuplenew_election_safemax_election_safer new_addressaddrr'r'r(rFsb              rrcCsP|dusJ||jkr||jt|S|jr$|j|jkr$||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)r/rrYmer)rgr/rr'r'r(rs   rtuple[int, Optional[str]]cCsztj}|dur |j}n||jkr||j||fS|jD] }||vr)t|||<q|jr9|j|jkr9||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)rKrr/rrYrrr)rgr/rr,rYr'r'r(rs      r$Mapping[_Address, ServerDescription]cCs*|D] }|jtjkrtjSqtjS)zCurrent topology type is ReplicaSetWithPrimary. Is primary still known? Pass in a dict of ServerDescriptions. Returns new topology type. )rGrrrrKrr)rgr=r'r'r(rs   rN)rr+rrr6r+)rr+rrr6r+) rgrr/r0rrr1r2r3r4r6r)rgrr/r0rrr6r)rgrr/r0rrr6r)rgrr6r);__doc__ __future__rrandomrtypingrrrrrr r r bson.min_keyr bson.objectidrpymongorpymongo.errorsrrpymongo.read_preferencesrrrrpymongo.server_descriptionrpymongo.server_selectorsrpymongo.server_typerpymongo.typingsrrrangerKr!r r*r&_ServerSelectorr+rrrrrrrrrbrrrrrr'r'r'r(s@  (       m h + X