o ȧe7 @s"UdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZddlmZzddlZWn eyFdZYnwzddlZWn eyXdZYnwdZdZdZedZeeejegZeee d <dZ!ee d <dZ"ej#e d <Gd d d e Z$de e%efde$fddZ&de$fddZ'de%fddZ(dej)de%dedej*fddZ+defddZ,de e%dee%de e%fdd Z-d!d"Z.  # d3d$e%d%e ed&e%d'e e%de%f d(d)Z/ d4d*e e%efd+e ee%de e%fd,d-Z0d.e%de%fd/d0Z1de%de%fd1d2Z2dS)5z:Internal helper functions for Abseil Python flags library.N)AnyDictIterableList NamedTupleOptionalSequenceSet)minidomP(g?u [- --„†-Ÿ-￾￿]disclaim_module_ids SPECIAL_FLAGS FLAGS_MODULEc@s$eZdZUdZejed<eed<dS)_ModuleObjectAndNamezkModule object and name. Fields: - module: object, module object. - module_name: str, module name. module module_nameN)__name__ __module__ __qualname____doc__types ModuleType__annotations__strrr3lib/python3.10/site-packages/absl/flags/_helpers.pyrPs   r globals_dictreturncCs8|dd}tj|d}t||dkrtjdS|S)aNReturns the module that defines a global environment, and its name. Args: globals_dict: A dictionary that should correspond to an environment providing the values of the globals. Returns: _ModuleObjectAndName - pair of module object & module name. Returns (None, None) if the module could not be identified. rN__main__r)getsysmodulesrargv)rnamerrrrget_module_object_and_name[s r%cCsVtdtD]}t|j}t|\}}t|tvr&|dur&t||Sqt d)a)Returns the module that's calling into this module. We generally use this function to get the name of the module calling a DEFINE_foo... function. Returns: The module object that called into this one. Raises: AssertionError: Raised when no calling module could be identified. NzNo module was found) ranger!getrecursionlimit _getframe f_globalsr%idr rAssertionError)depthZglobals_for_framerrrrr"get_calling_module_object_and_nameos  r.cCstjS)z?Returns the name of the module that's calling into this module.)r.rrrrrget_calling_modulesr/docr$valuecCsDt|}t|tr |}td|}||}||||S)aReturns an XML DOM element with name and text value. Args: doc: minidom.Document, the DOM document it should create nodes from. name: str, the tag of XML element. value: object, whose string representation will be used as the value of the XML element. Illegal or highly discouraged xml 1.0 characters are stripped. Returns: An instance of minidom.Element. ) r isinstanceboollower_ILLEGAL_XML_CHARS_REGEXsubZ createElementZ appendChildZcreateTextNode)r0r$r1serrrcreate_xml_dom_elements   r:c Cstjr tdus tdurtSz!ttjtjd}t d|d}|t kr(|WSt t dtWStttjfy?tYSw)zAReturns the integer width of help lines that is used in TextWrap.Ns1234Zhhr&COLUMNS)r!stdoutisattytermiosfcntl_DEFAULT_HELP_WIDTHZioctlZ TIOCGWINSZstructunpack_MIN_HELP_WIDTHintosgetenv TypeErrorIOErrorerror)datacolumnsrrrget_help_widthsrLattempt longopt_listc stdks|s gSdd|D}fdd|D}||d\}}|ttkr.gSg}|D]\}}||kr@||q2|S|S)z4Returns helpful similar matches for an invalid flag.cSsg|] }|ddqS)=r)split).0vrrr sz(get_flag_suggestions..cs&g|]}t|dt|fqS)r)_damerau_levenshteinlen)rRoptionrMrrrTsr)rVsort _SUGGESTION_ERROR_RATE_THRESHOLDappend) rMrNZ option_namesZ distancesZ least_errors_Z suggestionserrorsr$rrXrget_flag_suggestionss     r^csifdd||S)z6Returns Damerau-Levenshtein edit distance from a to b.cs||fvr ||fS|st|}nc|st|}n\t|dd|d||ddd|dd|dd|d|dk}t|dkrvt|dkrv|d|dkrv|d|dkrv|dd|ddd}||krv|}|||f<|S)z5Recursively defined string distance with memoization.r&NrrO)rVmin)xydtdistanceZmemorrres"    (8 z&_damerau_levenshtein..distancer)abrrdrrUs rUr2textlengthindentfirstline_indentcCs|durt}|dur d}|dur|}t||krtdt||kr'td|d}g}tj|||d}tj|||d}dd|DD]}|rT|||n| d|}qGd |S) aLWraps a given text to a maximum line length and returns it. It turns lines that only contain whitespace into empty lines, keeps new lines, and expands tabs using 4 spaces. Args: text: str, text to wrap. length: int, maximum length of a line, includes indentation. If this is None then use get_help_width() indent: str, indent for all but first line. firstline_indent: str, indent for first line; if None, fall back to indent. Returns: str, the wrapped text. Raises: ValueError: Raised if indent or firstline_indent not shorter than length. Nr2zLength of indent exceeds lengthz*Length of first line indent exceeds length)widthinitial_indentsubsequent_indentcss|]}|VqdSN)strip)rRprrr %ztext_wrap.. ) rLrV ValueError expandtabstextwrapZ TextWrapper splitlinesextendwrapr[join)rhrirjrkresultwrapperZsubsequent_wrapperZ paragraphrrr text_wraps0     rflag_map multi_flagsc cs|D]h\}}|durd|Vqt|tr&|r d|Vqd|Vqt|ttdfr7d||fVqz&|rM||vrM|D] }d|t|fVq@nd|ddd|DfVWqtymd||fVYqwdS) a3Convert a dict of values into process call parameters. This method is used to convert a dictionary into a sequence of parameters for a binary that parses arguments using this module. Args: flag_map: dict, a mapping where the keys are flag names (strings). values are treated according to their type: * If value is ``None``, then only the name is emitted. * If value is ``True``, then only the name is emitted. * If value is ``False``, then only the name prepended with 'no' is emitted. * If value is a string then ``--name=value`` is emitted. * If value is a collection, this will emit ``--name=value1,value2,value3``, unless the flag name is in ``multi_flags``, in which case this will emit ``--name=value1 --name=value2 --name=value3``. * Everything else is converted to string an passed as such. multi_flags: set, names (strings) of flags that should be treated as multi-flags. Yields: sequence of string suitable for a subprocess execution. Nz--%sz--no%sr2z--%s=%s,css|]}t|VqdSrp)r)rRitemrrrrs^rtz$flag_dict_to_args..)itemsr3r4bytestyperr|rG)rrkeyr1rrrrflag_dict_to_args0s*      r docstringcCs|sdSd}|}|}|ddD]}|}|r't|t|t|}q|dg}||krG|ddD] }|||dq9|rW|dsW||rW|drM|rh|dsh|d|rh|dr]d |S)a Removes indentation from triple-quoted strings. This is the function specified in PEP 257 to handle docstrings: https://www.python.org/dev/peps/pep-0257/. Args: docstring: str, a python docstring. Returns: str, docstring with indentation removed. r2i r&Nrru) rwrylstripr_rVrqr[rstrippopr|)rZ max_indentlinesrjlinestrippedZtrimmedrrrtrim_docstringds*      rcCsB|}tdtj}|d|}t|}tjdd|tjd}|S)z0Takes a __doc__ string and reformats it as help.z^[ ]+$r2z(?<=\S)\n(?=\S) )flags)rqrecompileMr7r)r0Zwhitespace_only_linerrr doc_to_helps  r)Nr2Nrp)3rrErrAr!rxrtypingrrrrrrrr Zxml.domr r? ImportErrorr>r@rCrZrr6setr+r"rr rDrrrrrrr%r.r/ZDocumentZElementr:rLr^rUrrrrrrrrs(             ?   4*