3 ]]|*@sdZddlZddlZddlmZddlmZmZmZm Z ddl m Z m Z ddl mZmZmZmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lm 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.YnXdZ0de0Z1dZ2dZ3dZ4dZ5dZ6dZ7dZ8dZ9dZ:dZ;dZdZ?dZ@dZAdZBdZCd ZDd!ZEd!ZFdZGdZHd&ZId'd(ZJd)d*ZKd+d,ZLeeee d-ZMd.d/ZNd0d1ZOd2d3ZPd4d5ZQd6d7ZRd8d9ZSd:d;ZTdd?ZVd@dAZWdBdCZXdDdEZYdFdGZZdHdIZ[dJdKZ\dLdMZ]dNdOZ^dPdQZ_dRdSZ`dTdUZadVdWZbdXdYZceddZd[d\gZed]d^Zfd_d`ZgdadbZhdcddZidedfZjdgdhZkdidjZldkdlZmdmdnZndodpZodqdrZpdsdtZqdudvZreddwdxdygZsdzd{Ztd|d}Zud~gdgdgdgdgdgdgdgdgd ZveneaefeVee\e\eOe[e\eTe^eWe`eceWeOeOe]e\eSeOe(e'ddeOeSeSeWeOeYeRed!ZweOeoeOeReOeSeOetebeUe\d Zxegehe_e eueWeWeped Zydddddddddd ZzdddddddddZ{ewj|exxValidate that 'value' is a positive integer or 0 or None. N)rJ)r@rAr4r4r5%validate_non_negative_integer_or_nonesrQcCs$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@rAr4r4r5rLs rLcCs|dkr |St||S)zEValidates that 'value' is an instance of `basestring` or `None`. N)rL)r@rAr4r4r5validate_string_or_nonesrSc 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=r r r.rEr?)r@rAr4r4r5validate_int_or_basestrings  rTc 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=r r r.rErJr?)r@rArHr4r4r5'validate_non_negative_int_or_basestring s    rUc 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)floatrEr?)r@rAerrmsgr4r4r5validate_positive_floats    rXcCs|dks|dkrdSt||S)zgValidates that 'value' is 0 or a positive float, or can be converted to 0 or a positive float. r0)rX)r@rAr4r4r5validate_positive_float_or_zero-srZcCs|dkr |St||dS)zcValidates a timeout specified in milliseconds returning a value in floating point seconds. Ng@@)rX)r@rAr4r4r5validate_timeout_or_none6sr[cCs8|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 NonerrYg@@)rrX)r@rAr4r4r5validate_timeout_or_zero?s r\cCs|dks|dkrdSt||S)zBValidates maxStalenessSeconds according to the Max Staleness Spec.r!z-1r,r,)rI)r@rAr4r4r5validate_max_stalenessLsr]cCst|tstd|f|S)z Validate a read preference. z%r is not a read preference.)r=rr?)r:rAr4r4r5validate_read_preferenceTs 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)rrE)r:rAr4r4r5validate_read_preference_mode\sr_cCs(|tkr$|dkr$td|ttf|S)z+Validate the authMechanism URI option. zCRAM-MD5z%s must be in %s)rrEtuple)r@rAr4r4r5validate_auth_mechanismhsrac 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_REPRESENTATIONSKeyErrorrEr`)r:rAr4r4r5validate_uuid_representationss rdc Cst|ts|g}g}xh|D]`}|dkr2|jiqy"|jtdd|jdDWqtk rxtd||fYqXqW|S)z:Parse readPreferenceTags if passed as a client kwarg. cSsg|]}|jdqS)r*)split).0tagr4r4r5 sz1validate_read_preference_tags..,z%r not a valid value for %s)r=listappenddictrf ExceptionrE)namerAZtag_setsZtag_setr4r4r5validate_read_preference_tags~s    rpZ SERVICE_NAMECANONICALIZE_HOST_NAMEZ SERVICE_REALMcCst||}i}x|jdD]v}y|jd\}}Wn"tk rRtd|fYnX|tkrptd|ttf|dkrt||||<q|||<qW|S)z!Validate authMechanismProperties.rjr*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.rq)rLrfrE_MECHANISM_PROPSr`rF)r@rApropsoptr9rHr4r4r5"validate_auth_mechanism_propertiess  rucCs"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) issubclassr MutableMappingr r?)r@rAr4r4r5validate_document_classs rxcCs(|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@rAr4r4r5validate_type_registrys rycCst|tstd|f|S)z!Validates that 'value' is a list.z%s must be a list)r=rkr?)r@rAr4r4r5 validate_lists rzcCs|dkr |St||S)z)Validates that 'value' is a list or None.N)rz)r@rAr4r4r5validate_list_or_nonesr{cCs"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=r Mappingrkr?)r@rAr4r4r5validate_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=r r|r?)r@rAr4r4r5validate_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=r rwr r?)r@rAr4r4r5validate_is_document_typesrcCs:|dkr |St||t|jddkr6td|f|S)zValidate the appname option.Nzutf-8z%s must be <= 128 bytes)rLlenencoderE)r@rAr4r4r5validate_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@rAr4r4r5validate_driver_or_nones  rcCs&|dkr |St|s"td|f|S)z%Validates that 'value' is a callable.Nz%s must be a callable)callablerE)r@rAr4r4r5validate_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/rE)rfirstr4r4r5validate_ok_for_replaces    rcCsJtd||stdt|t }tt|}|rF|jd rFtddS)zValidate an update document.updatezupdate cannot be emptyrz"update only works with $ operatorsN)r}rEr=rkrrr/)rZ is_documentrr4r4r5validate_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_HANDLERSrEr`)r:rAr4r4r5%validate_unicode_decode_error_handlersrcCs&|dk r"t|tj r"td||S)zValidate the tzinfo option Nz)%s must be an instance of datetime.tzinfo)r=datetimetzinfor?)r:rAr4r4r5validate_tzinfo's rjwtimeoutssl ssl_cert_reqsssl_match_hostname ssl_crlfile ssl_ca_certsZ ssl_certfilessl_pem_passphrase) journal wtimeoutmstlstlsallowinvalidcertificatestlsallowinvalidhostnames tlscrlfile tlscafiletlscertificatekeyfiletlscertificatekeyfilepasswordcGs t| S)N)rF)xr4r4r5[sr)!appnameZ authmechanismauthmechanismpropertiesZ authsourceZ compressorsconnecttimeoutmsheartbeatfrequencymsrZlocalthresholdms maxidletimemsZ maxpoolsizemaxstalenesssecondsZreadconcernlevelZreadpreferenceZreadpreferencetagsZ replicasetZ retryreadsZ retrywritesserverselectiontimeoutmssockettimeoutmsZ ssl_keyfilerrrrrrrrZ tlsinsecurewrZzlibcompressionlevel) connectZdriverfsyncZ minpoolsizeZsocketkeepaliverZtz_awareZunicode_decode_error_handlerZuuidrepresentationwaitqueuemultiplewaitqueuetimeoutms) Zdocument_classZ type_registryread_preferenceZevent_listenersrusernamepasswordZserver_selectorZauto_encryption_optsrr) rrrrrrrrrrenamed wTimeoutMStlsAllowInvalidCertificatestlsAllowInvalidHostnames tlsCRLFile tlsCAFiletlsCertificateKeyFilePasswordremovedqInstead of using waitQueueMultiple to bound queuing, limit the size of the thread pool in your application server)rrrrrrrrrrrrrrrrcCs,t||\}}|tkr$td|f||fS)z1Validate optional authentication parameters. z!Unknown authentication option: %s)validate _AUTH_OPTIONSr)r@rAr7r4r4r5validate_auth_options  rcCs&|j}tj|t}|||}||fS)z!Generic validation function. )r7 VALIDATORSgetr;)r@rAr7 validatorr4r4r5rs  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)Nr4)rr4r4r5rsz'get_validated_options..cs j|S)N) cased_key)r)optionsr4r5rscSs|jS)N)r7)rr4r4r5rscSs|S)Nr4)rr4r4r5rsN) r=_CaseInsensitiveDictionaryr URI_OPTIONS_VALIDATOR_MAPrr;rEr?rwarningswarnstr) rrZvalidated_optionsZget_normed_keyZget_setter_keyrtrAZ normed_keyrexcr4)rr5get_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_concernr4r4r5__init__s     zBaseObject.__init__cCs|jS)zdRead only access to the :class:`~bson.codec_options.CodecOptions` of this instance. )r)rr4r4r5r&szBaseObject.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)rr4r4r5r-szBaseObject.write_concerncCs|r|jrtS|jS)zKRead only access to the write concern of this instance or session. )Zin_transactionrr)rsessionr4r4r5_write_concern_for7s 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)rr4r4r5r?szBaseObject.read_preferencecCs|r|jp|jS|jS)zMRead only access to the read preference of this instance or session. )Z_txn_read_preferencer)rrr4r4r5_read_preference_forHszBaseObject._read_preference_forcCs|jS)zRead only access to the :class:`~pymongo.read_concern.ReadConcern` of this instance. .. versionadded:: 3.2 )r)rr4r4r5rPszBaseObject.read_concernN) rR __module__ __qualname____doc__rpropertyrrrrrrr4r4r4r5rs   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__datarrm)rargskwargsr4r4r5r[sz#_CaseInsensitiveDictionary.__init__cCs|j|jkS)N)r7r)rr9r4r4r5 __contains__`sz'_CaseInsensitiveDictionary.__contains__cCs t|jS)N)rr)rr4r4r5__len__csz"_CaseInsensitiveDictionary.__len__cCsdd|jDS)Ncss|] }|VqdS)Nr4)rgr9r4r4r5 gsz6_CaseInsensitiveDictionary.__iter__..)r)rr4r4r5__iter__fsz#_CaseInsensitiveDictionary.__iter__cstfddDS)Ncsi|]}j|j|qSr4)rr)rgk)rr4r5 jsz7_CaseInsensitiveDictionary.__repr__..)r)rr4)rr5__repr__isz#_CaseInsensitiveDictionary.__repr__cCs |j}||j|<||j|<dS)N)r7rr)rr9rAlc_keyr4r4r5 __setitem__ls z&_CaseInsensitiveDictionary.__setitem__cCs|j|jS)N)rr7)rr9r4r4r5 __getitem__qsz&_CaseInsensitiveDictionary.__getitem__cCs|j}|j|=|j|=dS)N)r7rr)rr9rr4r4r5 __delitem__tsz&_CaseInsensitiveDictionary.__delitem__cCsJt|tjstSt|t|kr$dSx |D]}||||kr*dSq*WdS)NFT)r=r r|NotImplementedr)rotherr9r4r4r5__eq__ys  z!_CaseInsensitiveDictionary.__eq__NcCs|jj|j|S)N)rrr7)rr9defaultr4r4r5rsz_CaseInsensitiveDictionary.getcOs*|j}|jj|d|jj|f||S)N)r7rpopr)rr9rrrr4r4r5rsz_CaseInsensitiveDictionary.popcCs"|jj\}}|jj|}||fS)N)rpopitemrr)rrrrAr4r4r5rs z"_CaseInsensitiveDictionary.popitemcCs|jj|jjdS)N)rclearr)rr4r4r5rs z _CaseInsensitiveDictionary.clearcCs6|j}||kr|j|S||j|<||j|<|SdS)N)r7rr)rr9rrr4r4r5 setdefaults    z%_CaseInsensitiveDictionary.setdefaultcCsJt|tr,x:|D]}||||j|<qWnx|D]}||||<q2WdS)N)r=rr)rrr9r4r4r5rs    z!_CaseInsensitiveDictionary.updatecCs|j|jS)N)rr7)rr9r4r4r5rsz$_CaseInsensitiveDictionary.cased_key)N)N)rRrrrrrrrrrrrrrrrrrrr4r4r4r5rZs   rii)r&)r'r(r))rr)rr)rr)rr)rr)rr)rr)rr)T)rrrZbsonrZ bson.binaryrrrrZbson.codec_optionsrrZbson.py3compatr r r r Z bson.raw_bsonr Z pymongo.authrZpymongo.compression_supportrrZpymongo.driver_inforZpymongo.encryption_optionsrZpymongo.errorsrZpymongo.monitoringrZpymongo.read_concernrZpymongo.read_preferencesrrZpymongo.ssl_supportrrZpymongo.write_concernrr collectionsrZ ORDERED_TYPES ImportErrorZ 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_SESSIONSr6r8r;rbrBrFrGrIrJrOrPrQrLrSrTrUrXrZr[r\r]r^r_rardrp frozensetrrrurxryrzr{r}r~rrrrrrrrrZURI_OPTIONS_ALIAS_MAPrZNONSPEC_OPTIONS_VALIDATOR_MAPZ KW_VALIDATORSZINTERNAL_URI_OPTION_NAME_MAPZURI_OPTIONS_DEPRECATION_MAProptnamealiasesaliascopyrZTIMEOUT_OPTIONSrrrrZWRITE_CONCERN_OPTIONSobjectrrwrr4r4r4r5s                                          $S