import argparse import sys import yaml from pathlib import Path from ee.part import PartDb, save_db from ee.xml import types def import_parts_yaml(in_file: Path, out_dir: Path): with in_file.open("r") as f: doc = yaml.safe_load(f) if not isinstance(doc, list): print("Bad YAML document, the root must be a list", file=sys.stderr) parts = PartDb() for item in doc: if not isinstance(item, dict): print("Bad YAML document, each part must be a dict", file=sys.stderr) return part_number_list = types.PartNumberList() mpn = item.get("mpn") assert isinstance(mpn, str) part_number_list.part_number.append(types.PartNumber(value=mpn)) id_ = mpn part = types.Part(id=id_) part.part_numbersProp = part_number_list parts.add_entry(part, True) print("Imported {} parts".format(parts.size())) save_db(out_dir, parts) parser = argparse.ArgumentParser() parser.add_argument("--in", dest="in_path", required=True, metavar="PARTS_YAML") parser.add_argument("--out", required=True, metavar="PART DB") args = parser.parse_args() import_parts_yaml(Path(args.in_path), Path(args.out))