stats: Allow autouser arg in parser

pull/1/head
Michael DM Dryden 2020-05-30 00:28:36 -04:00
rodzic 6856b82a25
commit 170a2dfed1
2 zmienionych plików z 17 dodań i 1 usunięć

13
main.py
Wyświetl plik

@ -107,7 +107,18 @@ def print_stats(update: Update, context: CallbackContext):
func = args.pop('func')
try:
if args['me']:
if args['user']:
try:
uid = args['user']
args['user'] = uid, stats.users[uid][0]
except KeyError:
send_help("unknown userid", context, update)
return
except KeyError:
pass
try:
if args['me'] and not args['user']: # Lets auto-user work by ignoring auto-input me arg
args['user'] = update.effective_user.id, update.effective_user.name
del args['me']
except KeyError:

Wyświetl plik

@ -449,6 +449,11 @@ def get_parser(runner: StatsRunner) -> InternalParser:
group = subparser.add_mutually_exclusive_group()
group.add_argument('-me', action='store_true', help='calculate stats for yourself')
group.add_argument('-user', type=int, help=argparse.SUPPRESS)
elif arg.name == 'autouser':
subparser.set_defaults(me=True)
subparser.add_argument('-user', type=int, help=argparse.SUPPRESS)
elif arg.name == 'kwargs':
pass
else:
if doc:
arg_doc = None