diff options
Diffstat (limited to 'src/ee/kicad/__init__.py')
| -rw-r--r-- | src/ee/kicad/__init__.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/ee/kicad/__init__.py b/src/ee/kicad/__init__.py index 5187a9c..d2d48da 100644 --- a/src/ee/kicad/__init__.py +++ b/src/ee/kicad/__init__.py @@ -29,6 +29,7 @@ def to_pandas(obj: Any, **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): fields = { "ref": c.ref, @@ -37,7 +38,7 @@ def to_pandas(obj: Any, **kwarg): "value": c.value, "footprint": c.footprint, } - fields.update({f.name:f.value for f in c.fields if f.is_custom}) + fields.update({f.name: f.value for f in c.fields if f.is_custom}) return fields components = sch.components @@ -56,8 +57,8 @@ def to_pandas(obj: Any, **kwarg): 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=columns).\ - set_index("ref").\ + return pandas.DataFrame(data=data, columns=columns). \ + set_index("ref"). \ sort_values(["ref_type", "ref_num"]) if isinstance(obj, Schematic): |