Add log level command line argument

Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
2023-07-29 21:07:59 -07:00
parent 918271926a
commit b190d5033d

View File

@@ -7,6 +7,8 @@ from .pull import pull
def parse_args():
parser = argparse.ArgumentParser(description="Run 4chan bans archiver")
parser.add_argument('--log-level', choices=['debug', 'info', 'warning', 'error', 'critical'], default='info')
subparsers = parser.add_subparsers(title="Commands", dest="command")
subcommands = set()
@@ -36,7 +38,13 @@ def parse_args():
async def main():
args = parse_args()
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
try:
level = getattr(logging, args.log_level.upper())
except AttributeError:
print(f"ERROR: no such logging level {args.log_level}. Exiting")
raise SystemExit()
logging.basicConfig(level=level, format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
match args.command:
case "pull":
await pull()