osi-dump 0.1.3.3.4.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

osi_dump/exporter/role_assignment/role_assignment_exporter.py

@@ -1,7 +1,7 @@
-from abc import ABC, abstractmethod
-
-
-class RoleAssignmentExporter(ABC):
-    @abstractmethod
-    def export_role_assignments(self, role_assignments, output_file: str):
-        pass
+from abc import ABC, abstractmethod
+from osi_dump.importer.role_assignment.role_assignment_importer import RoleAssignmentImporter
+
+class RoleAssignmentExporter(ABC):
+    @abstractmethod
+    def export_role_assignments(self, importer: RoleAssignmentImporter):
+        pass

osi_dump/exporter/router/excel_router_exporter.py

@@ -1,30 +1,30 @@
-import pandas as pd
-
-import logging
-
-from openpyxl import load_workbook
-
-from osi_dump import util
-from osi_dump.exporter.router.router_exporter import RouterExporter
-
-from osi_dump.model.router import Router
-
-logger = logging.getLogger(__name__)
-
-
-class ExcelRouterExporter(RouterExporter):
-    def __init__(self, sheet_name: str, output_file: str):
-        self.sheet_name = sheet_name
-        self.output_file = output_file
-
-    def export_routers(self, routers: list[Router]):
-
-        df = pd.DataFrame([router.model_dump() for router in routers])
-
-        logger.info(f"Exporting routers for {self.sheet_name}")
-        try:
-            util.export_data_excel(self.output_file, sheet_name=self.sheet_name, df=df)
-
-            logger.info(f"Exported routers for {self.sheet_name}")
-        except Exception as e:
-            logger.warning(f"Exporting routers for {self.sheet_name} error: {e}")
+import pandas as pd
+
+import logging
+
+from openpyxl import load_workbook
+
+from osi_dump import util
+from osi_dump.exporter.router.router_exporter import RouterExporter
+
+from osi_dump.model.router import Router
+
+logger = logging.getLogger(__name__)
+
+
+class ExcelRouterExporter(RouterExporter):
+    def __init__(self, sheet_name: str, output_file: str):
+        self.sheet_name = sheet_name
+        self.output_file = output_file
+
+    def export_routers(self, routers: list[Router]):
+
+        df = pd.DataFrame([router.model_dump() for router in routers])
+
+        logger.info(f"Exporting routers for {self.sheet_name}")
+        try:
+            util.export_data_excel(self.output_file, sheet_name=self.sheet_name, df=df)
+
+            logger.info(f"Exported routers for {self.sheet_name}")
+        except Exception as e:
+            logger.warning(f"Exporting routers for {self.sheet_name} error: {e}")

osi_dump/exporter/router/router_exporter.py

@@ -1,7 +1,7 @@
-from abc import ABC, abstractmethod
-
-
-class RouterExporter(ABC):
-    @abstractmethod
-    def export_routers(self, routers, output_file: str):
-        pass
+from abc import ABC, abstractmethod
+
+
+class RouterExporter(ABC):
+    @abstractmethod
+    def export_routers(self, routers, output_file: str):
+        pass

osi_dump/exporter/security_group/excel_security_group_exporter.py

@@ -0,0 +1,35 @@
+import logging
+import pandas as pd
+from typing import Generator
+from osi_dump import util
+from osi_dump.exporter.security_group.security_group_exporter import SecurityGroupExporter
+from osi_dump.model.security_group import SecurityGroup
+
+logger = logging.getLogger(__name__)
+
+class ExcelSecurityGroupExporter(SecurityGroupExporter):
+    def __init__(self, sheet_name: str, output_file: str):
+        self.sheet_name = sheet_name
+        self.output_file = output_file
+
+    def export_security_groups(self, security_groups: Generator[SecurityGroup, None, None]):
+        
+        data_generator = (sg.model_dump() for sg in security_groups)
+
+        df = pd.json_normalize(
+            data_generator,
+            record_path='rules',
+            meta=['security_group_id', 'name', 'project_id', 'description'],
+            record_prefix='rule.'
+        )
+
+        if df.empty:
+            logger.info(f"No security groups to export for {self.sheet_name}")
+            return
+            
+        logger.info(f"Exporting security groups for {self.sheet_name}")
+        try:
+            util.export_data_excel(self.output_file, sheet_name=self.sheet_name, df=df)
+            logger.info(f"Exported security groups for {self.sheet_name}")
+        except Exception as e:
+            logger.warning(f"Exporting security groups for {self.sheet_name} error: {e}")

