diff options
Diffstat (limited to 'src/ee/tools/part_apply_souffle_post.py')
| -rw-r--r-- | src/ee/tools/part_apply_souffle_post.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/ee/tools/part_apply_souffle_post.py b/src/ee/tools/part_apply_souffle_post.py index 961b0c0..6cc5f36 100644 --- a/src/ee/tools/part_apply_souffle_post.py +++ b/src/ee/tools/part_apply_souffle_post.py @@ -2,8 +2,10 @@ import argparse import csv from pathlib import Path +import ee.tools from ee import EeException from ee.db import ObjDb +from ee.logging import log from ee.part import Part, load_db, save_db, PartDb @@ -21,6 +23,7 @@ def work(in_path: Path, out_path: Path, work_dir: Path): mpn_csv = work_dir / "out" / "mpn.csv" + new_facts = 0 with (work_dir / "out" / "fact.csv").open("r") as f: reader = csv.reader(f, dialect="excel-tab") for uri, key, value in reader: @@ -32,8 +35,9 @@ def work(in_path: Path, out_path: Path, work_dir: Path): old_value = part.facts.get_value(key) if old_value is None: - print("New fact for {}: {}={}".format(uri, key, value)) + log.info("New fact for {}: {}={}".format(uri, key, value)) part.facts.add(key, value) + new_facts += 1 if mpn_csv.is_file(): with mpn_csv.open("r") as f: @@ -46,10 +50,12 @@ def work(in_path: Path, out_path: Path, work_dir: Path): part.add_mpn(mpn) + log.summary("New facts found: {}".format(new_facts)) save_db(out_path, out_parts) parser = argparse.ArgumentParser() +ee.tools.add_default_argparse_group(parser) parser.add_argument("--in", dest="in_path", @@ -65,5 +71,6 @@ parser.add_argument("--work", metavar="DIR") args = parser.parse_args() +ee.tools.process_default_argparse_group(args) work(Path(args.in_path), Path(args.out), Path(args.work)) |