kopia lustrzana https://github.com/mkdryden/telegram-stats-bot
stats: make get_parser handle bool arguments properly
rodzic
ec83a7a691
commit
2bd9de78d8
9
stats.py
9
stats.py
|
@ -456,12 +456,17 @@ def get_parser(runner: StatsRunner) -> InternalParser:
|
|||
elif arg.name == 'kwargs':
|
||||
pass
|
||||
else:
|
||||
if doc:
|
||||
arg_doc = None
|
||||
if doc:
|
||||
for line in doc:
|
||||
match = re.match(rf"^:param {arg.name}: (.*)", line)
|
||||
if match:
|
||||
arg_doc = match.group(1)
|
||||
subparser.add_argument(f"-{arg.name}", type=arg.annotation, help=arg_doc)
|
||||
|
||||
if arg.annotation == bool:
|
||||
subparser.add_argument(f"-{arg.name}".replace('_', '-'), action='store_true', help=arg_doc)
|
||||
else:
|
||||
subparser.add_argument(f"-{arg.name}".replace('_', '-'), type=arg.annotation, help=arg_doc,
|
||||
default=arg.default)
|
||||
|
||||
return parser
|
||||
|
|
Ładowanie…
Reference in New Issue