osi_dump/exporter/security_group/security_group_exporter.py

@@ -0,0 +1,8 @@
+from abc import ABC, abstractmethod
+from typing import Generator
+from osi_dump.model.security_group import SecurityGroup
+
+class SecurityGroupExporter(ABC):
+    @abstractmethod
+    def export_security_groups(self, security_groups: Generator[SecurityGroup, None, None]):
+        pass

osi_dump/exporter/volume/excel_volume_exporter.py

@@ -1,29 +1,29 @@
-import pandas as pd
-
-import logging
-
-
-from osi_dump import util
-
-from osi_dump.exporter.volume.volume_exporter import VolumeExporter
-
-from osi_dump.model.volume import Volume
-
-logger = logging.getLogger(__name__)
-
-
-class ExcelVolumeExporter(VolumeExporter):
-    def __init__(self, sheet_name: str, output_file: str):
-        self.sheet_name = sheet_name
-        self.output_file = output_file
-
-    def export_volumes(self, volumes: list[Volume]):
-        df = pd.DataFrame([volume.model_dump() for volume in volumes])
-
-        logger.info(f"Exporting volumes for {self.sheet_name}")
-        try:
-            util.export_data_excel(self.output_file, sheet_name=self.sheet_name, df=df)
-
-            logger.info(f"Exported volumes for {self.sheet_name}")
-        except Exception as e:
-            logger.warning(f"Exporting volumes for {self.sheet_name} error: {e}")
+import pandas as pd
+
+import logging
+
+
+from osi_dump import util
+
+from osi_dump.exporter.volume.volume_exporter import VolumeExporter
+
+from osi_dump.model.volume import Volume
+
+logger = logging.getLogger(__name__)
+
+
+class ExcelVolumeExporter(VolumeExporter):
+    def __init__(self, sheet_name: str, output_file: str):
+        self.sheet_name = sheet_name
+        self.output_file = output_file
+
+    def export_volumes(self, volumes: list[Volume]):
+        df = pd.DataFrame([volume.model_dump() for volume in volumes])
+
+        logger.info(f"Exporting volumes for {self.sheet_name}")
+        try:
+            util.export_data_excel(self.output_file, sheet_name=self.sheet_name, df=df)
+
+            logger.info(f"Exported volumes for {self.sheet_name}")
+        except Exception as e:
+            logger.warning(f"Exporting volumes for {self.sheet_name} error: {e}")

osi_dump/exporter/volume/volume_exporter.py

@@ -1,7 +1,7 @@
-from abc import ABC, abstractmethod
-
-
-class VolumeExporter(ABC):
-    @abstractmethod
-    def export_volumes(self, volumes, output_file: str):
-        pass
+from abc import ABC, abstractmethod
+
+
+class VolumeExporter(ABC):
+    @abstractmethod
+    def export_volumes(self, volumes, output_file: str):
+        pass

osi_dump/importer/external_port/external_port_importer.py

@@ -1,9 +1,9 @@
-from abc import ABC, abstractmethod
-
-from osi_dump.model.external_port import ExternalPort
-
-
-class ExternalPortImporter(ABC):
-    @abstractmethod
-    def import_external_ports(self) -> list[ExternalPort]:
-        pass
+from abc import ABC, abstractmethod
+
+from osi_dump.model.external_port import ExternalPort
+
+
+class ExternalPortImporter(ABC):
+    @abstractmethod
+    def import_external_ports(self) -> list[ExternalPort]:
+        pass

