U d\ @sdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZddlmZddlmZddlmZmZmZmZmZddlmZmZdd lm Z m!Z!dd l"m#Z#erdd l$m%Z%d Z&e'e&Z(d Z)e'e)Z*dZ+dddddZ,dddddZ-ddddddZ.e+fddddddZ/d d!d"hZ0dd#d$d%d&d'Z1d$d$d(d)d*Z2d$d$d(d+d,Z3d$d$d(d-d.Z4dWd0dd1d2d3d4Z5dXddddd1d6d7d8Z6e+fddd9d:d;d<Z7e8d=e9d>d?Z:e;d@dAdBdCdDdEgZdPdQdRdSdTZ?e@dUkrddlAZAzeAAe>ejBdVWn,ek rZCz eDeCW5dZC[CXYnXeEddS)Yz*Tools to parse and validate a MongoDB URI.) annotationsN) TYPE_CHECKINGAnyDictListMappingMutableMappingOptionalSizedTupleUnioncast) unquote_plus)_parse_ssl_options)INTERNAL_URI_OPTION_NAME_MAPSRV_SERVICE_NAMEURI_OPTIONS_DEPRECATION_MAP_CaseInsensitiveDictionaryget_validated_options)ConfigurationError InvalidURI)_HAVE_DNSPYTHON _SrvResolver)_Address) SSLContextz mongodb://zmongodb+srv://iistrbool)sreturncCsDtt|D]2}||dkr |||d}t||kr dSq dS)zCheck for unescaped percent signs. :Parameters: - `s`: A string. `s` can have things like '%25', '%2525', and '%E2%85%A8' but cannot have unquoted percent like '%foo'. %TF)rangelenr)risubr%6/tmp/pip-unpacked-wheel-oblwsawz/pymongo/uri_parser.py_unquoted_percent;s   r'zTuple[str, str])userinforcCsRd|ks|ddkst|r&td|d\}}}|sBtdt|t|fS)alValidates the format of user information in a MongoDB URI. Reserved characters that are gen-delimiters (":", "/", "?", "#", "[", "]", "@") as per RFC 3986 must be escaped. Returns a 2-tuple containing the unescaped username followed by the unescaped password. :Parameters: - `userinfo`: A string of the form : @:zXUsername and password must be escaped according to RFC 3986, use urllib.parse.quote_plusz'The empty string is not valid username.)countr'r partitionr)r(user_passwdr%r%r&parse_userinfoLs r1z Optional[int]z%Tuple[str, Optional[Union[str, int]]])entity default_portrcCsT|ddkrtd|d}|dkr8|dd|fS|d|||ddfS)aValidates an IPv6 literal host:port string. Returns a 2-tuple of IPv6 literal followed by port where port is default_port if it wasn't specified in entity. :Parameters: - `entity`: A string that represents an IPv6 literal enclosed in braces (e.g. '[::1]' or '[::1]:27017'). - `default_port`: The port number to use when one wasn't specified in entity. ]zNan IPv6 address literal must be enclosed in '[' and ']' according to RFC 2732.z]:r+N)find ValueError)r2r3r#r%r%r&parse_ipv6_literal_hostes r9rcCs|}|}|ddkr$t||\}}nF|dr6||fS|ddkrj|ddkrZtd|dd\}}t|tr|rt |dkst |dkrtd |t |}| |fS) aValidates a host string Returns a 2-tuple of host followed by port where port is default_port if it wasn't specified in the string. :Parameters: - `entity`: A host or host:port string where host could be a hostname or IP address. - `default_port`: The port number to use when one wasn't specified in entity. r[.sockr*r5r+zReserved characters such as ':' must be escaped according RFC 2396. An IPv6 address literal must be enclosed in '[' and ']' according to RFC 2732.iz-Port must be an integer between 0 and 65535: ) r9endswithr7r,r8split isinstancerisdigitintlower)r2r3hostportr%r%r& parse_host}s"     rDtlsallowinvalidcertificatesZtlsallowinvalidhostnamestlsdisableocspendpointcheckz Optional[str]r)optsdelimrcCst}||D]n}|d\}}|dkrB||g|q||kr\td|d|dkrn|}nt|}|||<q|S)zHelper method for split_options which creates the options dict. Also handles the creation of a list for the URI tag_sets/ readpreferencetags portion, and the use of a unicode options string. =ZreadpreferencetagszDuplicate URI option 'z'.Zauthmechanismproperties)rr=rA setdefaultappendwarningswarnr)rGrHoptionsZurioptkeyvaluevalr%r%r&_parse_optionss   rR)rNrcCs*|d}|dk rDtD],}||krd}t||d||fq|d}|dk rd|krxd}t|d|df|dkrd|d<|d}|dk rdD]$}||dkrd }t||fqd |kr&d |kr&d d d dd}||d ||d kr&d}t||d |d f|S)zRaise appropriate errors when conflicting TLS options are present in the options dictionary. :Parameters: - `options`: Instance of _CaseInsensitiveDictionary containing MongoDB URI options. tlsinsecureNz9URI options %s and %s cannot be specified simultaneously.rErFT tlscrlfile)rSrErFzDURI option %s=True cannot be specified when CRL checking is enabled.ssltlsr)rQrcSs"|dkr|dkSt|tr|S|S)N)truefalserW)r>r)rQr%r%r& truth_values  z-_handle_security_options..truth_valuez=Can not specify conflicting values for URI options %s and %s.)get_IMPLICIT_TLSINSECURE_OPTSr cased_key)rNrSopterr_msgZtlsallowinvalidcertsrTrYr%r%r&_handle_security_optionss>    r_cCst|D]}|tkrt|\}}|dkr|}||krhd}tj|||||ftdd||qd}tj||||ftddq|dkrd}tj||||ftddq|S)aMIssue appropriate warnings when deprecated options are present in the options dictionary. Removes deprecated option key, value pairs if the options dictionary is found to also have the renamed option. :Parameters: - `options`: Instance of _CaseInsensitiveDictionary containing MongoDB URI options. Zrenamedz0Deprecated option '%s' ignored in favor of '%s'.r6) stacklevelz,Option '%s' is deprecated, use '%s' instead.removedzOption '%s' is deprecated. %s.)listrrLrMr\DeprecationWarningpop)rNoptnamemodemessageZ newoptnameZwarn_msgr%r%r&_handle_option_deprecationss8   rhcCsX|d}|dk r$tD] }|||<qt|D]&}t|d}|dk r,||||<q,|S)zNormalizes option names in the options dictionary by converting them to their internally-used names. :Parameters: - `options`: Instance of _CaseInsensitiveDictionary containing MongoDB URI options. rSN)rZr[rbrrd)rNrSr]reZintnamer%r%r&_normalize_options s    riFzMapping[str, Any]zMutableMapping[str, Any])rGrMrcCs t||S)aValidates and normalizes options passed in a MongoDB URI. Returns a new dictionary of validated and normalized options. If warn is False then errors will be thrown for invalid options, otherwise they will be ignored and a warning will be issued. :Parameters: - `opts`: A dict of MongoDB URI options. - `warn` (optional): If ``True`` then warnings will be logged and invalid options will be ignored. Otherwise invalid options will cause errors. )r)rGrMr%r%r&validate_options7s rjT)rGvalidaterM normalizercCs|d}|d}zd|dkr0|dkr0tdnF|dkrDt|d}n2|dkrXt|d}n|ddkrrt|d}ntWntk rtdYnXt|}t|}|rt|}|rttt ||}| d d krtd |S) aTakes the options portion of a MongoDB URI, validates each option and returns the options in a dictionary. :Parameters: - `opt`: A string representing MongoDB URI options. - `validate`: If ``True`` (the default), validate and normalize all options. - `warn`: If ``False`` (the default), suppress all warnings raised during validation of options. - `normalize`: If ``True`` (the default), renames all options to their internally-used names. &;rz.Can not mix '&' and ';' for option separators.rIr5Nz(MongoDB URI options are key=value pairs. authsourcez1the authSource database cannot be an empty string) r7rrRr8r_rhrir rrjrZ)rGrkrMrlZand_idxZsemi_idxrNr%r%r& split_optionsGs.      rqzList[_Address])hostsr3rcCsFg}|dD]2}|std|}|dr0d}|t||q|S)aTakes a string of the form host1[:port],host2[:port]... and splits it into (host, port) tuples. If [:port] isn't present the default_port is used. Returns a set of 2-tuples containing the host name (or IP) followed by port number. :Parameters: - `hosts`: A string of the form host1[:port],host2[:port],... - `default_port`: The port number to use when one wasn't specified for a host. ,z)Empty host (or extra comma in host list).r;N)r=rr<rKrD)rrr3nodesr2rCr%r%r& split_hostsus  rur:z/ "$r4roZ authSource replicaset replicaSet loadbalanced loadBalancedr None)rtrNrcCsdt|dkr|drtd|dr`t|dkr, 'username': or None, 'password': or None, 'database': or None, 'collection': or None, 'options': , 'fqdn': or None } If the URI scheme is "mongodb+srv://" DNS SRV and TXT lookups will be done to build nodelist and options. :Parameters: - `uri`: The MongoDB URI to parse. - `default_port`: The port number to use when one wasn't specified for a host in the URI. - `validate` (optional): If ``True`` (the default), validate and normalize all options. Default: ``True``. - `warn` (optional): When validating, if ``True`` then will warn the user then ignore any invalid options or values. If ``False``, validation will error when options are unsupported or values are invalid. Default: ``False``. - `normalize` (optional): If ``True``, convert names of URI options to their internally-used names. Default: ``True``. - `connect_timeout` (optional): The maximum time in milliseconds to wait for a response from the DNS server. - 'srv_service_name` (optional): A custom SRV service name .. versionchanged:: 4.0 To better follow RFC 3986, unquoted percent signs ("%") are no longer supported. .. versionchanged:: 3.9 Added the ``normalize`` parameter. .. versionchanged:: 3.6 Added support for mongodb+srv:// URIs. .. versionchanged:: 3.5 Return the original value of the ``readPreference`` MongoDB URI option instead of the validated read preference mode. .. versionchanged:: 3.1 ``warn`` added so invalid options can be ignored. FNpythonzThe "dnspython" module must be installed to use mongodb+srv:// URIs. To fix this error install pymongo again: %s -m pip install pymongo>=4.3Tz)Invalid URI scheme: URI must begin with 'z' or ''z)Must provide at least one hostname or IP./rp?z8A '/' is required between the host list and any options..r+zBad database name "%s"ZsrvServiceNamer)z;Any '/' in a unix domain socket must be percent-encoded: %sZ srvMaxHostsZdirectConnectionz*Cannot specify directConnection=true with z URIs)r3z. URIs must include one, and only one, hostnamerz$ URIs must not include a port numberZconnectTimeoutMSzDOnly authSource, replicaSet, and loadBalanced are supported from DNSryz0You cannot specify loadBalanced with srvMaxHostsrwz.You cannot specify replicaSet with srvMaxHostsrVrUrWzDThe srvServiceName option is only allowed with 'mongodb+srv://' URIszAThe srvMaxHosts option is only allowed with 'mongodb+srv://' URIs)ZnodelistusernamepasswordZdatabase collectionrNfqdn) startswithSCHEME SCHEME_LEN SRV_SCHEMErsys executablerSRV_SCHEME_LENrrr-rr= _BAD_DB_CHARSsearchupdaterqrZr rpartitionr1rur"rZ get_hostsZ get_optionsset_ALLOWED_TXT_OPTSitemsr{)r|r3rkrMrlr}r~rZis_srvZ scheme_freeZ python_pathr.r0ZdbaserrNZ host_partr/Z path_partrGr(rrrrtrCZ dns_resolverZ dns_optionsZparsed_dns_optionsr]rQr%r%r& parse_uris<                   rzOptional[Mapping[str, Any]]zDict[str, SSLContext])kms_tls_optionsrcCs|siSt|tstdi}|D]\}}t|tsHtd|d|ddt|}t|}t|}t|}t |\}}|dkrt d|rt dd D]"}||krt d ||||<qq&|S) z!Parse KMS TLS connection options.zkms_tls_options must be a dictzkms_tls_options["z"] must be a dictrVTNz!TLS is required for KMS providerszInsecure TLS options prohibited)Z tlsInsecureZtlsAllowInvalidCertificatesZtlsAllowInvalidHostnamesZ$tlsDisableCertificateRevocationCheckz!Insecure TLS options prohibited: ) r>dict TypeErrorrrJrr_rirjrr)rZcontextsZproviderrG ssl_contextZallow_invalid_hostnamesnr%r%r&_parse_kms_tls_optionsUs.     r__main__r+)F)TFT)F__doc__ __future__rrerrLtypingrrrrrrr r r r r urllib.parserZpymongo.client_optionsrZpymongo.commonrrrrrZpymongo.errorsrrZpymongo.srv_resolverrrZpymongo.typingsrZpymongo.pyopenssl_contextrrr"rrrZ DEFAULT_PORTr'r1r9rDr[rRr_rhrirjrqrucompileescaper frozensetrr{rr__name__pprintargvexcprintexitr%r%r%r&st 4    )6(. 1!