From c34af93533845d0e41a43ead497e3ee946017cdf Mon Sep 17 00:00:00 2001 From: Andrei Stoica Date: Sat, 15 Oct 2022 14:19:15 -0400 Subject: [PATCH] moved to logger --- src/recipe_graph/insert_sites.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/src/recipe_graph/insert_sites.py b/src/recipe_graph/insert_sites.py index 50db413..a4a1c2b 100644 --- a/src/recipe_graph/insert_sites.py +++ b/src/recipe_graph/insert_sites.py @@ -4,6 +4,7 @@ from recipe_graph import db import json import argparse import logging +import sys def load_file(f_name: str): @@ -12,31 +13,32 @@ def load_file(f_name: str): return sites -def setup_argparser() -> argparse.Namespace: +def setup_argparser(args) -> argparse.Namespace: parser = argparse.ArgumentParser(description="Import recipes into database") parser.add_argument("file", type=str, help="JSON file with recipe site information") parser.add_argument("-v", "--verbose", action="store_true") - args = parser.parse_args() - return args + return parser.parse_args(args) -def setup_logging(args: argparse.Namespace): +def setup_logging(args: argparse.Namespace) -> logging.Logger: + logger = logging.Logger("insert_sites", logging.WARNING) if args.verbose: - logging.basicConfig(level=logging.INFO) + logger.setLevel(logging.INFO) logging.getLogger("sqlalchemy.engine").setLevel(logging.INFO) + return logger def main(): - args = setup_argparser() - setup_logging(args) + args = setup_argparser(sys.argv[1:]) + logger = setup_logging(args) S = db.get_session() sites = load_file(args.file) with S.begin() as session: for site in sites: - logging.info(f"Adding {site}") + logger.info(f"Adding {site}") session.add(db.RecipeSite(**site))