1 files changed, 100 insertions, 0 deletions
diff --git a/src/ee/digikey/import_parts.py b/src/ee/digikey/import_parts.py
new file mode 100644
index 0000000..748bbef
--- /dev/null
+++ b/src/ee/digikey/import_parts.py
@@ -0,0 +1,100 @@
+import os
+from typing import List, MutableMapping
+
+from ee.xml import bomFile
+from ee.xml.bom_file_utils import *
+from ee.xml.uris import DIGIKEY_URI
+
+__all__ = ["import_parts"]
+
+
+class Entry(object):
+    def __init__(self, new: bool, part: bomFile.Part):
+        self.new = new
+        self.part = part
+
+        self.pn = find_pn(part)
+        self.dpn = find_dpn(part, DIGIKEY_URI)
+
+
+def import_parts(in_path, out_path):
+    print("in: {}, out: {}".format(in_path, out_path))
+
+    in_file = bomFile.parse(in_path, True)
+    if in_file.partsProp is None:
+        in_file.partsProp = bomFile.PartList()
+    in_part_list = in_file.partsProp.partProp  # type: List[bomFile.Part]
+
+    print("in file: {} parts".format(len(in_part_list)))
+
+    if os.path.isfile(out_path):
+        out_file = bomFile.parse(out_path, True)
+    else:
+        out_file = bomFile.BomFile()
+
+    if out_file.partsProp is None:
+        out_file.partsProp = bomFile.PartList()
+    out_part_list = out_file.partsProp.partProp  # type: List[bomFile.Part]
+    print("out file: {} parts".format(len(out_part_list)))
+
+    existing_parts = []  # type: List[Entry]
+    pn_index = {}  # type: MutableMapping[str, Entry]
+    dpn_index = {}  # type: MutableMapping[str, Entry]
+    new_entry_added = 0
+
+    def add_entry(e: Entry):
+        existing_parts.append(e)
+        pn_index[e.pn] = e
+        dpn_index[e.dpn] = e
+
+        if e.new:
+            out_part_list.append(e.part)
+            nonlocal new_entry_added
+            new_entry_added = new_entry_added + 1
+
+    print("len(out_part_list)={}".format(len(out_part_list)))
+    for part in out_part_list:  # type: bomFile.Part
+        entry = Entry(False, part)
+        add_entry(entry)
+
+    print("loaded {} existing parts".format(len(existing_parts)))
+
+    for part in in_part_list:
+        pn_value = find_pn(part)
+
+        if pn_value is None:
+            print("Skipping part with no part number: id={}".format(part.idProp))
+            continue
+
+        entry = pn_index.get(pn_value)
+
+        if entry is not None:
+            print("Already imported pn_value={}".format(pn_value))
+            continue
+
+        print("Importing {}".format(pn_value))
+
+        pns = bomFile.PartNumberList()
+
+        if pn_value is not None:
+            pns.add_part_number(bomFile.PartNumber(value=pn_value))
+
+        dpn_value = find_dpn(part, DIGIKEY_URI)
+        if dpn_value is not None:
+            pns.add_part_number(bomFile.PartNumber(value=dpn_value, distributor=DIGIKEY_URI))
+
+        if len(pns.part_numberProp) == 0:
+            continue
+
+        new_part = bomFile.Part(part_numbers=pns)
+        entry = Entry(True, new_part)
+        add_entry(entry)
+
+    if new_entry_added:
+        print("Imported {} entries".format(new_entry_added))
+        tmp_path = out_path + ".tmp"
+        with open(tmp_path, "w") as f:
+            out_file.export(f, 0, name_="bom-file")
+        os.rename(tmp_path, out_path)
+    else:
+        print("no new entries")