pyedb 0.42.0__py3-none-any.whl → 0.44.0__py3-none-any.whl

pyedb/configuration/cfg_operations.py

@@ -24,9 +24,38 @@ from pyedb.configuration.cfg_common import CfgBase
 
 
 class CfgCutout(CfgBase):
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def get_parameters_from_edb(self):
+            if "pyedb_cutout" in self._pedb.stackup.all_layers:
+                polygons = self._pedb.layout.find_primitive(layer_name="pyedb_cutout")
+                if polygons:
+                    poly = polygons[0]
+                    self.parent.custom_extent = poly.polygon_data.points
+                    net_names = []
+                    for name, obj in self._pedb.nets.nets.items():
+                        if obj.primitives:
+                            if obj.primitives[0].layer.name == "pyedb_cutout":
+                                continue
+                            else:
+                                net_names.append(name)
+                    self.parent.reference_list = []
+                    self.parent.signal_list = net_names
+                return self.parent.export_properties()
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, **kwargs):
         self._pedb = pedb
-
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.signal_list = kwargs.get("signal_list")
         self.reference_list = kwargs.get("reference_list")
         self.extent_type = kwargs.get("extent_type")
@@ -52,23 +81,7 @@ class CfgCutout(CfgBase):
         self.keep_lines_as_path = kwargs.get("keep_lines_as_path")
 
     def get_data_from_db(self):
-        if "pyedb_cutout" in self._pedb.stackup.all_layers:
-            polygons = self._pedb.layout.find_primitive(layer_name="pyedb_cutout")
-            if polygons:
-                poly = polygons[0]
-                self.custom_extent = poly.polygon_data.points
-
-                net_names = []
-                for name, obj in self._pedb.nets.nets.items():
-                    if obj.primitives:
-                        if obj.primitives[0].layer.name == "pyedb_cutout":
-                            continue
-                        else:
-                            net_names.append(name)
-
-                self.reference_list = []
-                self.signal_list = net_names
-            return self.export_properties()
+        return self.api.get_parameters_from_edb()
 
     def export_properties(self):
         return {
@@ -79,24 +92,43 @@ class CfgCutout(CfgBase):
 
 
 class CfgOperations(CfgBase):
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def apply_on_edb(self):
+            if self.parent.op_cutout:
+                polygon_points = self._pedb.cutout(**self.parent.op_cutout.get_attributes())
+                if "pyedb_cutout" not in self._pedb.stackup.all_layers:
+                    self._pedb.stackup.add_document_layer(name="pyedb_cutout")
+                    self._pedb.modeler.create_polygon(
+                        polygon_points, layer_name="pyedb_cutout", net_name="pyedb_cutout"
+                    )
+
+        def get_parameter_from_edb(self):
+            self.parent.op_cutout = CfgCutout(self._pedb)
+            data_from_db = self.parent.op_cutout.get_data_from_db()
+            if data_from_db:
+                return {"cutout": data_from_db}
+            else:
+                return {}
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
     def __init__(self, pedb, data):
         self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.op_cutout = CfgCutout(pedb, **data["cutout"]) if "cutout" in data else None
 
     def apply(self):
         """Imports operation information from JSON."""
-        if self.op_cutout:
-            polygon_points = self._pedb.cutout(**self.op_cutout.get_attributes())
-            if "pyedb_cutout" not in self._pedb.stackup.all_layers:
-                self._pedb.stackup.add_document_layer(name="pyedb_cutout")
-                self._pedb.modeler.create_polygon(polygon_points, layer_name="pyedb_cutout", net_name="pyedb_cutout")
-
-            # create a polygon on pyedb layer
+        self.api.apply_on_edb()
 
     def get_data_from_db(self):
-        self.op_cutout = CfgCutout(self._pedb)
-        data_from_db = self.op_cutout.get_data_from_db()
-        if data_from_db:
-            return {"cutout": data_from_db}
-        else:
-            return {}
+        return self.api.get_parameter_from_edb()

pyedb/configuration/cfg_package_definition.py

@@ -21,7 +21,6 @@
 # SOFTWARE.
 
 from pyedb.configuration.cfg_common import CfgBase
-from pyedb.dotnet.database.definition.package_def import PackageDef
 
 
 class CfgPackage(CfgBase):
@@ -64,61 +63,123 @@ class CfgHeatSink(CfgBase):
 
 
 class CfgPackageDefinitions:
+    class Grpc:
+        def __init__(self, parent):
+            self.parent = parent
+            self._pedb = parent._pedb
+
+        def set_parameter_to_edb(self):
+            from pyedb.grpc.database.definition.package_def import PackageDef
+
+            for pkg in self.parent.packages:
+                comp_def_from_db = self._pedb.definitions.component[pkg.component_definition]
+                if pkg.name in self._pedb.definitions.package:
+                    self._pedb.definitions.package[pkg.name].delete()
+
+                if pkg.extent_bounding_box:
+                    package_def = PackageDef(self._pedb, name=pkg.name, extent_bounding_box=pkg.extent_bounding_box)
+                else:
+                    package_def = PackageDef(self._pedb, name=pkg.name, component_part_name=pkg.component_definition)
+                pkg.set_attributes(package_def)
+
+                if pkg.heatsink:
+                    attrs = pkg.heatsink.get_attributes()
+                    for attr, value in attrs.items():
+                        package_def.set_heatsink(**attrs)
+
+                comp_list = dict()
+                if pkg.apply_to_all:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def_from_db.components.items()
+                            if refdes not in pkg.components
+                        }
+                    )
+                else:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def_from_db.components.items()
+                            if refdes in pkg.components
+                        }
+                    )
+                for _, i in comp_list.items():
+                    i.package_def = pkg.name
+
+        def get_parameter_from_edb(self):
+            package_definitions = []
+            for pkg_name, pkg_obj in self._pedb.definitions.package.items():
+                pkg = {}
+                pkg_attrs = [i for i in dir(pkg_obj) if not i.startswith("_")]
+                pkg_attrs = {i for i in pkg_attrs if i in CfgPackage().__dict__}
+                for pkg_attr_name in pkg_attrs:
+                    pkg[pkg_attr_name] = getattr(pkg_obj, pkg_attr_name)
+                hs_obj = pkg_obj.heatsink
+                if hs_obj:
+                    hs = {}
+                    hs_attrs = [i for i in dir(hs_obj) if not i.startswith("_")]
+                    hs_attrs = [i for i in hs_attrs if i in CfgHeatSink().__dict__]
+                    for hs_attr_name in hs_attrs:
+                        hs[hs_attr_name] = getattr(hs_obj, hs_attr_name)
+                    pkg["heatsink"] = hs
+                package_definitions.append(pkg)
+
+            return package_definitions
+
+    class DotNet(Grpc):
+        def __init__(self, parent):
+            super().__init__(parent)
+
+        def set_parameter_to_edb(self):
+            from pyedb.dotnet.database.definition.package_def import PackageDef
+
+            for pkg in self.parent.packages:
+                comp_def_from_db = self._pedb.definitions.component[pkg.component_definition]
+                if pkg.name in self._pedb.definitions.package:
+                    self._pedb.definitions.package[pkg.name].delete()
+
+                if pkg.extent_bounding_box:
+                    package_def = PackageDef(self._pedb, name=pkg.name, extent_bounding_box=pkg.extent_bounding_box)
+                else:
+                    package_def = PackageDef(self._pedb, name=pkg.name, component_part_name=pkg.component_definition)
+                pkg.set_attributes(package_def)
+
+                if pkg.heatsink:
+                    attrs = pkg.heatsink.get_attributes()
+                    for attr, value in attrs.items():
+                        package_def.set_heatsink(**attrs)
+
+                comp_list = dict()
+                if pkg.apply_to_all:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def_from_db.components.items()
+                            if refdes not in pkg.components
+                        }
+                    )
+                else:
+                    comp_list.update(
+                        {
+                            refdes: comp
+                            for refdes, comp in comp_def_from_db.components.items()
+                            if refdes in pkg.components
+                        }
+                    )
+                for _, i in comp_list.items():
+                    i.package_def = pkg.name
+
     def __init__(self, pedb, data):
         self._pedb = pedb
