updated doc string and wrapper

This commit is contained in:
andy
2010-08-03 18:00:11 +02:00
parent 0e216d0fe0
commit f3568872af

View File

@@ -29,6 +29,14 @@ import gflags
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):
gflags.FlagValues.__init__(self)
self.__dict__['__dirty'] = []
@@ -125,7 +133,7 @@ class FlagValues(gflags.FlagValues):
FLAGS = FlagValues()
def party_wrapper(func):
def _wrapper(func):
def _wrapped(*args, **kw):
kw.setdefault('flag_values', FLAGS)
func(*args, **kw)
@@ -133,16 +141,16 @@ def party_wrapper(func):
return _wrapped
DEFINE_string = party_wrapper(gflags.DEFINE_string)
DEFINE_integer = party_wrapper(gflags.DEFINE_integer)
DEFINE_bool = party_wrapper(gflags.DEFINE_bool)
DEFINE_boolean = party_wrapper(gflags.DEFINE_boolean)
DEFINE_float = party_wrapper(gflags.DEFINE_float)
DEFINE_enum = party_wrapper(gflags.DEFINE_enum)
DEFINE_list = party_wrapper(gflags.DEFINE_list)
DEFINE_spaceseplist = party_wrapper(gflags.DEFINE_spaceseplist)
DEFINE_multistring = party_wrapper(gflags.DEFINE_multistring)
DEFINE_multi_int = party_wrapper(gflags.DEFINE_multi_int)
DEFINE_string = _wrapper(gflags.DEFINE_string)
DEFINE_integer = _wrapper(gflags.DEFINE_integer)
DEFINE_bool = _wrapper(gflags.DEFINE_bool)
DEFINE_boolean = _wrapper(gflags.DEFINE_boolean)
DEFINE_float = _wrapper(gflags.DEFINE_float)
DEFINE_enum = _wrapper(gflags.DEFINE_enum)
DEFINE_list = _wrapper(gflags.DEFINE_list)
DEFINE_spaceseplist = _wrapper(gflags.DEFINE_spaceseplist)
DEFINE_multistring = _wrapper(gflags.DEFINE_multistring)
DEFINE_multi_int = _wrapper(gflags.DEFINE_multi_int)
def DECLARE(name, module_string, flag_values=FLAGS):