o ]F @sdZddlZddlZddlmZmZerddlmZnddlmZddl m Z m Z m Z m Z ddlmZmZddlmZmZdZeeZd ZeeZd Zd d Zd dZefddZddhZeddeDBZddZddZ ddZ!ddZ"d-ddZ#d.d!d"Z$efd#d$Z%e&d%e'd&d'Z(e)gd(Z*ed dd dfd)d*Z+e,d+krddl-Z-ddl.Z.z e--e+e.j/d,WneyZ0z e1e0WYdZ0[0ndZ0[0wwe.2ddSdS)/z*Tools to parse and validate a MongoDB URI.N) string_typePY3) unquote_plus)get_validated_optionsINTERNAL_URI_OPTION_NAME_MAPURI_OPTIONS_DEPRECATION_MAP_CaseInsensitiveDictionary)ConfigurationError InvalidURI)_HAVE_DNSPYTHON _SrvResolverz mongodb://zmongodb+srv://iicCs\d|vs |ddkrtrd}nd}td||d\}}}|s&tdt|t|fS)aValidates the format of user information in a MongoDB URI. Reserved characters like ':', '/', '+' and '@' must be escaped following RFC 3986. Returns a 2-tuple containing the unescaped username followed by the unescaped password. :Paramaters: - `userinfo`: A string of the form : .. versionchanged:: 2.2 Now uses `urllib.unquote_plus` so `+` characters must be escaped. @:zurllib.parse.quote_pluszurllib.quote_pluszFUsername and password must be escaped according to RFC 3986, use %s().z'The empty string is not valid username.)countrr partitionr)userinfoquote_fnuser_passwdrD/var/www/html/env/lib/python3.10/site-packages/pymongo/uri_parser.pyparse_userinfo)srcCsT|ddkr td|d}|dkr|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]:rN)find ValueError)entity default_portirrrparse_ipv6_literal_hostEs   r"cCs|}|}|ddkrt||\}}n#|dr||fS|ddkr5|ddkr-td|dd\}}t|trU|rJt |dksJt |dkrQtd |ft |}| |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[.sockrrrzReserved 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: %s) r"endswithrrrsplit isinstancerisdigitintlower)rr hostportrrr parse_host[s"      r-tlsallowinvalidcertificatestlsallowinvalidhostnamescCsh|]}t|qSr)r).0krrr sr2cCslt}||D]+}|d\}}|dkr!||g|q||vr-td|ft|||<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.=readpreferencetagszDuplicate URI option '%s'.)rr&r* setdefaultappendwarningswarnr)optsdelimoptionsurioptkeyvaluerrr_parse_optionss r?cCs|d}|dur"tD]}||vr!d}t||d||fq d|vrLd|vrLdd}||d||dkrLd}t||d|df|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.ssltlscSs"|dvr|dkSt|tr|S|S)N)truefalserC)r'bool)valrrr truth_values  z-_handle_security_options..truth_valuez=Can not specify conflicting values for URI options %s and %s.)get_TLSINSECURE_EXCLUDE_OPTSr cased_key)r;r@opterr_msgrGrrr_handle_security_optionss" rMcCst|D]W}|tvr[t|\}}|dkrF|}||vr4d}tj|||||ftdd||qd}tj||||ftddq|dkr[d}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. renamedz0Deprecated option '%s' ignored in favor of '%s'.r) stacklevelz,Option '%s' is deprecated, use '%s' instead.removedzOption '%s' is deprecated. %s.)listrr7r8rJDeprecationWarningpop)r;optnamemodemessage newoptnamewarn_msgrrr_handle_option_deprecationss8    rYcCsf|d}|durtD] }t|d}| ||<q t|D]}t|d}|dur0||||<q|S)aNormalizes option names in the options dictionary by converting them to their internally-used names. Also handles use of the tlsInsecure option. :Parameters: - `options`: Instance of _CaseInsensitiveDictionary containing MongoDB URI options. r@N)rH_IMPLICIT_TLSINSECURE_OPTSrrQrS)r;r@rKintnamerTrrr_normalize_optionss     r\FcCs 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)r9r8rrrvalidate_optionss r]TcCs|d}|d}z1|dkr|dkrtd|dkr!t|d}n|dkr+t|d}n|ddkr8t|d}ntWn tyFtdwt|}t|}|rVt||}|r\t|}|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.r3rNz(MongoDB URI options are key=value pairs.)rr r?rrMrYr]r\)r9validater8 normalizeand_idxsemi_idxr;rrr split_optionss.      rdcCsFg}|dD]}|std|}|drd}|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&r r%r6r-)hostsr nodesrr,rrr split_hosts0s  rhr#z/ "$r) authsource authSource replicaset replicaSetcCs|trd}|td}n|tr"tstdd}|td}ntdttf|s0tdd}d} d} d} t} | d\} }}| sI|} d}|sSd | vrStd |r|d d krdt |d d}n"t t | d \} }}d | vr{| d d \} } t | rtd| |r| t||||| durt | } | durt | } d| vr| d\}}}t|\}} n| }d|vrtd| t |}d}|r:t|dd}t|d krtdtf|d \}}|durtdtf|p| d}t||d}|}|}|r+t||||}t|trtd|D]\}}|| vr)|| |<qd| vr9|r5dnd| d<nt||d}||| | | | |dS)aParse and validate a MongoDB URI. Returns a dict of the form:: { 'nodelist': , '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. .. 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. FNzCThe "dnspython" module must be installed to use mongodb+srv:// URIsTz4Invalid URI scheme: URI must begin with '%s' or '%s'z)Must provide at least one hostname or IP./?z8A '/' is required between the host list and any options.rr.zBad database name "%s"r z;Any '/' in a unix domain socket must be percent-encoded: %s)r z0%s URIs must include one, and only one, hostnamez&%s URIs must not include a port numberconnectTimeoutMS)connect_timeoutz5Only authSource and replicaSet are supported from DNSrArC)nodelistusernamepassworddatabase collectionr;fqdn) startswithSCHEME SCHEME_LEN SRV_SCHEMEr r SRV_SCHEME_LENr rrrmapr& _BAD_DB_CHARSsearchupdaterd rpartitionrrhlenrHr get_hosts get_optionsset_ALLOWED_TXT_OPTSitems)urir r`r8rarris_srv scheme_freerrdbaserwr; host_partr path_partr9rrfrxrgr, dns_resolver dns_optionsparsed_dns_optionsrKrFrrr parse_uriRs /            r__main__r)F)TFT)3__doc__rer7bson.py3compatrr urllib.parserurllibpymongo.commonrrrrpymongo.errorsr r pymongo.srv_resolverr r rzrr{r|r} DEFAULT_PORTrr"r-rZrIr?rMrYr\r]rdrhcompileescaper frozensetrr__name__pprintsysargvexcprintexitrrrrsh  &!$   *