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':
|
elif arg.name == 'kwargs':
|
||||||
pass
|
pass
|
||||||
else:
|
else:
|
||||||
if doc:
|
|
||||||
arg_doc = None
|
arg_doc = None
|
||||||
|
if doc:
|
||||||
for line in doc:
|
for line in doc:
|
||||||
match = re.match(rf"^:param {arg.name}: (.*)", line)
|
match = re.match(rf"^:param {arg.name}: (.*)", line)
|
||||||
if match:
|
if match:
|
||||||
arg_doc = match.group(1)
|
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
|
return parser
|
||||||
|
|
Ładowanie…
Reference in New Issue