+        if self._pedb.grpc:
+            self.api = self.Grpc(self)
+        else:
+            self.api = self.DotNet(self)
         self.packages = [CfgPackage(**package) for package in data]
 
     def apply(self):
-        for pkg in self.packages:
-            comp_def_from_db = self._pedb.definitions.component[pkg.component_definition]
-            if pkg.name in self._pedb.definitions.package:
-                self._pedb.definitions.package[pkg.name].delete()
-
-            if pkg.extent_bounding_box:
-                package_def = PackageDef(self._pedb, name=pkg.name, extent_bounding_box=pkg.extent_bounding_box)
-            else:
-                package_def = PackageDef(self._pedb, name=pkg.name, component_part_name=pkg.component_definition)
-            pkg.set_attributes(package_def)
-
-            if pkg.heatsink:
-                attrs = pkg.heatsink.get_attributes()
-                for attr, value in attrs.items():
-                    package_def.set_heatsink(**attrs)
-
-            comp_list = dict()
-            if pkg.apply_to_all:
-                comp_list.update(
-                    {
-                        refdes: comp
-                        for refdes, comp in comp_def_from_db.components.items()
-                        if refdes not in pkg.components
-                    }
-                )
-            else:
-                comp_list.update(
-                    {refdes: comp for refdes, comp in comp_def_from_db.components.items() if refdes in pkg.components}
-                )
-            for _, i in comp_list.items():
-                i.package_def = pkg.name
+        self.api.set_parameter_to_edb()
 
     def get_data_from_db(self):
-        package_definitions = []
-
-        for pkg_name, pkg_obj in self._pedb.definitions.package.items():
-            pkg = {}
-            pkg_attrs = [i for i in dir(pkg_obj) if not i.startswith("_")]
-            pkg_attrs = {i for i in pkg_attrs if i in CfgPackage().__dict__}
-            for pkg_attr_name in pkg_attrs:
-                pkg[pkg_attr_name] = getattr(pkg_obj, pkg_attr_name)
-
-            hs_obj = pkg_obj.heatsink
-            if hs_obj:
-                hs = {}
-                hs_attrs = [i for i in dir(hs_obj) if not i.startswith("_")]
-                hs_attrs = [i for i in hs_attrs if i in CfgHeatSink().__dict__]
-                for hs_attr_name in hs_attrs:
-                    hs[hs_attr_name] = getattr(hs_obj, hs_attr_name)
-                pkg["heatsink"] = hs
-            package_definitions.append(pkg)
-
-        return package_definitions
+        return self.api.get_parameter_from_edb()