updated doc string and wrapper
This commit is contained in:
@@ -29,6 +29,14 @@ import gflags
|
|||||||
|
|
||||||
|
|
||||||
class FlagValues(gflags.FlagValues):
|
class FlagValues(gflags.FlagValues):
|
||||||
|
"""Extension of gflags.FlagValues that allows undefined and runtime flags.
|
||||||
|
|
||||||
|
Unknown flags will be ignored when parsing the command line, but the
|
||||||
|
command line will be kept so that it can be replayed if new flags are
|
||||||
|
defined after the initial parsing.
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
gflags.FlagValues.__init__(self)
|
gflags.FlagValues.__init__(self)
|
||||||
self.__dict__['__dirty'] = []
|
self.__dict__['__dirty'] = []
|
||||||
@@ -125,7 +133,7 @@ class FlagValues(gflags.FlagValues):
|
|||||||
FLAGS = FlagValues()
|
FLAGS = FlagValues()
|
||||||
|
|
||||||
|
|
||||||
def party_wrapper(func):
|
def _wrapper(func):
|
||||||
def _wrapped(*args, **kw):
|
def _wrapped(*args, **kw):
|
||||||
kw.setdefault('flag_values', FLAGS)
|
kw.setdefault('flag_values', FLAGS)
|
||||||
func(*args, **kw)
|
func(*args, **kw)
|
||||||
@@ -133,16 +141,16 @@ def party_wrapper(func):
|
|||||||
return _wrapped
|
return _wrapped
|
||||||
|
|
||||||
|
|
||||||
DEFINE_string = party_wrapper(gflags.DEFINE_string)
|
DEFINE_string = _wrapper(gflags.DEFINE_string)
|
||||||
DEFINE_integer = party_wrapper(gflags.DEFINE_integer)
|
DEFINE_integer = _wrapper(gflags.DEFINE_integer)
|
||||||
DEFINE_bool = party_wrapper(gflags.DEFINE_bool)
|
DEFINE_bool = _wrapper(gflags.DEFINE_bool)
|
||||||
DEFINE_boolean = party_wrapper(gflags.DEFINE_boolean)
|
DEFINE_boolean = _wrapper(gflags.DEFINE_boolean)
|
||||||
DEFINE_float = party_wrapper(gflags.DEFINE_float)
|
DEFINE_float = _wrapper(gflags.DEFINE_float)
|
||||||
DEFINE_enum = party_wrapper(gflags.DEFINE_enum)
|
DEFINE_enum = _wrapper(gflags.DEFINE_enum)
|
||||||
DEFINE_list = party_wrapper(gflags.DEFINE_list)
|
DEFINE_list = _wrapper(gflags.DEFINE_list)
|
||||||
DEFINE_spaceseplist = party_wrapper(gflags.DEFINE_spaceseplist)
|
DEFINE_spaceseplist = _wrapper(gflags.DEFINE_spaceseplist)
|
||||||
DEFINE_multistring = party_wrapper(gflags.DEFINE_multistring)
|
DEFINE_multistring = _wrapper(gflags.DEFINE_multistring)
|
||||||
DEFINE_multi_int = party_wrapper(gflags.DEFINE_multi_int)
|
DEFINE_multi_int = _wrapper(gflags.DEFINE_multi_int)
|
||||||
|
|
||||||
|
|
||||||
def DECLARE(name, module_string, flag_values=FLAGS):
|
def DECLARE(name, module_string, flag_values=FLAGS):
|
||||||
|
|||||||
Reference in New Issue
Block a user