Elgato_dark/bunkerweb
1
0
Fork 0
mirror of https://github.com/bunkerity/bunkerweb synced 2026-05-24 09:28:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: handle ArgumentError in plugin command parsing and improve debug logging

Browse source
This commit is contained in:
Théophile Diot 2024-11-27 15:58:11 +01:00
parent b52476900d
commit b5316a151f
No known key found for this signature in database
GPG key ID: FA995104A0BA376A
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/common/cli/main.py
View file

@ -1,6 +1,7 @@
#!/usr/bin/env python3
from argparse import ArgumentParser
from argparse import ArgumentError, ArgumentParser
from contextlib import suppress
from os import _exit, getenv, sep
from os.path import join
from sys import exit as sys_exit, path as sys_path
@ -71,11 +72,14 @@ if __name__ == "__main__":
ret, err = cli.ban(args.ip, args.exp, args.reason)
elif args.command == "bans":
ret, err = cli.bans()
elif args.command == "plugin":
if args.debug:
logger.setLevel("DEBUG")
else:
with suppress(ArgumentError):
plugin_args = parser_plugin.parse_args()
ret, err = cli.custom(args.plugin_id, args.command, *args.arg, debug=args.debug)
if args.debug:
logger.setLevel("DEBUG")
ret, err = cli.custom(args.plugin_id, args.command, *args.arg, debug=args.debug)
if not ret:
logger.error(f"CLI command status : ❌ (fail)\n{err}")