digikey-search-parts: Adding summary in download report.

1 files changed, 31 insertions, 3 deletions

diff --git a/src/ee/digikey/search_parts.py b/src/ee/digikey/search_parts.py
index c0da731..703e65f 100644
--- a/src/ee/digikey/search_parts.py
+++ b/src/ee/digikey/search_parts.py
@@ -80,7 +80,7 @@ class QueryEngine(object):
     def pn_spn_search(self, pn, is_spn):
         s = "Searching for '{}'".format(pn)
         print(s, file=self.log)
-        print("=" * len(s) + "\n", file=self.log)
+        print("-" * len(s) + "\n", file=self.log)
 
         out_part = None
         result = None
@@ -220,24 +220,52 @@ def run_search_parts(in_path: Path, out_path: Path, log, cache_dir: Path, store_
     pn_queries = list(sorted(pn_qs))
     spn_queries = list(sorted(spn_qs))
 
-    print("Executing {} manufacturer product number searches\n\n".format(len(pn_queries)), file=log)
+    s = "Manufacturer product number searches"
+    print("{}\n{}\n".format(s, "=" * len(s)), file=log)
+    print("Executing {} manufacturer product number searches:\n\n".format(len(pn_queries)), file=log)
 
     exception = False
+    mpn_results = {}
     for q in pn_queries:
         res = engine.pn_search(q.pn)
+
+        if res in mpn_results:
+            mpn_results[res] = mpn_results[res] + 1
+        else:
+            mpn_results[res] = 1
+
         if res == SearchResponseTypes.EXCEPTION:
             exception = True
             break
 
+    spn_results = {}
     if not exception:
-        print("Executing {} supplier product number searches\n\n".format(len(spn_queries)), file=log)
+        if len(spn_queries) == 0:
+            print("Executing no supplier product number searches.\n\n".format(len(spn_queries)), file=log)
+        else:
+            print("Executing {} supplier product number searches:\n\n".format(len(spn_queries)), file=log)
 
         for q in spn_queries:
             res = engine.spn_search(q.spn)
+
+            if res in spn_results:
+                spn_results[res] = spn_results[res] + 1
+            else:
+                spn_results[res] = 1
+
             if res == SearchResponseTypes.EXCEPTION:
                 exception = True
                 break
 
+    header = "Statistics"
+    print("{}\n{}\n".format(header, "=" * len(header)), file=log)
+
+    for name, results in [("MPN", mpn_results), ("SPN", spn_results)]:
+        print("{} Searches:\n".format(name), file=log)
+        for res, count in results.items():
+            print("* {}: {}".format(res.name, count), file=log)
+        print("", file=log)
+
     part_db = PartDb()
     for part in engine.out_parts:
         part_db.add_entry(part, True)