From 122850d7a90428b6d7b92fe6100a1f2a6df2a1eb Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Sat, 30 Sep 2017 23:05:00 +0200
Subject: o Switching from YAML to INI files for downloaded facts. o Improved
 fact downloader.

---
 src/ee/tools/digikey_download_facts.py | 66 ++++++++++++++++++++++++----------
 1 file changed, 48 insertions(+), 18 deletions(-)

(limited to 'src/ee/tools')

diff --git a/src/ee/tools/digikey_download_facts.py b/src/ee/tools/digikey_download_facts.py
index 0550c5f..950623f 100644
--- a/src/ee/tools/digikey_download_facts.py
+++ b/src/ee/tools/digikey_download_facts.py
@@ -1,5 +1,6 @@
 import argparse
 from itertools import *
+from functools import total_ordering
 
 from colors import color
 
@@ -7,6 +8,22 @@ import ee.digikey as dk
 from ee.digikey import SearchResponseTypes, DigikeyProduct
 from ee.tools import mk_parents
 
+
+@total_ordering
+class Query(object):
+    def __init__(self, is_mpn, query):
+        self.is_mpn = is_mpn
+        self.query = query
+
+    def __eq__(self, other):
+        return (self.is_mpn, self.query) == (other.is_mpn, other.query)
+
+    def __lt__(self, other):
+        return (self.is_mpn, self.query) < (other.is_mpn, other.query)
+
+    def __hash__(self):
+        return hash((self.is_mpn, self.query))
+
 parser = argparse.ArgumentParser(description="Download facts about parts from Digi-Key")
 
 parser.add_argument("--out",
@@ -31,7 +48,7 @@ args = parser.parse_args()
 
 digikey = dk.Digikey()
 client = dk.DigikeyClient(digikey, on_download=lambda s: print(color(s, 'grey')))
-repo = dk.DigikeyRepository(args.out)
+repo = dk.DigikeyRepository(digikey, args.out)
 
 
 def on_product(product: DigikeyProduct):
@@ -41,7 +58,7 @@ def on_product(product: DigikeyProduct):
 parts = []
 
 if args.part:
-    parts.append(args.part)
+    [parts.append(Query(True, p)) for p in args.part]
 
 if args.sch:
     from ee.kicad import read_schematic, to_bom
@@ -49,39 +66,52 @@ if args.sch:
     for c in to_bom(sch):
         digikey = c.get_field("digikey")
         if digikey:
-            parts.append(digikey.value)
+            parts.append(Query(False, digikey.value))
         mpn = c.get_field("mpn")
         if mpn:
-            parts.append(mpn.value)
+            parts.append(Query(True, mpn.value))
 
-for p in parts:
-    print(color("Searching for {}".format(p), "white"))
+parts = sorted(set(parts))
 
-    if repo.has_product(p) and not args.force:
+for q in parts:
+    p = q.query
+    if repo.has_product(mpn = p if q.is_mpn else None, dpn = p if not q.is_mpn else None) and not args.force:
+        print(color("Already have facts for {}".format(p), "white"))
         continue
 
+    print(color("Searching for {}".format(p), "white"))
     response = client.search(p)
 
+    todos = []
+
     if response.response_type == SearchResponseTypes.SINGLE:
         p = response.products[0]
-        print(color("Direct match {}".format(p.mpn), "white"))
+        print(color("Direct match mpn/dpn: {}/{}".format(p.mpn, p.part_number), "white"))
         on_product(p)
     elif response.response_type == SearchResponseTypes.MANY:
-        hits = list(groupby(sorted(response.products), lambda p: p.mpn))
+        hits = [(mpn, list(products)) for mpn, products in groupby(sorted(response.products, key=lambda p: p.mpn), lambda p: p.mpn)]
 
         if len(hits) == 1:
             (mpn, products) = hits[0]
-            products = list(products)
-
-            if len(products) == 1:
-                print(color("Got many results, but they all point to the same part: {}".format(mpn), "white"))
-                on_product(products[0])
-                continue
 
-        for k, g in hits:
-            print(color("Got many results with many parts: {}: {}".format(k, list(g)), "white"))
-            on_product(list(g)[0])
+            part_number = products[0].part_number
+            print(color("Got many results, but they all point to the same part: {}. Will use {} for downloading attributes.".format(mpn, ", ".join([p.part_number for p in products])), "white"))
+            todos.append(part_number)
+        else:
+            for k, g in hits:
+                print(color("Got many results with many parts: {}: {}".format(k, list(g)), "white"))
+                on_product(list(g)[0])
     elif response.response_type == SearchResponseTypes.TOO_MANY:
         print(color("Too many results ({}), select a category first".format(response.count), 'red'))
     elif response.response_type == SearchResponseTypes.NO_MATCHES:
         print(color("Part not found", "orange"))
+
+    for part_number in todos:
+        response = client.search(part_number)
+
+        if response.response_type == SearchResponseTypes.SINGLE:
+            p = response.products[0]
+            print(color("Downloaded {}".format(p.mpn), "white"))
+            on_product(p)
+        else:
+            print("Got response of type {} when really expecting a single product.".format(response.response_type))
-- 
cgit v1.2.3