wip.

o Moving to more standard doit task_xxx methods.

2 files changed, 51 insertions, 63 deletions

diff --git a/src/ee/kicad/doit.py b/src/ee/kicad/doit.py
index 29438ad..577eb21 100644
--- a/src/ee/kicad/doit.py
+++ b/src/ee/kicad/doit.py
@@ -3,59 +3,47 @@ import os.path
 
 from configclass import Config
 
+import ee.kicad
+import ee.kicad.pcb
 from ee.fact import DataSetManager
 
 logger = logging.getLogger(__name__)
 
+_config_template = Config({
+    "sch": None,
+    "kicad_pcb": None,
+    "gerber_dir": None,
+    "gerber_zip": None,
+    "data_set_dir": None,
+})
 
-class KicadDoitTasks(object):
-    config = Config({
-        "sch": None,
-        "kicad_pcb": None,
-        "gerber_dir": None,
-        "gerber_zip": None,
-        "data_set_dir": None,
-    })
+_config = None
 
-    def __init__(self, *args, **kwargs):
-        self.config = self.config.make(kwargs)
+_dsm = None  # type: DataSetManager
 
-        formatter = logging.Formatter("%(levelname)s: %(message)s")
-        ch = logging.StreamHandler()
-        ch.setFormatter(formatter)
-        logger.addHandler(ch)
 
-        logger.setLevel(logging.DEBUG)
+def init(**kwargs):
+    global _config
+    _config = _config_template.make(kwargs)
 
-    def tasks(self, *args, **kwargs):
-        kicad_pcb = self.config["kicad_pcb"]
-        sch = self.config["sch"]
-        tasks = []
+    formatter = logging.Formatter("%(levelname)s: %(message)s")
+    ch = logging.StreamHandler()
+    ch.setFormatter(formatter)
+    logger.addHandler(ch)
 
-        dsm = DataSetManager(self.config["data_set_dir"])
+    logger.setLevel(logging.DEBUG)
 
-        gerber_dir = self.config["gerber_dir"]
-        gerber_zip = self.config["gerber_zip"]
-        if kicad_pcb and gerber_dir:
-            tasks.append(task_kicad_gerber(kicad_pcb, gerber_dir, gerber_zip))
+    global _dsm
+    _dsm = DataSetManager(_config["data_set_dir"])
 
-        sch_ds = task_kicad_sch_to_data_set(dsm, sch) \
-            if sch else None
 
-        pcb_ds = task_kicad_pcb_to_data_set(dsm, kicad_pcb) \
-            if kicad_pcb else None
-
-        component_ds = task_kicad_create_component_data_set(dsm) \
-            if sch_ds else None
-
-        tasks = [component_ds, sch_ds, pcb_ds,]
-        return tasks
-
-
-def task_kicad_gerber(kicad_pcb: str, gerber_dir: str, gerber_zip: str, name="kicad-gerber"):
-    import ee.kicad
+def task_kicad_gerber():
+    kicad_pcb = _config["kicad_pcb"]
+    gerber_dir = _config["gerber_dir"]
+    gerber_zip = _config["gerber_zip"]
 
     gerber_zip = gerber_zip or "{}.zip".format(gerber_dir)
+
     # logger.info("gerber_zip={}".format(gerber_zip))
 
     eg = next((p for p in (os.path.join(p, "export_gerber.py") for p in ee.kicad.__path__) if os.path.isfile(p)), None)
