Add log level command line argument
Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user