all tools: Creating a shared group of argument options. Processing this

shared group too, changing the log level when logging options are
specified.

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