@@ -81,22 +69,23 @@ def task_kicad_gerber(kicad_pcb: str, gerber_dir: str, gerber_zip: str, name="ki
                 z.write(p, arcname=p.relative_to(gerber_dir))
 
     return {
-        "name": name,
+        "basename": "kicad-gerber",
         "targets": [gerber_zip],
         "actions": [mkdir, export_gerber, make_zip],
         "file_dep": [kicad_pcb],
     }
 
 
-def task_kicad_sch_to_data_set(dsm: DataSetManager, sch, name="kicad-sch-to-data-set"):
+def task_kicad_sch_to_data_set():
     out_data_set = "kicad-sch"
     in_data_sets = []
 
+    sch = _config["sch"]
+
     def action():
-        import ee.kicad
         from ee.kicad.model import ComponentField
 
-        with dsm.create_rw(out_data_set, inputs=in_data_sets) as ds:
+        with _dsm.create_rw(out_data_set, inputs=in_data_sets) as ds:
             schematics = ee.kicad.read_schematics(sch)
             for c in [c for c in schematics.components]:
                 o = ds.get_object("kicad-schematic-component", c.timestamp)
@@ -113,33 +102,32 @@ def task_kicad_sch_to_data_set(dsm: DataSetManager, sch, name="kicad-sch-to-data
                         o.set(f.name, str(f.value))
 
     return {
-        "name": name,
-        "file_dep": [sch] + [dsm.metafile_for_ds(ds) for ds in in_data_sets],
+        "basename": "kicad-sch-to-data-set",
+        "file_dep": [sch] + [_dsm.metafile_for_ds(ds) for ds in in_data_sets],
         "actions": [action],
-        "targets": [dsm.metafile_for_ds(out_data_set)],
+        "targets": [_dsm.metafile_for_ds(out_data_set)],
     }
 
 
-def task_kicad_pcb_to_data_set(dsm: DataSetManager, pcb_path, name="kicad-pcb-to-data-set"):
+def task_kicad_pcb_to_data_set():
+    kicad_pcb = _config["kicad_pcb"]
+
     out_data_set = "kicad-pcb"
     in_data_sets = []
 
     def action():
-        import ee.kicad.pcb
-        from ee.kicad.pcb import FpText
-
-        logger.debug("Parsing PCB {}".format(pcb_path))
+        logger.debug("Parsing PCB {}".format(kicad_pcb))
 
-        with dsm.create_rw(out_data_set, inputs=in_data_sets) as ds:
+        with _dsm.create_rw(out_data_set, inputs=in_data_sets) as ds:
             # [ds.delete(o) for o in ds.items(object_type="kicad-pcb-component")]
 
-            pcb: ee.kicad.pcb.KicadPcb = ee.kicad.pcb.parse(pcb_path)
+            pcb = ee.kicad.pcb.parse(kicad_pcb)  # type: ee.kicad.pcb.KicadPcb
             for _m in pcb.modules:
-                m: ee.kicad.pcb.Module = _m
+                m = _m  # type: ee.kicad.pcb.Module
 
                 o = ds.get_object("kicad-pcb-component", m.tstamp)
 
-                ref_text: FpText = next((t for t in m.fp_texts if t.kind == "reference"), None)
+                ref_text = next((t for t in m.fp_texts if t.kind == "reference"), None)
                 o.set("reference", ref_text.value)
 
                 x, y, rot = m.at
@@ -149,29 +137,29 @@ def task_kicad_pcb_to_data_set(dsm: DataSetManager, pcb_path, name="kicad-pcb-to
                 o.set("layer", m.layer)
 
     return {
-        "name": name,
-        "file_dep": [pcb_path] + [dsm.metafile_for_ds(ds) for ds in in_data_sets],
+        "basename": "kicad-pcb-to-data-set",
+        "file_dep": [kicad_pcb] + [_dsm.metafile_for_ds(ds) for ds in in_data_sets],
         "actions": [action],
-        "targets": [dsm.metafile_for_ds(out_data_set)],
+        "targets": [_dsm.metafile_for_ds(out_data_set)],
     }
 
 
-def task_kicad_create_component_data_set(dsm: DataSetManager, name="kicad-create-component-data-set"):
+def task_kicad_create_component_data_set():
     out_data_set = "components"
     in_data_sets = ["kicad-sch", "kicad-pcb"]
 
     def action(targets, *args, **kwargs):
         logger.info("targets={}, args={}, kwargs={}".format(targets, args, kwargs))
 
-        with dsm.create_rw(out_data_set, inputs=in_data_sets) as ds:
+        with _dsm.create_rw(out_data_set, inputs=in_data_sets) as ds:
             items = ds.items()
             logger.info("Got {} objects".format(len(items)))
             for o in items:
                 logger.info("processing {}".format(o.key))
 
     return {
-        "name": name,
-        "file_dep": [dsm.metafile_for_ds(ds) for ds in in_data_sets],
+        "basename": "kicad-create-component-data-set",
+        "file_dep": [_dsm.metafile_for_ds(ds) for ds in in_data_sets],
         "actions": [action],
-        "targets": [dsm.metafile_for_ds(out_data_set)],
+        "targets": [_dsm.metafile_for_ds(out_data_set)],
     }
diff --git a/src/ee/kicad/export_gerber.py b/src/ee/kicad/export_gerber.py
index 511f9ee..c4c72fd 100755
--- a/src/ee/kicad/export_gerber.py
+++ b/src/ee/kicad/export_gerber.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python2
 from __future__ import print_function
 import sys
 import os