osi_dump/importer/external_port/openstack_external_port_importer.py

@@ -1,177 +1,177 @@
-import logging
-
-import ipaddress
-
-import concurrent
-
-import numpy as np
-
-from openstack.connection import Connection
-from openstack.network.v2.port import Port as OSPort
-from openstack.network.v2.subnet import Subnet as OSSubnet
-from openstack.network.v2.floating_ip import FloatingIP as OSFloatingIP
-from openstack.network.v2.router import Router as OSRouter
-from openstack.network.v2.network import Network as OSNetwork
-
-from osi_dump.importer.external_port.external_port_importer import ExternalPortImporter
-from osi_dump.model.external_port import ExternalPort
-
-import osi_dump.api.neutron as osi_neutron
-
-logger = logging.getLogger(__name__)
-
-
-class OpenStackExternalPortImporter(ExternalPortImporter):
-    def __init__(self, connection: Connection):
-        self.connection = connection
-
-    def _get_external_ports(self) -> list[OSPort]:
-
-        # Get all external networks
-        networks = self.connection.network.networks(
-            is_router_external=True,
-        )
-
-        external_ports = []
-
-        for network in networks:
-            ex_ports = list(self.connection.network.ports(network_id=network.id))
-
-            for ex_port in ex_ports:
-                external_ports.append(ex_port)
-
-        # Get all port on those networks
-
-        return external_ports
-
-    def import_external_ports(self) -> list[ExternalPort]:
-        """Import external_ports information from Openstack
-
-
-        Raises:
-            Exception: Raises exception if fetching external_port failed
-
-        Returns:
-            list[Instance]: _description_
-        """
-
-        logger.info(f"Importing external_ports for {self.connection.auth['auth_url']}")
-
-        try:
-            os_ports: list[OSPort] = self._get_external_ports()
-        except Exception as e:
-            raise Exception(
-                f"Can not fetch external_ports for {self.connection.auth['auth_url']} {e}"
-            ) from e
-
-        external_ports: list[ExternalPort] = []
-
-        with concurrent.futures.ThreadPoolExecutor() as executor:
-            futures = [
-                executor.submit(self._get_external_port_info, external_port)
-                for external_port in os_ports
-            ]
-            for future in concurrent.futures.as_completed(futures):
-                external_ports.append(future.result())
-
-        logger.info(f"Imported external_ports for {self.connection.auth['auth_url']}")
-
-        external_ports = sorted(
-            external_ports,
-            key=lambda external_port: (
-                external_port.network_id,
-                (
-                    ipaddress.ip_address(external_port.ip_address)
-                    if external_port.ip_address
-                    else ipaddress.ip_address("0.0.0.0")
-                ),
-            ),
-        )
-
-        return external_ports
-
-    def _get_external_port_info(self, external_port: OSPort) -> ExternalPort:
-
-        subnet_id = None
-        ip_address = None
-        subnet_cidr = None
-
-        try:
-            ip_address = external_port.fixed_ips[0]["ip_address"]
-        except Exception as e:
-            logger.warning(f"No IP address found for port {external_port.id}")
-
-        try:
-            subnet_id = external_port.fixed_ips[0]["subnet_id"]
-        except Exception as e:
-            logger.warning(f"No subnet ID found for {external_port.id}")
-
-        try:
-            subnet: OSSubnet = self.connection.get_subnet_by_id(subnet_id)
-
-            subnet_cidr = subnet.cidr
-        except Exception as e:
-            logger.warning(f"No subnet cidr found for port {external_port.id}")
-
-        mapped_device_project = ["network:floatingip", "network:router_gateway"]
-
-        project_id = (
-            external_port.project_id
-            if external_port.device_owner not in mapped_device_project
-            else self._map_project_id(
-                device_owner=external_port.device_owner,
-                device_id=external_port.device_id,
-            )
-        )
-        
-        network: OSNetwork = self.connection.get_network(name_or_id=external_port.network_id)
-
-        vlan_id = None
-        
-        try: 
-            if not network.segments: 
-                vlan_id = network.provider_segmentation_id
-            else: 
-                logger.warning(f"Multiple segments mapped for net id {network.id}")
-        except Exception as e: 
-            logger.warning(f"Can not get vlan ID for external port {external_port.id}, net id {network.id}")
-        
-        external_port_ret = ExternalPort(
-            port_id=external_port.id,
-            project_id=project_id,
-            network_id=external_port.network_id,
-            network_name=network.name,
-            subnet_id=subnet_id,
-            subnet_cidr=subnet_cidr,
-            ip_address=ip_address,
-            allowed_address_pairs=external_port.allowed_address_pairs,
-            device_id=external_port.device_id,
-            device_owner=external_port.device_owner,
-            status=external_port.status,
-            vlan_id=vlan_id
-        )
-
-        return external_port_ret
-
-    def _map_project_id(self, device_owner: str, device_id: str) -> str:
-
-        if device_owner == "network:router_gateway":
-            return self._map_project_id_router(device_id=device_id)
-        elif device_owner == "network:floatingip":
-            return self._map_project_id_floating(device_id=device_id)
-
-        return ""
-
-    def _map_project_id_router(self, device_id: str) -> str:
-        project_id = osi_neutron.get_router_project(
-            self.connection, router_id=device_id
-        )
-
-        return project_id
-
-    def _map_project_id_floating(self, device_id: str) -> str:
-        project_id = osi_neutron.get_floating_ip_project(
-            self.connection, floating_ip_id=device_id
-        )
-
-        return project_id
+import logging
+
+import ipaddress
+
+import concurrent
+
+import numpy as np
+
+from openstack.connection import Connection
+from openstack.network.v2.port import Port as OSPort
+from openstack.network.v2.subnet import Subnet as OSSubnet
+from openstack.network.v2.floating_ip import FloatingIP as OSFloatingIP
+from openstack.network.v2.router import Router as OSRouter
+from openstack.network.v2.network import Network as OSNetwork
+
+from osi_dump.importer.external_port.external_port_importer import ExternalPortImporter
+from osi_dump.model.external_port import ExternalPort
+
+import osi_dump.api.neutron as osi_neutron
+
+logger = logging.getLogger(__name__)
+
+
+class OpenStackExternalPortImporter(ExternalPortImporter):
+    def __init__(self, connection: Connection):
+        self.connection = connection
+
+    def _get_external_ports(self) -> list[OSPort]:
+
+        # Get all external networks
+        networks = self.connection.network.networks(
+            is_router_external=True,
+        )
+
+        external_ports = []
+
+        for network in networks:
+            ex_ports = list(self.connection.network.ports(network_id=network.id))
+
+            for ex_port in ex_ports:
+                external_ports.append(ex_port)
+
+        # Get all port on those networks
+
+        return external_ports
+
+    def import_external_ports(self) -> list[ExternalPort]:
+        """Import external_ports information from Openstack
+
+
+        Raises:
+            Exception: Raises exception if fetching external_port failed
+
+        Returns:
+            list[Instance]: _description_
+        """
+
+        logger.info(f"Importing external_ports for {self.connection.auth['auth_url']}")
+
+        try:
+            os_ports: list[OSPort] = self._get_external_ports()
+        except Exception as e:
+            raise Exception(
+                f"Can not fetch external_ports for {self.connection.auth['auth_url']} {e}"
+            ) from e
+
+        external_ports: list[ExternalPort] = []
+
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            futures = [
+                executor.submit(self._get_external_port_info, external_port)
+                for external_port in os_ports
+            ]
+            for future in concurrent.futures.as_completed(futures):
+                external_ports.append(future.result())
+
+        logger.info(f"Imported external_ports for {self.connection.auth['auth_url']}")
+
+        external_ports = sorted(
+            external_ports,
+            key=lambda external_port: (
+                external_port.network_id,
+                (
+                    ipaddress.ip_address(external_port.ip_address)
+                    if external_port.ip_address
+                    else ipaddress.ip_address("0.0.0.0")
+                ),
+            ),
+        )
+
+        return external_ports
+
+    def _get_external_port_info(self, external_port: OSPort) -> ExternalPort:
+
+        subnet_id = None
+        ip_address = None
+        subnet_cidr = None
+
+        try:
+            ip_address = external_port.fixed_ips[0]["ip_address"]
+        except Exception as e:
+            logger.warning(f"No IP address found for port {external_port.id}")
+
+        try:
+            subnet_id = external_port.fixed_ips[0]["subnet_id"]
+        except Exception as e:
+            logger.warning(f"No subnet ID found for {external_port.id}")
+
+        try:
+            subnet: OSSubnet = self.connection.get_subnet_by_id(subnet_id)
+
+            subnet_cidr = subnet.cidr
+        except Exception as e:
+            logger.warning(f"No subnet cidr found for port {external_port.id}")
+
+        mapped_device_project = ["network:floatingip", "network:router_gateway"]
+
+        project_id = (
+            external_port.project_id
+            if external_port.device_owner not in mapped_device_project
+            else self._map_project_id(
+                device_owner=external_port.device_owner,
+                device_id=external_port.device_id,
+            )
+        )
+        
+        network: OSNetwork = self.connection.get_network(name_or_id=external_port.network_id)
+
+        vlan_id = None
+        
+        try: 
+            if not network.segments: 
+                vlan_id = network.provider_segmentation_id
+            else: 
+                logger.warning(f"Multiple segments mapped for net id {network.id}")
+        except Exception as e: 
+            logger.warning(f"Can not get vlan ID for external port {external_port.id}, net id {network.id}")
+        
+        external_port_ret = ExternalPort(
+            port_id=external_port.id,
+            project_id=project_id,
+            network_id=external_port.network_id,
+            network_name=network.name,
+            subnet_id=subnet_id,
+            subnet_cidr=subnet_cidr,
+            ip_address=ip_address,
+            allowed_address_pairs=external_port.allowed_address_pairs,
+            device_id=external_port.device_id,
+            device_owner=external_port.device_owner,
+            status=external_port.status,
+            vlan_id=vlan_id
+        )
+
+        return external_port_ret
+
+    def _map_project_id(self, device_owner: str, device_id: str) -> str:
+
+        if device_owner == "network:router_gateway":
+            return self._map_project_id_router(device_id=device_id)
+        elif device_owner == "network:floatingip":
+            return self._map_project_id_floating(device_id=device_id)
+
+        return ""
+
+    def _map_project_id_router(self, device_id: str) -> str:
+        project_id = osi_neutron.get_router_project(
+            self.connection, router_id=device_id
+        )
+
+        return project_id
+
+    def _map_project_id_floating(self, device_id: str) -> str:
+        project_id = osi_neutron.get_floating_ip_project(
+            self.connection, floating_ip_id=device_id
+        )
+
+        return project_id

osi_dump/importer/flavor/flavor_importer.py

@@ -1,9 +1,9 @@
-from abc import ABC, abstractmethod
-
-from osi_dump.model.flavor import Flavor
-
-
-class FlavorImporter(ABC):
-    @abstractmethod
-    def import_flavors(self) -> list[Flavor]:
-        pass
+from abc import ABC, abstractmethod
+
+from osi_dump.model.flavor import Flavor
+
+
+class FlavorImporter(ABC):
+    @abstractmethod
+    def import_flavors(self) -> list[Flavor]:
+        pass