8dZddlmZddlmZddlmZddlZddlZddlmZe hdZ Gdd ejZ Gd d ejZ Gd d ejZGddejZdZdZy)a This module provides argparse integration with absl.flags. argparse_flags.ArgumentParser is a drop-in replacement for argparse.ArgumentParser. It takes care of collecting and defining absl flags in argparse. Here is a simple example: # Assume the following absl.flags is defined in another module: # # from absl import flags # flags.DEFINE_string('echo', None, 'The echo message.') # parser = argparse_flags.ArgumentParser( description='A demo of absl.flags and argparse integration.') parser.add_argument('--header', help='Header message to print.') # The parser will also accept the absl flag `--echo`. # The `header` value is available as `args.header` just like a regular # argparse flag. The absl flag `--echo` continues to be available via # `absl.flags.FLAGS` if you want to access it. args = parser.parse_args() # Example usages: # ./program --echo='A message.' --header='A header' # ./program --header 'A header' --echo 'A message.' Here is another example demonstrates subparsers: parser = argparse_flags.ArgumentParser(description='A subcommands demo.') parser.add_argument('--header', help='The header message to print.') subparsers = parser.add_subparsers(help='The command to execute.') roll_dice_parser = subparsers.add_parser( 'roll_dice', help='Roll a dice.', # By default, absl flags can also be specified after the sub-command. # To only allow them before sub-command, pass # `inherited_absl_flags=None`. inherited_absl_flags=None) roll_dice_parser.add_argument('--num_faces', type=int, default=6) roll_dice_parser.set_defaults(command=roll_dice) shuffle_parser = subparsers.add_parser('shuffle', help='Shuffle inputs.') shuffle_parser.add_argument( 'inputs', metavar='I', nargs='+', help='Inputs to shuffle.') shuffle_parser.set_defaults(command=shuffle) args = parser.parse_args(argv[1:]) args.command(args) # Example usages: # ./program --echo='A message.' roll_dice --num_faces=6 # ./program shuffle --echo='A message.' 1 2 3 4 There are several differences between absl.flags and argparse_flags: 1. Flags defined with absl.flags are parsed differently when using the argparse parser. Notably: 1) absl.flags allows both single-dash and double-dash for any flag, and doesn't distinguish them; argparse_flags only allows double-dash for flag's regular name, and single-dash for flag's `short_name`. 2) Boolean flags in absl.flags can be specified with `--bool`, `--nobool`, as well as `--bool=true/false` (though not recommended); in argparse_flags, it only allows `--bool`, `--nobool`. 2. Help related flag differences: 1) absl.flags does not define help flags, absl.app does that; argparse_flags defines help flags unless passed with `add_help=False`. 2) absl.app supports `--helpxml`; argparse_flags does not. 3) argparse_flags supports `-h`; absl.app does not. )absolute_import)division)print_functionN)flags>helphelpxmlundefokflagfilehelpfull helpshortc:eZdZdZfdZdfd ZdZdZxZS)ArgumentParserz:Custom ArgumentParser class to support special absl flags.c P|jdd}|dk7rtdj||jdtj |_tt|&d i||jrX|jddtjtj|jdttjd |j rL|jd tjtj |j|j y y )aInitializes ArgumentParser. Args: **kwargs: same as argparse.ArgumentParser, except: 1. It also accepts `inherited_absl_flags`: the absl flags to inherit. The default is the global absl.flags.FLAGS instance. Pass None to ignore absl flags. 2. The `prefix_chars` argument must be the default value '-'. Raises: ValueError: Raised when prefix_chars is not '-'. prefix_chars-zTargparse_flags.ArgumentParser only supports "-" as the prefix character, found "{}".inherited_absl_flagsz --helpshortr)actiondefaultrz --helpfullzshow full help message and exitz --undefok)rrN)get ValueErrorformatpoprFLAGS_inherited_absl_flagssuperr__init__add_help add_argumentargparseSUPPRESS_HelpFullAction_define_absl_flags)selfkwargsr __class__s 4platform/bq/third_party/absl/flags/argparse_flags.pyrzArgumentParser.__init__rs ::nc2Ls  ##)6,#7 99 "(,BEKK!PD .$(262 }}  ##(*;*;=  ##*KM !!  x00x7H7HJ d889"cF|tjdd}|jr|jj|d}t }t |d|}t t|#||\}}||ur||_ |jr]t|drt|j|}|` |jj |jj||fS||fS#tj$r'}|j!t#|Yd}~||fSd}~wwxYw)NT) force_gnur )sysargvrread_flags_from_filesobjectgetattrrrparse_known_argsr hasattr_strip_undefok_argsmark_as_parsedvalidate_all_flagsrIllegalFlagValueErrorerrorstr)r$args namespaceundefok_missingr er&s r'r1zArgumentParser.parse_known_argss' | XXab\d !! ' ' = = $> dhOiOz&_strip_undefok_args..qsD1C 1Cs,c3&K|] }d|z yw)rtNrrs r'rz&_strip_undefok_args..rs?s)r>split _is_undefok)r r9 undefok_namesargs r'r3r3ns[ Ds1CDDMS????M G4C{3 'FC4D G + Hs AAc|jdsy|jdr|dd}n|dd}d|vr|jdd\}}n|}||vryy) zGReturns whether we can ignore arg based on a set of undefok flag names.rFrHrrNr*=T)rxr)rrarg_without_dashrA_s r'rrxsg   ^^D12w12w $$S!,GD! D ]  r()rY __future__rrrr r,abslrrnr@rActionrRrPr"r3rrr(r'rsJX'% |'X,,|'~!4(//!4H1414h%hoo%Pr(