From a00dcfc7f6d12442fca8c5b7d43f12b707f0c877 Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Tue, 9 Apr 2019 06:03:33 +0200 Subject: o Renaming part-create-distributor-search-list to pn-part-search-list. Making the search more narrow, it is only responsible to find parts based on MPN or SPN. --- src/ee/part/pn_part_search_list.py | 44 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 src/ee/part/pn_part_search_list.py (limited to 'src/ee/part/pn_part_search_list.py') diff --git a/src/ee/part/pn_part_search_list.py b/src/ee/part/pn_part_search_list.py new file mode 100644 index 0000000..049d0b8 --- /dev/null +++ b/src/ee/part/pn_part_search_list.py @@ -0,0 +1,44 @@ +from pathlib import Path + +from ee.part import PartDb, load_db, save_db, Part, fact_keys +from ee.xml import types + +__all__ = ["pn_part_search_list"] + +ignored_part_classes = [ + "mechanical" +] + + +def pn_part_search_list(in_path: Path, out_path: Path, supplier: str): + in_parts = load_db(in_path) + out_parts = PartDb() + + # print("loaded {} existing parts".format(in_parts.size())) + + for xml in in_parts.iterparts(): + part = Part(xml) + pn_value = next((p.valueProp for p in part.get_mpns()), None) + + part_class = part.find_fact(fact_keys.part_class) + if part_class: + if part_class.valueProp in ignored_part_classes: + continue + + if pn_value is None: + refs = [ref.referenceProp for ref in part.get_schematic_references()] + print("Skipping part with no part number: schematic reference: {}".format(", ".join(refs))) + continue + + entry = out_parts.find_by_pn(pn_value) + + if entry is not None: + continue + + new_part = types.Part() + new_part.referencesProp = xml.referencesProp + + out_parts.add_entry(new_part, True) + + # print("Saving {} work parts".format(out_parts.size())) + save_db(out_path, out_parts) -- cgit v1.2.3