3 ]]F@sdZddlZddlZddlmZmZer6ddlmZn ddlmZddl m Z m Z m Z m Z ddlmZmZddlmZmZdZeeZd ZeeZd Zd d Zd dZefddZddhZeddeDBZddZddZ ddZ!ddZ"d0ddZ#d1d!d"Z$efd#d$Z%ej&d%ej'd&d'Z(e)d(d)d*d+gZ*ed dd dfd,d-Z+e,d.krddl-Z-ddl.Z.ye-j-e+e.j/d/Wn,ek rZ0ze1e0WYddZ0[0XnXe.j2ddS)2z*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|ks|jddkr0tr d}nd}td||jd\}}}|sLtdt|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)userinfoZquote_fnuser_passwdr5/tmp/pip-build-20mum3z4/pymongo/pymongo/uri_parser.pyparse_userinfo)srcCsT|jddkrtd|jd}|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. ]rzNan IPv6 address literal must be enclosed in '[' and ']' according to RFC 2732.z]:Nrr)find ValueError)entity default_portirrrparse_ipv6_literal_hostEs  r!cCs|}|}|ddkr$t||\}}nF|jdr6||fS|jdd krj|jddkrZtd|jdd\}}t|tr|j st |dkst |dkrtd|ft |}|j |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[z.sockrrzReserved 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: %sr) r!endswithrrrsplit isinstancerisdigitintlower)rrhostportrrr parse_host[s    " r+ZtlsallowinvalidcertificatesZtlsallowinvalidhostnamescCsh|] }t|qSr)r).0krrr sr.cCspt}xd|j|D]V}|jd\}}|jdkrD|j|gj|q||kr\tjd|ft|||<qW|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 '%s'.)rr$r( setdefaultappendwarningswarnr)optsdelimoptionsZurioptkeyvaluerrr_parse_optionss r9cCs|jd}|dk rHx4tD],}||krd}t||jd|j|fqWd|krd|krdd}||jd||jdkrd}t||jd|jdf|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.sslZtlscSs"|dkr|dkSt|tr|S|S)Ntruefalse)r<r=)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)r6r:opterr_msgr@rrr_handle_security_optionss  rFcCsxt|D]}|tkr t|\}}|dkr|}||krjd}tj||j||j|ftdd|j|q d}tj||j||ftddq |dkr d}tj||j||ftddq W|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'.r) stacklevelz,Option '%s' is deprecated, use '%s' instead.removedzOption '%s' is deprecated. %s.)listrr2r3rCDeprecationWarningpop)r6optnamemodemessageZ newoptnameZwarn_msgrrr_handle_option_deprecationss.       rOcCsn|jd}|dk r6x"tD]}tj|d}| ||<qWx2t|D]&}tj|d}|dk r@|j|||<q@W|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)rA_IMPLICIT_TLSINSECURE_OPTSrrIrK)r6r:rDZintnamerLrrr_normalize_optionss    rQFcCs 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)r4r3rrrvalidate_optionss rRTc Cs|jd}|jd}yd|dkr0|dkr0tdnF|dkrDt|d}n2|dkrXt|d}n|jdd krrt|d}ntWntk rtdYnXt|}t|}|rt||}|rt|}|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.r/rNz(MongoDB URI options are key=value pairs.r)rr r9rrFrOrRrQ)r4validater3 normalizeZand_idxZsemi_idxr6rrr split_optionss*       rWcCsJg}x@|jdD]2}|s td|}|jdr2d}|jt||qW|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).z.sockN)r$r r#r1r+)hostsrnodesrr*rrr split_hosts0s  r[r"z/ "$rZ authsourceZ authSourceZ replicasetZ replicaSetcCs|jtrd}|td}n8|jtrDts2tdd}|td}ntdttf|s`tdd}d} d} d} t} |j d\} }}| s|} d}| rd | krtd |r.|d d krt |d d}nHt t |j d \} }}d | kr| j d d \} } t j| rtd| |r.| jt||||| dk r@t | } | dk rRt | } d| krz| jd\}}}t|\}} n| }d|krtd| t |}d}|rt|dd}t|d krtdtf|d \}}|dk rtdtf|p| jd}t||d}|j}|j}|rrt||||}t|trHtdx(|jD]\}}|| krR|| |<qRWd| kr|rdnd| d<n t||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)rz0%s URIs must include one, and only one, hostnamez&%s URIs must not include a port numberZconnectTimeoutMS)connect_timeoutz5Only authSource and replicaSet are supported from DNSr;r<)ZnodelistusernamepasswordZdatabase collectionr6fqdn) startswithSCHEME SCHEME_LEN SRV_SCHEMEr r SRV_SCHEME_LENr rrrmapr$ _BAD_DB_CHARSsearchupdaterW rpartitionrr[lenrAr Z get_hostsZ get_optionsset_ALLOWED_TXT_OPTSitems)urirrUr3rVr`Zis_srvZ scheme_freerrZdbasercr6Z host_partrZ path_partr4rrYrdrZr*Z dns_resolverZ dns_optionsZparsed_dns_optionsrDr?rrr parse_uriRs/                     rt__main__r)F)TFT)3__doc__rer2Zbson.py3compatrr urllib.parserurllibZpymongo.commonrrrrZpymongo.errorsr r Zpymongo.srv_resolverr r rfrorgrhriZ DEFAULT_PORTrr!r+rPrBr9rFrOrQrRrWr[compileescaperk frozensetrqrt__name__pprintsysargvexcprintexitrrrrsV  & !$  *