3 O6b{-@sdZddlZddlZddlmZddlmZddlmZm Z ddl m Z m Z m Z mZmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+yddl,m-Z-ee-fZ.Wne/k r,efZ.YnXerBddl0m1Z1n ddl2m1Z1dZ3de3Z4dZ5dZ6dZ7dZ8dZ9dZ:dZ;dZdZ?dZ@d ZAd!ZBdZCdZDdZEdZFd"ZGd#ZHd#ZIdZJdZKd'ZLd(d)ZMd*d+ZNd,d-ZOejPejQejRejSejTd.ZUd/d0ZVd1d2ZWd3d4ZXd5d6ZYd7d8ZZd9d:Z[d;d<Z\d=d>Z]d?d@Z^dAdBZ_dCdDZ`dEdFZadGdHZbdIdJZcdKdLZddMdNZedOdPZfdQdRZgdSdTZhdUdVZidWdXZjdYdZZkd[d\Zlemd]d^d_d`gZndadbZodcddZpdedfZqdgdhZrdidjZsdkdlZtdmdnZudodpZvdqdrZwdsdtZxdudvZydwdxZzdydzZ{d{d|Z|emd}d~dgZ}ddZ~ddZdgdgdgdgdgdgdgdgdgd Zewejeoe^eefeWedeWecede\ege_eiele_eWeWeWeeefe[eWe(e'ddeWe[e[e_ddeWeaeZed$ZeWexeyeWeZeWe[eWe~eke]edd Zepeqehe ee_e_ezed Zddddddddddd Zddddddddddd Zejex)r?r@r3r3r4validate_integers  rFcCs$t||}|dkr td|f|S)zKValidate that 'value' is a positive integer, which does not include 0. rz*The value of %s must be a positive integer)rFrD)r?r@valr3r3r4validate_positive_integers   rHcCs$t||}|dkr td|f|S)z6Validate that 'value' is a positive integer or 0. rz.The value of %s must be a non negative integer)rFrD)r?r@rGr3r3r4validate_non_negative_integers   rIcCs(|dkr |St||}t|dj|S)z6Validates that 'value' is file-like and readable. Nr)validate_stringopenclose)r?r@r3r3r4validate_readables  rNcCs|dkr |St||S)z9Validate that 'value' is a positive integer or None. N)rH)r?r@r3r3r4!validate_positive_integer_or_nonesrOcCs|dkr |St||S)z>Validate that 'value' is a positive integer or 0 or None. N)rI)r?r@r3r3r4%validate_non_negative_integer_or_nonesrPcCs$t|tr|Std|tjfdS)zbValidates that 'value' is an instance of `basestring` for Python 2 or `str` for Python 3. z2Wrong type for %s, value must be an instance of %sN)r<r r>__name__)r?r@r3r3r4rKs rKcCs|dkr |St||S)zEValidates that 'value' is an instance of `basestring` or `None`. N)rK)r?r@r3r3r4validate_string_or_nonesrRc CsHt|tr|St|tr6yt|Stk r4|SXtd|fdS)z4Validates that 'value' is an integer or string. z7Wrong type for %s, value must be an integer or a stringN)r<rr r-rDr>)r?r@r3r3r4validate_int_or_basestrings  rSc CsVt|tr|St|trDy t|}Wntk r8|SXt||Std|fdS)z4Validates that 'value' is an integer or string. zDWrong type for %s, value must be an non negative integer or a stringN)r<rr r-rDrIr>)r?r@rGr3r3r4'validate_non_negative_int_or_basestrings    rTc Csxd|f}y t|}Wn6tk r2t|Yntk rLt|YnXd|ko`dknsttd|f|S)z[Validates that 'value' is a float, or can be converted to one, and is positive. z%s must be an integer or floatrgeAz3%s must be greater than 0 and less than one billion)floatrDr>)r?r@errmsgr3r3r4validate_positive_floats    rWcCs|dks|dkrdSt||S)zgValidates that 'value' is 0 or a positive float, or can be converted to 0 or a positive float. r0)rW)r?r@r3r3r4validate_positive_float_or_zero2srYcCs|dkr |St||dS)zcValidates a timeout specified in milliseconds returning a value in floating point seconds. Ng@@)rW)r?r@r3r3r4validate_timeout_or_none;srZcCs8|dkrtd|f|dks&|dkr*dSt||dS)zValidates a timeout specified in milliseconds returning a value in floating point seconds for the case where None is an error and 0 is valid. Setting the timeout to nothing in the URI string is a config error. Nz%s cannot be NonerrXg@@)rrW)r?r@r3r3r4validate_timeout_or_zeroDs r[cCs*|dks|dks|dkrdSt||dS)zValidates a timeout specified in milliseconds returning a value in floating point seconds. value=0 and value="0" are treated the same as value=None which means unlimited timeout. NrrXg@@)rW)r?r@r3r3r4 validate_timeout_or_none_or_zeroQsr\cCs|dks|dkrdSt||S)zBValidates maxStalenessSeconds according to the Max Staleness Spec.r!z-1r+r+)rH)r?r@r3r3r4validate_max_staleness[sr]cCst|tstd|f|S)z Validate a read preference. z%r is not a read preference.)r<rr>)r9r@r3r3r4validate_read_preferencecs r^cCs|tkrtd|f|S)zValidate read preference mode for a MongoReplicaSetClient. .. versionchanged:: 3.5 Returns the original ``value`` instead of the validated read preference mode. z!%s is not a valid read preference)rrD)r9r@r3r3r4validate_read_preference_modeksr_cCs(|tkr$|dkr$td|ttf|S)z+Validate the authMechanism URI option. zCRAM-MD5z%s must be in %s)r rDtuple)r?r@r3r3r4validate_auth_mechanismwsrac Cs6yt|Stk r0td|ttfYnXdS)zAValidate the uuid representation option selected in the URI. z7%s is an invalid UUID representation. Must be one of %sN)_UUID_REPRESENTATIONSKeyErrorrDr`)r9r@r3r3r4validate_uuid_representations rdc Cst|ts|g}g}x|D]}|dkr2|jiqyDi}x0|jdD]"}|jd\}}t||t|<qDW|j|Wqtk rtd||fYqXqW|S)z:Parse readPreferenceTags if passed as a client kwarg. ,r)z%r not a valid value for %s)r<listappendsplitr ExceptionrD)namer@Ztag_setsZtag_settagstagr8rGr3r3r4validate_read_preference_tagss"   rnZ SERVICE_NAMECANONICALIZE_HOST_NAMEZ SERVICE_REALMAWS_SESSION_TOKENcCst||}i}x|jdD]}y|jd\}}Wn.tk r^d|krLd}td|fYnX|tkr|td|ttf|dkrt||||<qt|||<qW|S)z!Validate authMechanismProperties.rfr)rpz_AWS_SESSION_TOKEN:, did you forget to percent-escape the token with quote_plus?zTauth mechanism properties must be key:value pairs like SERVICE_NAME:mongodb, not %s.zA%s is not a supported auth mechanism property. Must be one of %s.ro)rKrirD_MECHANISM_PROPSr`rEr)r?r@propsoptr8rGr3r3r4"validate_auth_mechanism_propertiess" rtcCs"t|tjtfstd|f|S)z#Validate the document_class option.zh%s must be dict, bson.son.SON, bson.raw_bson.RawBSONDocument, or a sublass of collections.MutableMapping) issubclassrMutableMappingr r>)r?r@r3r3r4validate_document_classs rwcCs(|dk r$t|t r$td|tf|S)z"Validate the type_registry option.Nz%s must be an instance of %s)r<rr>)r?r@r3r3r4validate_type_registrys rxcCst|tstd|f|S)z!Validates that 'value' is a list.z%s must be a list)r<rgr>)r?r@r3r3r4 validate_lists rycCs|dkr |St||S)z)Validates that 'value' is a list or None.N)ry)r?r@r3r3r4validate_list_or_nonesrzcCs"t|tjtfstd|fdS)z/Validates that 'value' is a list or a document.zw%s must either be a list or an instance of dict, bson.son.SON, or any other type that inherits from collections.MappingN)r<rMappingrgr>)r?r@r3r3r4validate_list_or_mappingsr|cCst|tjstd|fdS)z=Validate the type of method arguments that expect a document.zf%s must be an instance of dict, bson.son.SON, or any other type that inherits from collections.MappingN)r<rr{r>)r?r@r3r3r4validate_is_mappings r}cCs"t|tjtfstd|fdS)zEValidate the type of method arguments that expect a MongoDB document.z%s must be an instance of dict, bson.son.SON, bson.raw_bson.RawBSONDocument, or a type that inherits from collections.MutableMappingN)r<rrvr r>)r?r@r3r3r4validate_is_document_typesr~cCs:|dkr |St||t|jddkr6td|f|S)zValidate the appname option.Nzutf-8z%s must be <= 128 bytes)rKlenencoderD)r?r@r3r3r4validate_appname_or_nones  rcCs(|dkr |St|ts$td|f|S)z Validate the driver keyword arg.Nz$%s must be an instance of DriverInfo)r<rr>)r?r@r3r3r4validate_driver_or_nones  rcCs(|dkr |St|ts$td|f|S)z$Validate the server_api keyword arg.Nz#%s must be an instance of ServerApi)r<rr>)r?r@r3r3r4validate_server_api_or_nones  rcCs&|dkr |St|s"td|f|S)z%Validates that 'value' is a callable.Nz%s must be a callable)callablerD)r?r@r3r3r4validate_is_callable_or_nones rcCs<td||r8t|t r8tt|}|jdr8tddS)z Validate a replacement document. replacement$z'replacement can not include $ operatorsN)r}r<r nextiterr.rD)rfirstr3r3r4validate_ok_for_replace#s    rcCsJtd||stdt|t }tt|}|rF|jd rFtddS)zValidate an update document.updatezupdate cannot be emptyrz"update only works with $ operatorsN)r|rDr<rgrrr.)rZ is_documentrr3r3r4validate_ok_for_update-s   rstrictreplaceignorecCs |tkrtd|ttf|S)zFValidate the Unicode decode error handler option of CodecOptions. z@%s is an invalid Unicode decode error handler. Must be one of %s)_UNICODE_DECODE_ERROR_HANDLERSrDr`)r9r@r3r3r4%validate_unicode_decode_error_handler=srcCs&|dk r"t|tj r"td||S)zValidate the tzinfo option Nz)%s must be an instance of datetime.tzinfo)r<datetimetzinfor>)r9r@r3r3r4validate_tzinfoGs rjwtimeoutssl ssl_cert_reqsssl_match_hostname ssl_crlfile ssl_ca_certs ssl_certfilessl_pem_passphrase) journal wtimeoutmstlstlsallowinvalidcertificatestlsallowinvalidhostnames tlscrlfile tlscafiletlscertificatekeyfiletlscertificatekeyfilepasswordcGs t| S)N)rE)xr3r3r4~srcGs t| S)N)rE)rr3r3r4rs)$appnameZ authmechanismauthmechanismpropertiesZ authsourceZ compressorsconnecttimeoutmsZdirectconnectionheartbeatfrequencymsrZlocalthresholdms maxidletimemsZ maxpoolsizemaxstalenesssecondsZreadconcernlevelZreadpreferenceZreadpreferencetagsZ replicasetZ retryreadsZ retrywritesZ loadbalancedserverselectiontimeoutmssockettimeoutms ssl_keyfilerrrrrrrrtlsdisableocspendpointcheckZ tlsinsecurewrZzlibcompressionlevel) connectZdriverZ server_apifsyncZ minpoolsizeZsocketkeepaliverZtz_awareZunicode_decode_error_handlerZuuidrepresentationwaitqueuemultiplewaitqueuetimeoutms) Zdocument_classZ type_registryread_preferenceZevent_listenersrusernamepasswordZserver_selectorZauto_encryption_optsrrZssl_check_ocsp_endpoint) rrrrrrrrrrrenamed wTimeoutMStlsAllowInvalidCertificatestlsAllowInvalidHostnames tlsCRLFile tlsCAFileremovedInstead of using ssl_certfile to specify the certificate file, use tlsCertificateKeyFile to pass a single file containing both the client certificate and the private keyInstead of using ssl_keyfile to specify the private keyfile, use tlsCertificateKeyFile to pass a single file containing both the client certificate and the private keytlsCertificateKeyFilePasswordqInstead of using waitQueueMultiple to bound queuing, limit the size of the thread pool in your application server) rrrrrrrrrrrrrrrrrrcCs,t||\}}|tkr$td|f||fS)z1Validate optional authentication parameters. z!Unknown authentication option: %s)validate _AUTH_OPTIONSr)r?r@r6r3r3r4validate_auth_options  rcCs&|j}tj|t}|||}||fS)z!Generic validation function. )r6 VALIDATORSgetr:)r?r@r6 validatorr3r3r4rs  rc sttr&t}dd}fdd}ni}dd}dd}xtD]v\}}||}ytj|t}|||}Wn>tttfk r} z|rt j t | nWYdd} ~ XqDX||||<qDW|S)aValidate each entry in options and raise a warning if it is not valid. Returns a copy of options with invalid entries removed. :Parameters: - `opts`: A dict containing MongoDB URI options. - `warn` (optional): If ``True`` then warnings will be logged and invalid options will be ignored. Otherwise, invalid options will cause errors. cSs|S)Nr3)rr3r3r4rsz'get_validated_options..cs j|S)N) cased_key)r)optionsr3r4rscSs|jS)N)r6)rr3r3r4rscSs|S)Nr3)rr3r3r4rsN) r<_CaseInsensitiveDictionaryrURI_OPTIONS_VALIDATOR_MAPrr:rDr>rwarningswarnstr) rrZvalidated_optionsZget_normed_keyZget_setter_keyrsr@Z normed_keyrexcr3)rr4get_validated_optionss& rrrc@sXeZdZdZddZeddZeddZdd Zed d Z d d Z eddZ dS) BaseObjectzA base class that provides attributes and methods common to multiple pymongo classes. SHOULD NOT BE USED BY DEVELOPERS EXTERNAL TO MONGODB. cCsjt|tstd||_t|ts0td|f||_t|tsHtd||_t|ts`td||_ dS)NzDcodec_options must be an instance of bson.codec_options.CodecOptionszT%r is not valid for read_preference. See pymongo.read_preferences for valid options.zGwrite_concern must be an instance of pymongo.write_concern.WriteConcernzDread_concern must be an instance of pymongo.read_concern.ReadConcern) r<rr>_BaseObject__codec_optionsr_BaseObject__read_preferencer_BaseObject__write_concernr_BaseObject__read_concern)self codec_optionsr write_concern read_concernr3r3r4__init__=s     zBaseObject.__init__cCs|jS)zdRead only access to the :class:`~bson.codec_options.CodecOptions` of this instance. )r)rr3r3r4rUszBaseObject.codec_optionscCs|jS)zRead only access to the :class:`~pymongo.write_concern.WriteConcern` of this instance. .. versionchanged:: 3.0 The :attr:`write_concern` attribute is now read only. )r)rr3r3r4r\szBaseObject.write_concerncCs|r|jrtS|jS)zKRead only access to the write concern of this instance or session. )Zin_transactionrr)rsessionr3r3r4_write_concern_forfs zBaseObject._write_concern_forcCs|jS)zRead only access to the read preference of this instance. .. versionchanged:: 3.0 The :attr:`read_preference` attribute is now read only. )r)rr3r3r4rnszBaseObject.read_preferencecCs|r|jp|jS|jS)zMRead only access to the read preference of this instance or session. )Z_txn_read_preferencer)rrr3r3r4_read_preference_forwszBaseObject._read_preference_forcCs|jS)zRead only access to the :class:`~pymongo.read_concern.ReadConcern` of this instance. .. versionadded:: 3.2 )r)rr3r3r4rszBaseObject.read_concernN) rQ __module__ __qualname____doc__rpropertyrrrrrrr3r3r3r4r6s   rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ d"ddZ ddZ ddZddZd#ddZddZd d!ZdS)$rcOs i|_i|_|jt||dS)N)%_CaseInsensitiveDictionary__casedkeys _CaseInsensitiveDictionary__datardict)rargskwargsr3r3r4rsz#_CaseInsensitiveDictionary.__init__cCs|j|jkS)N)r6r)rr8r3r3r4 __contains__sz'_CaseInsensitiveDictionary.__contains__cCs t|jS)N)rr)rr3r3r4__len__sz"_CaseInsensitiveDictionary.__len__cCsdd|jDS)Ncss|] }|VqdS)Nr3).0r8r3r3r4 sz6_CaseInsensitiveDictionary.__iter__..)r)rr3r3r4__iter__sz#_CaseInsensitiveDictionary.__iter__cstfddDS)Ncsi|]}j|j|qSr3)rr)rk)rr3r4 sz7_CaseInsensitiveDictionary.__repr__..)r)rr3)rr4__repr__sz#_CaseInsensitiveDictionary.__repr__cCs |j}||j|<||j|<dS)N)r6rr)rr8r@lc_keyr3r3r4 __setitem__s z&_CaseInsensitiveDictionary.__setitem__cCs|j|jS)N)rr6)rr8r3r3r4 __getitem__sz&_CaseInsensitiveDictionary.__getitem__cCs|j}|j|=|j|=dS)N)r6rr)rr8rr3r3r4 __delitem__sz&_CaseInsensitiveDictionary.__delitem__cCsJt|tjstSt|t|kr$dSx |D]}||||kr*dSq*WdS)NFT)r<rr{NotImplementedr)rotherr8r3r3r4__eq__s  z!_CaseInsensitiveDictionary.__eq__NcCs|jj|j|S)N)rrr6)rr8defaultr3r3r4rsz_CaseInsensitiveDictionary.getcOs*|j}|jj|d|jj|f||S)N)r6rpopr)rr8rrrr3r3r4rsz_CaseInsensitiveDictionary.popcCs"|jj\}}|jj|}||fS)N)rpopitemrr)rrrr@r3r3r4rs z"_CaseInsensitiveDictionary.popitemcCs|jj|jjdS)N)rclearr)rr3r3r4rs z _CaseInsensitiveDictionary.clearcCs6|j}||kr|j|S||j|<||j|<|SdS)N)r6rr)rr8rrr3r3r4 setdefaults    z%_CaseInsensitiveDictionary.setdefaultcCsJt|tr,x:|D]}||||j|<qWnx|D]}||||<q2WdS)N)r<rr)rrr8r3r3r4rs    z!_CaseInsensitiveDictionary.updatecCs|j|jS)N)rr6)rr8r3r3r4rsz$_CaseInsensitiveDictionary.cased_key)N)N)rQrrrrrrrrrrrrrrrrrrr3r3r3r4rs   rii)r&)rr'r()rr)rr)rr)rr)rr)rr)rr)rr)rr)rr)T)rrrZbsonrZ bson.binaryrZbson.codec_optionsrrZbson.py3compatrrrr r Z bson.raw_bsonr Z pymongo.authr Zpymongo.compression_supportr rZpymongo.driver_inforZpymongo.server_apirZpymongo.encryption_optionsrZpymongo.errorsrZpymongo.monitoringrZpymongo.read_concernrZpymongo.read_preferencesrrZpymongo.ssl_supportrrZpymongo.write_concernrr collectionsrZ ORDERED_TYPES ImportError urllib.parserurllibZ MAX_BSON_SIZEZMAX_MESSAGE_SIZEZMIN_WIRE_VERSIONZMAX_WIRE_VERSIONZMAX_WRITE_BATCH_SIZEZMIN_SUPPORTED_SERVER_VERSIONZMIN_SUPPORTED_WIRE_VERSIONZMAX_SUPPORTED_WIRE_VERSIONZHEARTBEAT_FREQUENCYZKILL_CURSOR_FREQUENCYZEVENTS_QUEUE_FREQUENCYZSERVER_SELECTION_TIMEOUTZMIN_HEARTBEAT_INTERVALZMIN_SRV_RESCAN_INTERVALZCONNECT_TIMEOUTZ MAX_POOL_SIZEZ MIN_POOL_SIZEZMAX_IDLE_TIME_MSZMAX_IDLE_TIME_SECZWAIT_QUEUE_TIMEOUTZLOCAL_THRESHOLD_MSZ RETRY_WRITESZ RETRY_READSZCOMMAND_NOT_FOUND_CODESZUNAUTHORIZED_CODESZ_MAX_END_SESSIONSr5r7r:Z UNSPECIFIEDZSTANDARDZ PYTHON_LEGACYZ JAVA_LEGACYZ CSHARP_LEGACYrbrArErFrHrIrNrOrPrKrRrSrTrWrYrZr[r\r]r^r_rardrn frozensetrqrtrwrxryrzr|r}r~rrrrrrrrrZURI_OPTIONS_ALIAS_MAPrZNONSPEC_OPTIONS_VALIDATOR_MAPZ KW_VALIDATORSZINTERNAL_URI_OPTION_NAME_MAPZURI_OPTIONS_DEPRECATION_MAProptnamealiasesaliascopyrZTIMEOUT_OPTIONSrrrrZWRITE_CONCERN_OPTIONSobjectrrvrr3r3r3r4s                                                $S