andrei-stoica
/
recipe-graph
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

moved to logger

This commit is contained in:
Andrei Stoica 2022-10-15 14:19:15 -04:00
parent 9b9e629548
commit c34af93533
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/recipe_graph/insert_sites.py
View File

@ -4,6 +4,7 @@ from recipe_graph import db
import json import json
import argparse import argparse
import logging import logging
import sys
def load_file(f_name: str): def load_file(f_name: str):
@ -12,31 +13,32 @@ def load_file(f_name: str):
return sites return sites
def setup_argparser() -> argparse.Namespace: def setup_argparser(args) -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Import recipes into database") parser = argparse.ArgumentParser(description="Import recipes into database")
parser.add_argument("file", type=str, help="JSON file with recipe site information") parser.add_argument("file", type=str, help="JSON file with recipe site information")
parser.add_argument("-v", "--verbose", action="store_true") parser.add_argument("-v", "--verbose", action="store_true")
args = parser.parse_args() return parser.parse_args(args)
return 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: if args.verbose:
logging.basicConfig(level=logging.INFO) logger.setLevel(logging.INFO)
logging.getLogger("sqlalchemy.engine").setLevel(logging.INFO) logging.getLogger("sqlalchemy.engine").setLevel(logging.INFO)
return logger
def main(): def main():
args = setup_argparser() args = setup_argparser(sys.argv[1:])
setup_logging(args) logger = setup_logging(args)
S = db.get_session() S = db.get_session()
sites = load_file(args.file) sites = load_file(args.file)
with S.begin() as session: with S.begin() as session:
for site in sites: for site in sites:
logging.info(f"Adding {site}") logger.info(f"Adding {site}")
session.add(db.RecipeSite(**site)) session.add(db.RecipeSite(**site))