o sDiV@sdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZddlmZddlmZddlmZddlmZmZmZmZdd lmZmZdd lmZer\dd lm Z d Z!e"e!Z#d Z$e"e$Z%dZ&e'gdZ(dVddZ)dWddZ*dXddZ+e&fdYd!d"Z,hd#Z-dZd(d)Z.d[d+d,Z/d[d-d.Z0d[d/d0Z1d\d]d5d6Z2 7d^d_d:d;Z3e&fd`d>d?Z4e5d@e6dAdBZ7e'gdCZ8dadGdHZ9dbdMdNZ:e&d7d1d7dfdcdRdSZ;dddTdUZ: @:zXUsername and password must be escaped according to RFC 3986, use urllib.parse.quote_plusz&The empty string is not valid username)countrPr partitionr )rQuser_passwdrNrNrOparse_userinfo}s r[entity default_port Optional[int]%tuple[str, Optional[Union[str, int]]]cCsT|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. :param entity: A string that represents an IPv6 literal enclosed in braces (e.g. '[::1]' or '[::1]:27017'). :param 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]:rUN)find ValueError)r\r]rLrNrNrOparse_ipv6_literal_hosts  rercCs|}|}|ddkrt||\}}n#|dr||fS|ddkr5|ddkr-td|dd\}}t|tro|s[t dd |DrW|D] }| rVtd |qItd t |d ksgt |dkrktd t |}| |fS)aqValidates 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. :param entity: A host or host:port string where host could be a hostname or IP address. :param default_port: The port number to use when one wasn't specified in entity. r[.sockrTrarUzReserved characters such as ':' must be escaped according RFC 2396. An IPv6 address literal must be enclosed in '[' and ']' according to RFC 2732.css |] }|p |VqdSN)isspaceisdigit).0crNrNrO szparse_host..z$Port contains whitespace character: zPort contains non-digit characters. Hint: username and password must be escaped according to RFC 3986, use urllib.parse.quote_plusiz+Port must be an integer between 0 and 65535) reendswithrcrVrdsplit isinstancerErjallriintlower)r\r]hostportrlrNrNrO parse_hosts4     rv>tlsallowinvalidhostnamestlsallowinvalidcertificatestlsdisableocspendpointcheckoptsdelim Optional[str]rcCst}||D]9}|d\}}|dkr!||g|q||vr0tjd|ddd|dkr9|}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. =readpreferencetagszDuplicate URI option 'z'.rb stacklevelauthmechanismproperties)rrors setdefaultappendwarningswarnr )rzr{optionsurioptkeyvaluevalrNrNrO_parse_optionss   rrcCs|d}|dur"tD]}||vr!d}t||d||fq |d}|durDd|vr.truth_valuez=Can not specify conflicting values for URI options %s and %s.)rrrFr)get_IMPLICIT_TLSINSECURE_OPTSr cased_key)rropterr_msgtlsallowinvalidcertsrrrNrNrO_handle_security_optionssB     rcCst|D]W}|tvr[t|\}}|dkrF|}||vr4d}tj|||||ftdd||qd}tj||||ftddq|dkr[d}tj||||ftddq|S)a;Issue 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. :param options: Instance of _CaseInsensitiveDictionary containing MongoDB URI options. renamedz0Deprecated option '%s' ignored in favor of '%s'.rbrz,Option '%s' is deprecated, use '%s' instead.removedzOption '%s' is deprecated. %s.)listrrrrDeprecationWarningpop)roptnamemodemessage newoptnamewarn_msgrNrNrO_handle_option_deprecations2s:   rcCsX|d}|durtD]}|||<q t|D]}t|d}|dur)||||<q|S)zNormalizes option names in the options dictionary by converting them to their internally-used names. :param options: Instance of _CaseInsensitiveDictionary containing MongoDB URI options. rN)rrrrr)rrrrintnamerNrNrO_normalize_optionsYs    rFMapping[str, Any]rMutableMapping[str, Any]cCs 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. :param opts: A dict of MongoDB URI options. :param warn: If ``True`` then warnings will be logged and invalid options will be ignored. Otherwise invalid options will cause errors. )r)rzrrNrNrOvalidate_optionsos rTvalidate normalizecCs|d}|d}z1|dkr|dkrtd|dkr!t|d}n|dkr+t|d}n|ddkr8t|d}ntWn tyGtddwt|}t|}|rVt|}|rkttt ||}| d d krktd |S) aTakes the options portion of a MongoDB URI, validates each option and returns the options in a dictionary. :param opt: A string representing MongoDB URI options. :param validate: If ``True`` (the default), validate and normalize all options. :param warn: If ``False`` (the default), suppress all warnings raised during validation of options. :param normalize: If ``True`` (the default), renames all options to their internally-used names. &;rz-Can not mix '&' and ';' for option separatorsr}raNz'MongoDB URI options are key=value pairs authsourcez1the authSource database cannot be an empty string) rcrrrdrrrr rrr)rzrrrand_idxsemi_idxrrNrNrO split_options~s2       rhostslist[_Address]cCsFg}|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. :param hosts: A string of the form host1[:port],host2[:port],... :param default_port: The port number to use when one wasn't specified for a host. ,z(Empty host (or extra comma in host list)rgN)rorrnrrv)rr]nodesr\rurNrNrO split_hostss  rrfz/ "$r`)rr replicasetr- loadbalancedrrrNonecCsht|dkr|drtd|dr0t|dkrtd|dr'td|dr2tddSdS) NrUdirectconnectionz8Cannot specify multiple hosts with directConnection=truerz4Cannot specify multiple hosts with loadBalanced=truez;Cannot specify directConnection=true with loadBalanced=truerz0Cannot specify replicaSet with loadBalanced=true)rKrr)rrrNrNrO_check_optionss    rkms_tls_optionsOptional[Mapping[str, Any]]is_syncdict[str, SSLContext]c Cs|siSt|ts tdi}|D]S\}}t|ts$td|d|ddt|}t|}t|}ttt |}t ||\}}|durLt d|rRt dd D]}||vrat d ||||<qTq|S) z!Parse KMS TLS connection options.zkms_tls_options must be a dictzkms_tls_options["z"] must be a dictr7TNz!TLS is required for KMS providerszInsecure TLS options prohibited)r?r8r9r=z!Insecure TLS options prohibited: ) rpdict TypeErroritemsrrrrr rr r) rrcontextsproviderrrz ssl_contextallow_invalid_hostnamesnrNrNrO_parse_kms_tls_optionss0     ruri srv_max_hostsdict[str, Any]cCs&|trd}|td}n'|tr*ts!tjpd}td|d}|td}n t dtdtd|s;t dd} d} d} d} t } | d \}}}d |vr[| d \}}} n|}| r{t | } d | vro| d d \} } t| rzt d | nd} |r| t||||d|vr|d\}}}t|\} } n|}d |vrt d|t |}d}|p| d}|r| drtdtdt|dd}t|d krt td|d\}}|durt tdn|s| ddurtd|s|rtdt||d}t|| || | | | | |dS)NFpythonzThe "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?/.rUzBad database name "%s"rSz;Any '/' in a unix domain socket must be percent-encoded: %sr4rz*Cannot specify directConnection=true with z URIs)r]z. URIs must include one, and only one, hostnamerz$ URIs must not include a port numberr5zDThe srvServiceName option is only allowed with 'mongodb+srv://' URIszAThe srvMaxHosts option is only allowed with 'mongodb+srv://' URIs)nodelistusernamepassworddatabase collectionrfqdn) startswithSCHEME SCHEME_LEN SRV_SCHEMEr sys executablerSRV_SCHEME_LENrrrWr ro _BAD_DB_CHARSsearchupdater rpartitionr[rrrKr)rr]rrrris_srv scheme_free python_pathrXrZdbaserrhost_plus_db_partrYrz host_partrQrrrrurNrNrO _validate_uris            rcCsNi}tD]}||vr||||<||q|D]\}}|||<q|Srh) URI_OPTIONSrsrr)rcase_sensitiveoptionkvrNrNrO_make_options_case_sensitive^s    r)rDrErFrG)rQrErFrR)r\rEr]r^rFr_)r\rEr]r^rFr)rzrEr{r|rFr)rrrFr)F)rzrrrGrFr)TFT) rzrErrGrrGrrGrFr)rrEr]r^rFr)rrrrrFr)rrrrGrFr)rrEr]r^rrGrrGrrGrr^rFr)rrrFr)=__doc__ __future__rrerrtypingrrrrrrr r urllib.parser !pymongo.asynchronous.srv_resolverr pymongo.client_optionsr pymongo.commonrrrrpymongo.errorsrrpymongo.typingsrpymongo.pyopenssl_contextrrrKrrr DEFAULT_PORT frozensetrrPr[rervrrrrrrrrcompileescaper_ALLOWED_TXT_OPTSrrrrrNrNrNrOs` (     4  4   5 '  -  & `