aboutsummaryrefslogtreecommitdiff
path: root/src/ee
diff options
context:
space:
mode:
Diffstat (limited to 'src/ee')
-rw-r--r--src/ee/tools/__init__.py21
-rw-r--r--src/ee/tools/digikey_download_facts.py7
-rw-r--r--src/ee/tools/kicad_make_bom.py37
3 files changed, 60 insertions, 5 deletions
diff --git a/src/ee/tools/__init__.py b/src/ee/tools/__init__.py
index e69de29..893ee31 100644
--- a/src/ee/tools/__init__.py
+++ b/src/ee/tools/__init__.py
@@ -0,0 +1,21 @@
+import os.path
+
+
+def _mkdir_and_open(path):
+ dirname = os.path.dirname(path)
+
+ if len(dirname) > 0:
+ if not os.path.isdir(dirname):
+ os.mkdir(dirname)
+
+ return open(path, "w")
+
+
+def mk_parents(path: str):
+ dirname = os.path.dirname(path)
+
+ if len(dirname) == 0:
+ return
+
+ if not os.path.isdir(dirname):
+ os.mkdir(dirname)
diff --git a/src/ee/tools/digikey_download_facts.py b/src/ee/tools/digikey_download_facts.py
index 08a8029..c97ba9d 100644
--- a/src/ee/tools/digikey_download_facts.py
+++ b/src/ee/tools/digikey_download_facts.py
@@ -7,6 +7,7 @@ from colors import color
import ee.digikey as dk
from ee.digikey import SearchResponseTypes, DigikeyProduct
+from ee.tools import mk_parents
parser = argparse.ArgumentParser(description="Download facts about parts from Digi-Key")
@@ -41,11 +42,7 @@ def on_product(product: DigikeyProduct):
y = product.to_yaml()
filename = mpn_to_path(product.mpn)
- dirname = os.path.dirname(filename)
-
- if not os.path.isdir(dirname):
- os.mkdir(dirname)
-
+ mk_parents(filename)
with open(filename, "w") as f:
yaml.dump(y, f, encoding="utf-8", allow_unicode=True)
diff --git a/src/ee/tools/kicad_make_bom.py b/src/ee/tools/kicad_make_bom.py
new file mode 100644
index 0000000..518cf75
--- /dev/null
+++ b/src/ee/tools/kicad_make_bom.py
@@ -0,0 +1,37 @@
+import argparse
+import ee.kicad as kicad
+from ee.tools import mk_parents
+from xml.etree import ElementTree
+from xml.dom import minidom
+
+parser = argparse.ArgumentParser(description="Create a bom XML file from a KiCAD schematic")
+
+parser.add_argument("--sch",
+ required=True,
+ metavar="SCH",
+ help="The schematic to read")
+
+parser.add_argument("--out",
+ metavar="FILE",
+ help="The output file")
+
+parser.add_argument("--pretty",
+ action='store_true',
+ help="Pretty print the XML")
+
+args = parser.parse_args()
+
+sch = kicad.read_schematic(args.sch)
+
+bom = kicad.to_bom(sch)
+xml = ElementTree.tostring(bom, encoding='unicode')
+
+if args.pretty:
+ xml = minidom.parseString(xml).toprettyxml(indent=" ")
+
+if args.out:
+ mk_parents(args.out)
+ with open(args.out, "w") as f:
+ f.write(xml)
+else:
+ print(xml)