U Ucv4@s*dZddlZddlZddlZddlZddlZddlZz ddlZWnek rXdZYnXz ddl Z Wnek r~dZ YnXdZ dZ dZ e dZeeejegZdZdZGdddedd Zd d Zd d ZddZddZddZddZddZd!ddZd"ddZ ddZ!dd Z"dS)#z:Internal helper functions for Abseil Python flags library.NP(g?u [- --„†-Ÿ-￾￿]c@seZdZdZdS)_ModuleObjectAndNamezkModule object and name. Fields: - module: object, module object. - module_name: str, module name. N)__name__ __module__ __qualname____doc__r r 7/tmp/pip-unpacked-wheel-yg88_g5m/absl/flags/_helpers.pyrJsrzmodule module_namecCs6|dd}tj|d}t||dkr0tjdn|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)Z globals_dictnamemoduler r r get_module_object_and_nameTs rcCsZtdtD]>}t|j}t|\}}t|tkr|dk rt||Sqt ddS)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_globalsriddisclaim_module_idsrAssertionError)depthZglobals_for_framer module_namer r r "get_calling_module_object_and_namefs   rcCstjS)z?Returns the name of the module that's calling into this module.)rrr r r r get_calling_module|srcCsDt|}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. ) str isinstanceboollower_ILLEGAL_XML_CHARS_REGEXsub createElement appendChildZcreateTextNode)docrvalueser r r create_xml_dom_elements    r,c CstjrtdkstdkrtSzBttjtjd}t d|d}|t krP|WSt t dtWStttjfk rtYSXdS)zAReturns the integer width of help lines that is used in TextWrap.NZ1234hhrCOLUMNS)r stdoutisattytermiosfcntl_DEFAULT_HELP_WIDTHZioctl TIOCGWINSZstructunpack_MIN_HELP_WIDTHintosgetenv TypeErrorIOErrorerror)datacolumnsr r r get_help_widthsr@c stdks|sgSdd|D}fdd|D}||d\}}|ttkr\gSg}|D] \}}||kr||qdqqd|S)z4Returns helpful similar matches for an invalid flag.cSsg|]}|ddqS)=r)split).0vr r r sz(get_flag_suggestions..cs&g|]}t|dt|fqS)r)_damerau_levenshteinlen)rDoptionattemptr r rFsr)rHsort _SUGGESTION_ERROR_RATE_THRESHOLDappend) rKZ longopt_list option_namesZ distancesZ least_errors_Z suggestionserrorsrr rJr get_flag_suggestionss     rRcsifdd||S)z6Returns Damerau-Levenshtein edit distance from a to b.cs||fkr||fS|s&t|}n|s4t|}nt|dd|d||ddd|dd|dd|d|dk}t|dkrt|dkr|d|dkr|d|dkr|dd|ddd}||kr|}|||f<|S)z5Recursively defined string distance with memoization.rNrrA)rHmin)xydtdistancememor r rYs"    (8 z&_damerau_levenshtein..distancer )abr rXr rGsrGrcCs|dkrt}|dkrd}|dkr&|}t||kr:tdt||krNtd|d}g}tj|||d}tj|||d}dd|DD](}|r|||n | d|}qd |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. NrzLength of indent exceeds lengthz*Length of first line indent exceeds length)widthinitial_indentsubsequent_indentcss|]}|VqdSN)strip)rDpr r r sztext_wrap.. ) r@rH ValueError expandtabstextwrap TextWrapper splitlinesextendwraprNjoin)textlengthindentZfirstline_indentresultwrapperZsubsequent_wrapperZ paragraphr r r text_wraps8    rsc cs|D]\}}|dkr$d|Vqt|trJ|r>d|Vqd|Vqt|ttdfrld||fVqzL|r||kr|D]}d|t|fVq~nd|ddd|DfVWqtk rd||fVYqXqdS) 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%srz--%s=%s,css|]}t|VqdSra)r )rDitemr r r rdJsz$flag_dict_to_args..)itemsr!r"bytestyper rmr;)Zflag_mapZ multi_flagskeyr)rur r r flag_dict_to_argss      "rzcCs|sdSd}|}|}|ddD]&}|}|r(t|t|t|}q(|dg}||kr|ddD]}|||dqr|r|ds|q|r|ds|dqd |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. ri rNrre) rgrjlstriprSrHrbrNrstrippoprm)Z docstringZ max_indentlinesrplinestrippedZtrimmedr r r trim_docstringPs$      rcCsB|}tdtj}|d|}t|}tjdd|tjd}|S)z0Takes a __doc__ string and reformats it as help.z^[ ]+$rz(?<=\S)\n(?=\S) )flags)rbrecompileMr%r)r(Zwhitespace_only_liner r r doc_to_helpzs  r)NrN)N)#r collectionsr9rr5r rhr2 ImportErrorr1r3r7rMrr$setrrrrZ SPECIAL_FLAGSZ FLAGS_MODULE namedtuplerrrrr,r@rRrGrsrzrrr r r r sJ         9 2*