diff options
Diffstat (limited to 'src/ee/kicad/__init__.py')
| -rw-r--r-- | src/ee/kicad/__init__.py | 24 |
1 files changed, 18 insertions, 6 deletions
diff --git a/src/ee/kicad/__init__.py b/src/ee/kicad/__init__.py index b270ba1..adf8ddd 100644 --- a/src/ee/kicad/__init__.py +++ b/src/ee/kicad/__init__.py @@ -25,27 +25,39 @@ def to_pandas(obj: Any, **kwarg): return False return True - def to_pandas_schematic(sch: Schematic, **kwarg): + def to_pandas_schematic(sch: Schematic): + # These fields will always be put first. + special_fields = ["ref", "ref_type", "ref_num", "value"] def make_dict(c: Component): - return { + fields = { "ref": c.ref, "ref_type": c.ref_type, "ref_num": c.ref_num, - "value": c.value + "value": c.value, + "footprint": c.footprint, } + fields.update({f.name:f.value for f in c.fields if f.is_custom}) + return fields components = sch.components filters = [] - if not kwarg.get("include_pwr", False): + include_pwr = kwarg.get("include_pwr", False) + if not include_pwr: filters.append(lambda c: not c.is_pwr) + include_flg = kwarg.get("include_flg", False) + if not include_flg: + filters.append(lambda c: c.ref_type != "#FLG") + data = [make_dict(c) for c in components if run_filter(filters, c)] + columns = set([key for row in data for key in list(row)]) - set(special_fields) + columns = special_fields + list(columns) - return pandas.DataFrame(data=data, columns=["ref", "ref_type", "ref_num", "value"]).\ + return pandas.DataFrame(data=data, columns=columns).\ set_index("ref").\ - sort_index() + sort_values(["ref_type", "ref_num"]) if isinstance(obj, Schematic): return to_pandas_schematic(obj) |