o Si3 @sUdZddlmZmZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z dZdZdZed ZeejehZeeed <dZe ed <dZejdBed <Gd dde Zdeee fdedBfddZ defddZ!defddZ"de j#dede de j$fddZ%defddZ&dedeede'efd d!Z(d"d#Z)  $ d4d%ed&edBd'ed(edBdef d)d*Z* d5d+eee fd,eedBdeefd-d.Z+d/edefd0d1Z,dedefd2d3Z-dS)6z:Internal helper functions for Abseil Python flags library.)IterableSequenceN)Any NamedTuple)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__strrrG/home/ubuntu/.local/lib/python3.10/site-packages/absl/flags/_helpers.pyr Fs   r globals_dictreturncCsJz |d}tj|}Wn tyYdSwt||dkr"tjdS|S)aFReturns 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 if the module could not be identified. rN__main__r)sysmodulesKeyErrorr argv)rnamer rrrget_module_object_and_nameQs  r!cCsNtdtD]}t|j}t|}|dur"t|jtvr"|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) rangergetrecursionlimit _getframe f_globalsr!idr r AssertionError)depthglobals_for_framer rrr"get_calling_module_object_and_nameis 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_REGEXsub createElement appendChildcreateTextNode)r-r r.serrrcreate_xml_dom_elements   r:cCstjtdfd}|jS)zAReturns the integer width of help lines that is used in TextWrap.r")fallback)shutilget_terminal_size_DEFAULT_HELP_WIDTHcolumns)sizerrrget_help_widthsrAattempt 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)rGoptionrBrrrIsr)rKsort _SUGGESTION_ERROR_RATE_THRESHOLDappend) rBrC option_names distances least_errors_ suggestionserrorsr rrMrget_flag_suggestionss     rWcsifdd||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"NrrD)rKmin)xydtdistancememorrr^s"    (8 z&_damerau_levenshtein..distancer)abrr]rrJs rJr/textlengthindentfirstline_indentcCs|durt}|dur d}|dur|}t||krtdt||kr'td|d}g}tj|||d}tj|||d}dd|DD]}|rT|||n| d|}qGd |S) a:Wraps 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: Text to wrap. length: Maximum length of a line, includes indentation. If this is `None` then use `get_help_width()`. indent: Indent for all but first line. firstline_indent: Indent for first line. If `None`, fall back to `indent`. Returns: The wrapped text. Raises: ValueError: Raised if indent or firstline_indent not shorter than length. Nr/zLength of indent exceeds lengthz*Length of first line indent exceeds length)widthinitial_indentsubsequent_indentcss|]}|VqdSN)strip)rGprrr ztext_wrap.. ) rArK ValueError expandtabstextwrap TextWrapper splitlinesextendwraprPjoin)rbrcrdreresultwrappersubsequent_wrapper paragraphrrr text_wraps0     r|flag_map multi_flagsc cs|D]f\}}|durd|Vqt|tr&|r d|Vqd|Vqt|ttfr5d||fVqz&|rK||vrK|D] }d|t|fVq>nd|ddd|DfVWqtykd||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%sz--%s=%s,css|]}t|VqdSrj)r)rGitemrrrrmLrnz$flag_dict_to_args..)itemsr0r1bytesrrw TypeError)r}r~keyr.rrrrflag_dict_to_argss*      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. r/i r"Nrro) rqrtlstriprXrKrkrPrstrippoprw)r max_indentlinesrdlinestrippedtrimmedrrrtrim_docstringRs*      rcCsB|}tdtj}|d|}t|}tjdd|tjd}|S)z0Takes a __doc__ string and reformats it as help.z^[ ]+$r/z(?<=\S)\n(?=\S) )flags)rkrecompileMr4r)r-whitespace_only_linerrr doc_to_help|s  r)Nr/Nrj).rcollections.abcrrrr<rrrrtypingrrxml.domrr>_MIN_HELP_WIDTHrOrr3r'rrr setintrr r rr dictrr!r+r,DocumentElementr:rAlistrWrJr|rrrrrrrs        ?   4*