moved to logger

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))