3 O6bH@sdZddlmZmZddlmZddlmZddlm Z m Z dZ dZ dZ dZd Zd1ZddZddZddZddZGdddeZGdddeZGdddeZGdddeZGdd d eZGd!d"d"eZeeeeefZd3d#d$Zd4ZGd*d+d+eZd,d-Z Gd.d/d/eZ!d0S)5zBUtilities for choosing which member of a replica set to read from.)abc integer_types)max_staleness_selectors)ConfigurationError) member_with_tags_server_selector#secondary_with_tags_server_selectorprimaryprimaryPreferred secondarysecondaryPreferrednearestcCsj|dkr |St|ts$td|ft|dkr>td|fx&|D]}t|tjsDtd|fqDW|S)z3Validate tag sets for a MongoReplicaSetClient. Nz#Tag sets %r invalid, must be a listrzETag sets %r invalid, must be None or contain at least one set of tagszqTag set %r invalid, must be an instance of dict, bson.son.SON or other type that inherits from collection.Mapping) isinstancelist TypeErrorlen ValueErrorrMapping)tag_setstagsrW/var/www/html/sandeepIITI/myenv/lib/python3.6/site-packages/pymongo/read_preferences.py_validate_tag_sets(s      rcCsd|S)Nz6maxStalenessSeconds must be a positive integer, not %sr) max_stalenessrrr_invalid_max_staleness_msg@srcCs:|dkr dSt|ts"tt||dkr6tt||S)zValidate max_staleness.rrr)rrrrr)rrrr_validate_max_stalenessFs   rcCs(|dkr dSt|ts$td|f|S)zValidate hedge.Nz"hedge must be a dictionary, not %r)rdictr)hedgerrr_validate_hedgeTs  r"c@seZdZdZd%Zd'd d Zed d Zed dZeddZ eddZ eddZ eddZ eddZ eddZddZddZdd Zd!d"Zd#d$ZdS)( _ServerModez)Base class for all read preferences. __mongos_mode__mode __tag_sets__max_staleness__hedgeNrcCs2t||_||_t||_t||_t||_dS)N) _MONGOS_MODES_ServerMode__mongos_mode_ServerMode__moder_ServerMode__tag_setsr_ServerMode__max_stalenessr"_ServerMode__hedge)selfmoderrr!rrr__init__fs    z_ServerMode.__init__cCs|jjS)z*The name of this read preference. ) __class____name__)r/rrrnamemsz_ServerMode.namecCs|jS)z1The mongos mode of this read preference. )r*)r/rrr mongos_modessz_ServerMode.mongos_modecCsTd|ji}|jdigfkr$|j|d<|jdkr8|j|d<|jdifkrP|j|d<|S)z'Read preference as a document. r0NrrZmaxStalenessSecondsr!r)r*r,r-r.)r/docrrrdocumentys     z_ServerMode.documentcCs|jS)z3The mode of this read preference instance. )r+)r/rrrr0sz_ServerMode.modecCs|jrt|jSigS)a{Set ``tag_sets`` to a list of dictionaries like [{'dc': 'ny'}] to read only from members whose ``dc`` tag has the value ``"ny"``. To specify a priority-order for tag sets, provide a list of tag sets: ``[{'dc': 'ny'}, {'dc': 'la'}, {}]``. A final, empty tag set, ``{}``, means "read from any member that matches the mode, ignoring tags." MongoReplicaSetClient tries each set of tags in turn until it finds a set of tags with at least one matching member. .. seealso:: `Data-Center Awareness `_ )r,r)r/rrrrs z_ServerMode.tag_setscCs|jS)zThe maximum estimated length of time (in seconds) a replica set secondary can fall behind the primary in replication before it will no longer be selected for operations, or -1 for no maximum.)r-)r/rrrrsz_ServerMode.max_stalenesscCs|jS)aThe read preference ``hedge`` parameter. A dictionary that configures how the server will perform hedged reads. It consists of the following keys: - ``enabled``: Enables or disables hedged reads in sharded clusters. Hedged reads are automatically enabled in MongoDB 4.4+ when using a ``nearest`` read preference. To explicitly enable hedged reads, set the ``enabled`` key to ``true``:: >>> Nearest(hedge={'enabled': True}) To explicitly disable hedged reads, set the ``enabled`` key to ``False``:: >>> Nearest(hedge={'enabled': False}) .. versionadded:: 3.11 )r.)r/rrrr!sz_ServerMode.hedgecCs|jdkrdSdS)aThe wire protocol version the server must support. Some read preferences impose version requirements on all servers (e.g. maxStalenessSeconds requires MongoDB 3.4 / maxWireVersion 5). All servers' maxWireVersion must be at least this read preference's `min_wire_version`, or the driver raises :exc:`~pymongo.errors.ConfigurationError`. rrr)r-)r/rrrmin_wire_versions z_ServerMode.min_wire_versioncCsd|j|j|j|jfS)Nz+%s(tag_sets=%r, max_staleness=%r, hedge=%r))r4r,r-r.)r/rrr__repr__sz_ServerMode.__repr__cCs>t|tr:|j|jko8|j|jko8|j|jko8|j|jkStS)N)rr#r0rrr!NotImplemented)r/otherrrr__eq__s      z_ServerMode.__eq__cCs ||k S)Nr)r/r<rrr__ne__sz_ServerMode.__ne__cCs|j|j|j|jdS)zeReturn value of object for pickling. Needed explicitly because __slots__() defined. )r0rrr!)r+r,r-r.)r/rrr __getstate__sz_ServerMode.__getstate__cCsD|d|_t|j|_t|d|_t|d|_t|d|_dS)zRestore from pickling.r0rrr!N) r+r)r*rr,rr-r"r.)r/valuerrr __setstate__s   z_ServerMode.__setstate__)r$r%r&r'r(r)NrN)r3 __module__ __qualname____doc__ __slots__r1propertyr4r5r7r0rrr!r9r:r=r>r?rArrrrr#_s"         r#cs<eZdZdZfZfddZddZddZdd ZZ S) PrimaryaGPrimary read preference. * When directly connected to one mongod queries are allowed if the server is standalone or a replica set primary. * When connected to a mongos queries are sent to the primary of a shard. * When connected to a replica set queries are sent to the primary of the replica set. cstt|jtdS)N)superrGr1_PRIMARY)r/)r2rrr1szPrimary.__init__cCs|jS)z*Apply this read preference to a Selection.)primary_selection)r/ selectionrrr__call__szPrimary.__call__cCsdS)Nz Primary()r)r/rrrr:szPrimary.__repr__cCst|tr|jtkStS)N)rr#r0rIr;)r/r<rrrr=s  zPrimary.__eq__) r3rBrCrDrEr1rLr:r= __classcell__rr)r2rrGs  rGcs.eZdZdZfZd fdd ZddZZS) PrimaryPreferredaPrimaryPreferred read preference. * When directly connected to one mongod queries are allowed to standalone servers, to a replica set primary, or to replica set secondaries. * When connected to a mongos queries are sent to the primary of a shard if available, otherwise a shard secondary. * When connected to a replica set queries are sent to the primary if available, otherwise a secondary. .. note:: When a :class:`~pymongo.mongo_client.MongoClient` is first created reads will be routed to an available secondary until the primary of the replica set is discovered. :Parameters: - `tag_sets`: The :attr:`~tag_sets` to use if the primary is not available. - `max_staleness`: (integer, in seconds) The maximum estimated length of time a replica set secondary can fall behind the primary in replication before it will no longer be selected for operations. Default -1, meaning no maximum. If it is set, it must be at least 90 seconds. - `hedge`: The :attr:`~hedge` to use if the primary is not available. .. versionchanged:: 3.11 Added ``hedge`` parameter. Nrcstt|jt|||dS)N)rHrNr1_PRIMARY_PREFERRED)r/rrr!)r2rrr1#s zPrimaryPreferred.__init__cCs&|jr |jSt|jtj|j|SdS)z(Apply this read preference to Selection.N)r rJrrrselectr)r/rKrrrrL's zPrimaryPreferred.__call__r)NrN)r3rBrCrDrEr1rLrMrr)r2rrNsrNcs.eZdZdZfZd fdd ZddZZS) SecondaryaSecondary read preference. * When directly connected to one mongod queries are allowed to standalone servers, to a replica set primary, or to replica set secondaries. * When connected to a mongos queries are distributed among shard secondaries. An error is raised if no secondaries are available. * When connected to a replica set queries are distributed among secondaries. An error is raised if no secondaries are available. :Parameters: - `tag_sets`: The :attr:`~tag_sets` for this read preference. - `max_staleness`: (integer, in seconds) The maximum estimated length of time a replica set secondary can fall behind the primary in replication before it will no longer be selected for operations. Default -1, meaning no maximum. If it is set, it must be at least 90 seconds. - `hedge`: The :attr:`~hedge` for this read preference. .. versionchanged:: 3.11 Added ``hedge`` parameter. Nrcstt|jt|||dS)N)rHrQr1 _SECONDARY)r/rrr!)r2rrr1Ks zSecondary.__init__cCst|jtj|j|S)z(Apply this read preference to Selection.)rrrrPr)r/rKrrrrLOszSecondary.__call__r)NrN)r3rBrCrDrEr1rLrMrr)r2rrQ2srQcs.eZdZdZfZd fdd ZddZZS) SecondaryPreferredaSecondaryPreferred read preference. * When directly connected to one mongod queries are allowed to standalone servers, to a replica set primary, or to replica set secondaries. * When connected to a mongos queries are distributed among shard secondaries, or the shard primary if no secondary is available. * When connected to a replica set queries are distributed among secondaries, or the primary if no secondary is available. .. note:: When a :class:`~pymongo.mongo_client.MongoClient` is first created reads will be routed to the primary of the replica set until an available secondary is discovered. :Parameters: - `tag_sets`: The :attr:`~tag_sets` for this read preference. - `max_staleness`: (integer, in seconds) The maximum estimated length of time a replica set secondary can fall behind the primary in replication before it will no longer be selected for operations. Default -1, meaning no maximum. If it is set, it must be at least 90 seconds. - `hedge`: The :attr:`~hedge` for this read preference. .. versionchanged:: 3.11 Added ``hedge`` parameter. Nrcstt|jt|||dS)N)rHrSr1_SECONDARY_PREFERRED)r/rrr!)r2rrr1ts zSecondaryPreferred.__init__cCs(t|jtj|j|}|r|S|jSdS)z(Apply this read preference to Selection.N)rrrrPrrJ)r/rKZ secondariesrrrrLxs zSecondaryPreferred.__call__r)NrN)r3rBrCrDrEr1rLrMrr)r2rrSWsrScs.eZdZdZfZd fdd ZddZZS) NearestavNearest read preference. * When directly connected to one mongod queries are allowed to standalone servers, to a replica set primary, or to replica set secondaries. * When connected to a mongos queries are distributed among all members of a shard. * When connected to a replica set queries are distributed among all members. :Parameters: - `tag_sets`: The :attr:`~tag_sets` for this read preference. - `max_staleness`: (integer, in seconds) The maximum estimated length of time a replica set secondary can fall behind the primary in replication before it will no longer be selected for operations. Default -1, meaning no maximum. If it is set, it must be at least 90 seconds. - `hedge`: The :attr:`~hedge` for this read preference. .. versionchanged:: 3.11 Added ``hedge`` parameter. Nrcstt|jt|||dS)N)rHrUr1_NEAREST)r/rrr!)r2rrr1s zNearest.__init__cCst|jtj|j|S)z(Apply this read preference to Selection.)rrrrPr)r/rKrrrrLszNearest.__call__r)NrN)r3rBrCrDrEr1rLrMrr)r2rrUsrUcCsB|tkr4|digfkrtd|dkr.tdtSt|||S)Nz4Read preference primary cannot be combined with tagsrzCRead preference primary cannot be combined with maxStalenessSecondsr)rIrrG_ALL_READ_PREFERENCES)r0rrrrrmake_read_preferencesrXPRIMARYPRIMARY_PREFERRED SECONDARYSECONDARY_PREFERREDNEARESTc@s.eZdZdZeZeZeZ e Z e Z dS)ReadPreferenceaDAn enum that defines the read preference modes supported by PyMongo. See :doc:`/examples/high_availability` for code examples. A read preference is used in three cases: :class:`~pymongo.mongo_client.MongoClient` connected to a single mongod: - ``PRIMARY``: Queries are allowed if the server is standalone or a replica set primary. - All other modes allow queries to standalone servers, to a replica set primary, or to replica set secondaries. :class:`~pymongo.mongo_client.MongoClient` initialized with the ``replicaSet`` option: - ``PRIMARY``: Read from the primary. This is the default, and provides the strongest consistency. If no primary is available, raise :class:`~pymongo.errors.AutoReconnect`. - ``PRIMARY_PREFERRED``: Read from the primary if available, or if there is none, read from a secondary. - ``SECONDARY``: Read from a secondary. If no secondary is available, raise :class:`~pymongo.errors.AutoReconnect`. - ``SECONDARY_PREFERRED``: Read from a secondary if available, otherwise from the primary. - ``NEAREST``: Read from any member. :class:`~pymongo.mongo_client.MongoClient` connected to a mongos, with a sharded cluster of replica sets: - ``PRIMARY``: Read from the primary of the shard, or raise :class:`~pymongo.errors.OperationFailure` if there is none. This is the default. - ``PRIMARY_PREFERRED``: Read from the primary of the shard, or if there is none, read from a secondary of the shard. - ``SECONDARY``: Read from a secondary of the shard, or raise :class:`~pymongo.errors.OperationFailure` if there is none. - ``SECONDARY_PREFERRED``: Read from a secondary of the shard if available, otherwise from the shard primary. - ``NEAREST``: Read from any shard member. N)r3rBrCrDrGrYrNrZrQr[rSr\rUr]rrrrr^s 1r^cCs tj|S)z7Get the read preference mode from mongos/uri name. )r)index)r4rrrread_pref_mode_from_namesr`c@s0eZdZdZddZddZddZdd Zd S) MovingAveragez0Tracks an exponentially-weighted moving average.cCs d|_dS)N)average)r/rrrr1szMovingAverage.__init__cCs6|dkr dS|jdkr||_nd|jd||_dS)Nrg?g?)rb)r/samplerrr add_samples  zMovingAverage.add_samplecCs|jS)z6Get the calculated average, or None if no samples yet.)rb)r/rrrgetszMovingAverage.getcCs d|_dS)N)rb)r/rrrresetszMovingAverage.resetN)r3rBrCrDr1rdrerfrrrrras  raN)r r rrrr)r)rYrZr[r\r])"rDZbson.py3compatrrZpymongorZpymongo.errorsrZpymongo.server_selectorsrrrIrOrRrTrVr)rrrr"objectr#rGrNrQrSrUrWrXZ_MODESr^r`rarrrrsH    -%.%  9