U ~fh@s2dZddlmZddlmZmZmZmZmZm Z m Z ddl m Z er\ddl mZddlmZe dZeeefZeeZGdd 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"Zd#S)$zFCriteria to select some ServerDescriptions from a TopologyDescription.) annotations) TYPE_CHECKINGAnyMappingOptionalSequenceTypeVarcast) 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|Srrr5rrrany_server_selector`sr7cCs|dd|jDS)NcSsg|]}|jr|qSr)Z is_readable.0srrr fsz,readable_server_selector..rrr6rrrreadable_server_selectordsr=cCs|dd|jDS)NcSsg|]}|jr|qSr) is_writabler8rrrr;lsz,writable_server_selector..r<r6rrrwritable_server_selectorjsr?cCs|dd|jDS)NcSsg|]}|jtjkr|qSr)rr Z RSSecondaryr8rrrr;rs z-secondary_server_selector..r<r6rrrr#psr#cCs|dd|jDS)NcSsg|]}|jtjkr|qSr)rr Z RSArbiterr8rrrr;xs z+arbiter_server_selector..r<r6rrrarbiter_server_selectorvsr@cCst|pt|S)z6Like PrimaryPreferred but doesn't use tags or latency.)r?r#r6rrr"writable_preferred_server_selector|srATagSet)tag_setr5rcs.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)rDr"value)rCrr tags_matchsz(apply_single_tag_set..tags_matchcsg|]}|jr|qSr)tagsr8)rGrrr;s z(apply_single_tag_set..r<)rCr5r)rCrGrapply_single_tag_sets rITagSets)tag_setsr5rcCs*|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. )rIr)rKr5rCZ with_tag_setrrrapply_tag_setss   rLcCst|t|S)z2All near-enough secondaries matching the tag sets.)rLr#rKr5rrr#secondary_with_tags_server_selectorsrNcCst|t|S)z.All near-enough members matching the tag sets.)rLr=rMrrr member_with_tags_server_selectorsrON) r2 __future__rtypingrrrrrrr Zpymongo.server_typer Zpymongo.server_descriptionr Zpymongo.topology_descriptionr r strrBrJrr7r=r?r#r@rArIrLrNrOrrrrs( $    @