U d@s6dZddlmZddlmZmZmZmZmZm Z m Z m Z ddl m Z er`ddlmZddlmZe dZeeefZe eZGdd d Zddd d d Zd d d d dZd d d ddZd d d ddZd d d ddZd d d ddZdd d dddZdd d dddZdd d ddd Zdd d dd!d"Z d#S)$zFCriteria to select some ServerDescriptions from a TopologyDescription.) annotations) TYPE_CHECKINGAnyListMappingOptionalSequenceTypeVarcast) SERVER_TYPE)ServerDescription)TopologyDescriptionTc@seZdZdZedddddZddddd d d Zddd d dZddddZe ddddZ e ddddZ e ddddZ ddddZ dddddZd S)! Selectionz.Input or output of a server selector function.r )topology_descriptionreturncCs:|j}d}|D]}|jtjkr|}q(qt||j|j|SN) known_servers server_typer Z RSPrimaryrcommon_wire_version)clsrrprimarysdrHz>Selection.secondary_with_max_last_write_date..)key)secondary_server_selectorrmax)rZ secondariesrrr"secondary_with_max_last_write_dateDsz,Selection.secondary_with_max_last_write_datecCs|jr|jgng}||Sr)rr)rZ primariesrrrprimary_selectionLszSelection.primary_selectionintcCs|jjSr)rheartbeat_frequencyrrrrr)QszSelection.heartbeat_frequencycCs|jjSr)r topology_typer*rrrr+UszSelection.topology_typeboolcCs t|jSr)r,rr*rrr__bool__YszSelection.__bool__r )itemrcCs |j|Sr)r)rr.rrr __getitem__\szSelection.__getitem__N)__name__ __module__ __qualname____doc__ classmethodrrrr&propertyr'r)r+r-r/rrrrr s r) selectionrcCs|Srrr6rrrany_server_selector`sr8cCs|dd|jDS)NcSsg|]}|jr|qSr)Z is_readable.0srrr fsz,readable_server_selector..rrr7rrrreadable_server_selectordsr>cCs|dd|jDS)NcSsg|]}|jr|qSr) is_writabler9rrrr<lsz,writable_server_selector..r=r7rrrwritable_server_selectorjsr@cCs|dd|jDS)NcSsg|]}|jtjkr|qSr)rr Z RSSecondaryr9rrrr<rs z-secondary_server_selector..r=r7rrrr$psr$cCs|dd|jDS)NcSsg|]}|jtjkr|qSr)rr Z RSArbiterr9rrrr<xs z+arbiter_server_selector..r=r7rrrarbiter_server_selectorvsrAcCst|pt|S)z6Like PrimaryPreferred but doesn't use tags or latency.)r@r$r7rrr"writable_preferred_server_selector|srBTagSet)tag_setr6rcs.dddfdd |fdd|jDS)zAll servers matching one tag set. A tag set is a dict. A server matches if its tags are a superset: A server tagged {'a': '1', 'b': '2'} matches the tag set {'a': '1'}. The empty tag set {} matches any server. zMapping[str, Any]r,) server_tagsrcs0D]"\}}||ks$|||krdSqdS)NFT)items)rEr#value)rDrr tags_matchsz(apply_single_tag_set..tags_matchcsg|]}|jr|qSr)tagsr9)rHrrr<s z(apply_single_tag_set..r=)rDr6r)rDrHrapply_single_tag_sets rJTagSets)tag_setsr6rcCs*|D]}t||}|r|Sq|gS)aNAll servers match a list of tag sets. tag_sets is a list of dicts. The empty tag set {} matches any server, and may be provided at the end of the list as a fallback. So [{'a': 'value'}, {}] expresses a preference for servers tagged {'a': 'value'}, but accepts any server if none matches the first preference. )rJr)rLr6rDZ with_tag_setrrrapply_tag_setss   rMcCst|t|S)z2All near-enough secondaries matching the tag sets.)rMr$rLr6rrr#secondary_with_tags_server_selectorsrOcCst|t|S)z.All near-enough members matching the tag sets.)rMr>rNrrr member_with_tags_server_selectorsrPN)!r3 __future__rtypingrrrrrrr r Zpymongo.server_typer Zpymongo.server_descriptionr Zpymongo.topology_descriptionr rstrrCrKrr8r>r@r$rArBrJrMrOrPrrrrs( (    @