o 2hoN@s&dZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZed Zed ejd ZejGd d d e eZGdddeeZGdddeeZGdddeeZGddde eee eZ Gddde eZ!dS)zContains Flag class - information about single command-line flag. Do NOT import this module directly. Import the flags package and use the aliases defined at the package level instead. )abcN) AnyDictGenericIterableListOptionalTypeTypeVarUnion)minidom)_argument_parser) _exceptions)_helpers_T_ET)boundc@sReZdZUdZeeed<eeed<eeeefed<e j eed<       dFde j ed ee j ed edeeeefd eed eed e de de de de de ddfddZ edeefddZejdeefddZddZddZddZd d!Zd"d#Zd$d%Zd&eeefdd'fd(d)Zdeedeefd*d+Zd,eeefddfd-d.Zd,eeefdeefd/d0ZdGd1d2Zdefd3d4Zdeedefd5d6Z deeeefddfd7d8Z!deeefdeefd9d:Z"defd;d<Z# dHd=e$j%d>ed?e de$j&fd@dAZ'deedefdBdCZ(d=e$j%de)e$j&fdDdEZ*dS)IFlaga Information about a command-line flag. Attributes: name: the name for this flag default: the default value for this flag default_unparsed: the unparsed default value for this flag. default_as_str: default value as repr'd string, e.g., "'true'" (or None) value: the most recent parsed value of this flag set by :meth:`parse` help: a help string or None if no help is available short_name: the single letter alias for this flag (or None) boolean: if 'true', this flag does not accept arguments present: true if this flag was parsed from command line flags parser: an :class:`~absl.flags.ArgumentParser` object serializer: an ArgumentSerializer object allow_override: the flag may be redefined without raising an error, and newly defined flag overrides the old one. allow_override_cpp: use the flag from C++ if available the flag definition is replaced by the C++ flag after init allow_hide_cpp: use the Python flag despite having a C++ flag with the same name (ignore the C++ flag) using_default_value: the flag value has not been set by user allow_overwrite: the flag may be parsed more than once without raising an error, the last set value will be used allow_using_method_names: whether this flag can be defined even if it has a name that conflicts with a FlagValues method. validators: list of the flag validators. The only public method of a ``Flag`` object is :meth:`parse`, but it is typically only called by a :class:`~absl.flags.FlagValues` object. The :meth:`parse` method is a thin wrapper around the :meth:`ArgumentParser.parse()` method. The parsed value is saved in ``.value``, and the ``.present`` attribute is updated. If this flag was already present, an Error is raised. :meth:`parse` is also called during ``__init__`` to parse the default value and initialize the ``.value`` attribute. This enables other python modules to safely use flags even if the ``__main__`` module neglects to parse the command line arguments. The ``.present`` attribute is cleared after ``__init__`` parsing. If the default value is set to ``None``, then the ``__init__`` parsing step is skipped and the ``.value`` attribute is initialized to None. Note: The default value is also presented to the user in the help string, so it is important that it be a legal value for this flag. defaultdefault_as_strdefault_unparsedparserNFT serializername help_string short_namebooleanallow_overrideallow_override_cppallow_hide_cppallow_overwriteallow_using_method_namesreturnc Cs||_|sd}||_||_||_d|_||_||_||_| |_| |_ | |_ | |_ d|_ d|_ g|_|j r<|jr)Nzcan't pickle Flag objectsr?r1r,r,r- __getstate__r:zFlag.__getstate__cCstdt|j)Nz>%s does not support shallow copies. Use copy.deepcopy instead.)r@type__name__r1r,r,r-__copy__sz Flag.__copy__memozFlag[_T]cCs"tt|}t|j||_|Sr/)object__new__rCcopydeepcopy__dict__)r+rFresultr,r,r- __deepcopy__szFlag.__deepcopy__cCsH|durdS|jrt|j|S|jr|rtdStdStt|S)z$Returns parsed flag value as string.Ntruefalse)rrepr serializerstrr3r,r,r-_get_parsed_value_as_strings z Flag._get_parsed_value_as_stringargumentcCsB|jr|jstd|j||jf|||_|jd7_dS)z}Parses string and sets flag value. Args: argument: str or the correct flag value type, argument to be parsed. z#flag --%s=%s: already defined as %sN)r%r rIllegalFlagValueErrorrr2_parse)r+rTr,r,r-parses   z Flag.parsec CsDz|j|WStttfy!}z td|j||fd}~ww)zInternal parse function. It returns the parsed value, and does not modify class states. Args: argument: str or the correct flag value type, argument to be parsed. Returns: The parsed value. zflag --%s=%s: %sN)rrXr@ ValueError OverflowErrorrrVr)r+rTer,r,r-rWs z Flag._parsecCs|j|_d|_d|_dS)NTr)rr2r&r%r1r,r,r-unparses z Flag.unparsecCs ||jS)zSerializes the flag.) _serializer2r1r,r,r-rQs zFlag.serializecCsV|durdS|jr|rd|jSd|jS|js td|jd|j|j|fS)zInternal serialize function.Nz--%sz--no%s"Serializer not present for flag %sz--%s=%s)rrrrr)rQr3r,r,r-r]s  zFlag._serializecCsF||_|dur d|_n|||_||j|_|jr!|j|_dSdS)z@Changes the default value (and current value too) for this Flag.N)rr_parse_from_defaultrSrr&r2r3r,r,r-r*s  zFlag._set_defaultcCs ||Sr/)rWr3r,r,r-r`s zFlag._parse_from_defaultcCs |jS)aReturns a str that describes the type of the flag. NOTE: we use strings, and not the types.*Type constants because our flags can have more exotic types, e.g., 'comma separated list of strings', 'whitespace separated list of strings', etc. r flag_typer1r,r,r-rbs zFlag.flag_typedoc module_nameis_keycCs&|d}|r|t|dd|t|d||t|d|j|jr4|t|d|j|jrB|t|d|j|jr[t|j t s[|j durX|j |j }nd }n|j }|t|d || |j }|t|d ||t|d |||D]}||q|S) a6Returns an XML element that contains this flag's information. This is information that is relevant to all flags (e.g., name, meaning, etc.). If you defined a flag that has some other pieces of info, then please override _ExtraXMLInfo. Please do NOT override this method. Args: doc: minidom.Document, the DOM document it should create nodes from. module_name: str,, the name of the module that defines this flag. is_key: boolean, True iff this flag is key for main module. Returns: A minidom.Element instance. flagkeyyesfilerrmeaningNr^rcurrentrC) createElement appendChildrcreate_xml_dom_elementrrr$rr;rrRrQ_serialize_value_for_xmlr2rb_extra_xml_dom_elements)r+rcrdreelementdefault_serializedvalue_serializedr[r,r,r-_create_xml_dom_elementsB     zFlag._create_xml_dom_elementcCs|S)z:Returns the serialized value, for use in an XML help text.r,r3r,r,r-ro<szFlag._serialize_value_for_xmlcCs |j|S)aReturns extra info about this flag in XML. "Extra" means "not already included by _create_xml_dom_element above." Args: doc: minidom.Document, the DOM document it should create nodes from. Returns: A list of minidom.Element. )r_custom_xml_dom_elements)r+rcr,r,r-rp@s zFlag._extra_xml_dom_elements)NFFFFTF)r"NF)+rD __module__ __qualname____doc__rr__annotations__rRr r ArgumentParserArgumentSerializerboolr.propertyr2setterr6r9r=rArBrErintrrMrSrXrWr\rQr]r*r`rbr DocumentElementrtrorrpr,r,r,r-r$s  0         *     6rc sNeZdZdZ d dedeeeefdeedeeddf fdd ZZ S) BooleanFlagaBasic boolean flag. Boolean flags do not take any arguments, and their value is either ``True`` (1) or ``False`` (0). The false value is specified on the command line by prepending the word ``'no'`` to either the long or the short flag name. For example, if a Boolean flag was created whose long name was ``'update'`` and whose short name was ``'x'``, then this flag could be explicitly unset through either ``--noupdate`` or ``--nox``. Nrrr$rr"c s,t}tj|d||||dfi|dSNT)r BooleanParsersuperr.)r+rrr$rargsp __class__r,r-r._s$zBooleanFlag.__init__r/) rDrwrxryrRr rr}r. __classcell__r,r,rr-rRsrc sveZdZUdZejed<  ddedeedeede ed eed e f fd d Z d e j dee jfddZZS)EnumFlagzFBasic enum flag; its value can be any string from list of enum_values.rNTrrr$ enum_valuesrcase_sensitivec  sTt||}t} tj|| ||||fi|||_dd|j|jf|_dS)N<%s>: %s|) r EnumParserr|rr.rjoinrr$) r+rrr$rrrrrgrr,r-r.ps zEnumFlag.__init__rcr"cCs*g}|jjD] }|t|d|q|SN enum_value)rrappendrrnr+rcelementsrr,r,r-rps  z EnumFlag._extra_xml_dom_elementsr)rDrwrxryr rrzrRrrr}r.r rrrrprr,r,rr-rks.  rc s~eZdZUdZejed<  ddedee e efde ede e d e ed e f fd d Z d ejdeejfddZZS) EnumClassFlagz5Basic enum flag; its value is an enum class's member.rNFrrr$ enum_classrrc  s\tj||d}tj| d} tj|| ||||fi|||_dd|j|jf|_dS)Nr) lowercaserr) r EnumClassParserEnumClassSerializerrr.rr member_namesr$) r+rrr$rrrrrrrr,r-r.s zEnumClassFlag.__init__rcr"cC0g}|jjjD] }|t|d|q |Srrr __members__keysrrrnrr,r,r-rp z%EnumClassFlag._extra_xml_dom_elements)NF)rDrwrxryr rrzrRr rrr r}r.r rrrrprr,r,rr-rs.  rcseZdZdZfddZdeeeeeffddZ deeeeefde effdd Z d e e edeffd d Z d dZdejde ejfddZZS) MultiFlagaeA flag that can appear multiple time on the command-line. The value of such a flag is a list that contains the individual values from all the appearances of that flag on the command-line. See the __doc__ for Flag for most behavior of this class. Only differences in behavior are described here: * The default value may be either a single value or an iterable of values. A single value is transformed into a single-item list of that value. * The value of the flag is always a list, even if the option was only supplied once, and even if the default value is a single value cs$tj|i||jd7_dS)Nz4; repeat this option to specify a list of values)rr.r$)r+rkwargsrr,r-r.szMultiFlag.__init__ argumentscCsH||}|jr|jdusJ|j|n||_|jt|7_dS)aParses one or more arguments with the installed parser. Args: arguments: a single argument or a list of arguments (typically a list of default values); a single argument is converted internally into a list containing one item. N)rWr%r2extendlen)r+r new_valuesr,r,r-rXs zMultiFlag.parser"csBt|tr |g}nt|tjrt|}n|g}fdd|DS)Ncg|] }tt|qSr,)rrrW).0itemrr+r,r- sz$MultiFlag._parse..)r;rRrrlist)r+rarguments_listrr1r-rWs   zMultiFlag._parser2cs@js tdj|durdSfdd|D}d|S)See base class.r_Nr^crr,)rrr])r value_itemrr,r-rsz(MultiFlag._serialize.. )rrr)rr)r+r2serialized_itemsrr1r-r]s  zMultiFlag._serializecCsd|jS)rzmulti rar1r,r,r-rbszMultiFlag.flag_typerccCs6g}t|jdr|jjD] }|t|d|q |S)Nrr)hasattrrrrrrnrr,r,r-rps  z!MultiFlag._extra_xml_dom_elements)rDrwrxryr.r rRrrrXrrWrr]rbr rrrprr,r,rr-rs (rc seZdZUdZejeed< ddede de eee eefdede ed e f fd d Z d ejd eejfddZddZZS)MultiEnumClassFlagzA multi_enum_class flag. See the __doc__ for MultiFlag for most behaviors of this class. In addition, this class knows how to handle enum.Enum instances as values for this flag type. rFrrNrrrc  sdtj||d}tjd| d}tj|||||fi|||_||_dd|j|p,df|_ dS)Nr,)list_seprz<<%s>: %s; repeat this option to specify a list of valuesrr#) r rEnumClassListSerializerrr.rrrrr$) r+rrrrrrrrrr,r-r. s zMultiEnumClassFlag.__init__rcr"cCrrrrr,r,r-rp#rz*MultiEnumClassFlag._extra_xml_dom_elementscCs6|dur|jstd|j|j|}|Sd}|S)rNr_r^)rrr)rrQ)r+r2rsr,r,r-ro,s z+MultiEnumClassFlag._serialize_value_for_xmlrv)rDrwrxryr rrrzrRr rr r}r.r rrrrprorr,r,rr-rs*   r)"ry collectionsrrIenum functoolstypingrrrrrrr r r xml.domr absl.flagsr rrrEnumrtotal_orderingrr}rrRrrrrr,r,r,r-s* ,    / "T