osi-dump 0.1.2.9__tar.gz → 0.1.3__tar.gz

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: osi-dump
-Version: 0.1.2.9
+Version: 0.1.3
 Summary: OpenStack information dump tool
 Author: TVKain
 License: Apache-2.0

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "osi-dump"
-version = "0.1.2.9"
+version = "0.1.3"
 description = "OpenStack information dump tool"
 readme = "README.md"
 authors = [{ name = "TVKain" }]

osi_dump-0.1.3/src/osi_dump/api/neutron.py

@@ -0,0 +1,42 @@
+import logging
+
+from openstack.connection import Connection
+from openstack.identity.v3.service import Service
+from openstack.load_balancer.v2.load_balancer import LoadBalancer
+
+logger = logging.getLogger(__name__)
+
+
+def get_floating_ip_project(connection: Connection, floating_ip_id: str):
+    neutron_endpoint = connection.endpoint_for(
+        service_type="network", interface="public"
+    )
+
+    try:
+        url = f"{neutron_endpoint}/v2.0/floatingips/{floating_ip_id}?fields=project_id"
+        response = connection.session.get(url)
+
+    except Exception as e:
+        print(e)
+
+    data = response.json()
+
+    return data["floatingip"]["project_id"]
+
+
+def get_router_project(connection: Connection, router_id: str):
+    octavia_endpoint = connection.endpoint_for(
+        service_type="network", interface="internal"
+    )
+
+    try:
+        url = f"{octavia_endpoint}/v2.0/routers/{router_id}?fields=project_id"
+
+        response = connection.session.get(url)
+
+    except Exception as e:
+        print(e)
+
+    data = response.json()
+
+    return data["router"]["project_id"]

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/src/osi_dump/importer/external_port/openstack_external_port_importer.py

@@ -15,6 +15,8 @@ from openstack.network.v2.router import Router as OSRouter
 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__)
 
 
@@ -146,11 +148,15 @@ class OpenStackExternalPortImporter(ExternalPortImporter):
         return ""
 
     def _map_project_id_router(self, device_id: str) -> str:
-        router: OSRouter = self.connection.get_router(name_or_id=device_id)
+        project_id = osi_neutron.get_router_project(
+            self.connection, router_id=device_id
+        )
 
-        return router.project_id
+        return project_id
 
     def _map_project_id_floating(self, device_id: str) -> str:
-        floating_ip: OSFloatingIP = self.connection.get_floating_ip(id=device_id)
+        project_id = osi_neutron.get_floating_ip_project(
+            self.connection, floating_ip_id=device_id
+        )
 
-        return floating_ip.project_id
+        return project_id

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/src/osi_dump/importer/instance/openstack_instance_importer.py

@@ -5,6 +5,8 @@ import concurrent
 from openstack.connection import Connection
 from openstack.compute.v2.server import Server
 
+from openstack.compute.v2.flavor import Flavor as OSFlavor
+
 from osi_dump.importer.instance.instance_importer import InstanceImporter
 from osi_dump.model.instance import Instance
 
@@ -87,6 +89,28 @@ class OpenStackInstanceImporter(InstanceImporter):
                 f"Unable to obtain IP address information for instance {server.name}: {e}"
             )
 
+        vgpus = None
+        vgpu_type = None
+
+        vgpu_metadata_property = "pci_passthrough:alias"
+
+        try:
+            flavor: OSFlavor = self.connection.get_flavor(
+                name_or_id=server.flavor["id"]
+            )
+
+            vgpu_prop: str = flavor.extra_specs[vgpu_metadata_property]
+
+            vgpu_props = vgpu_prop.split(":")
+
+            vgpu_type = vgpu_props[0]
+            vgpus = int(vgpu_props[1])
+
+        except Exception as e:
+            logger.warning(
+                f"Unable to obtain GPU information for instance {server.name}: {e}"
+            )
+
         instance = Instance(
             instance_id=server.id,
             instance_name=server.name,
@@ -101,6 +125,8 @@ class OpenStackInstanceImporter(InstanceImporter):
             vcpus=server.flavor["vcpus"],
             created_at=server.created_at,
             updated_at=server.updated_at,
+            vgpus=vgpus,
+            vgpu_type=vgpu_type,
         )
 
         return instance

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/src/osi_dump/model/instance.py

@@ -16,6 +16,8 @@ class Instance(BaseModel):
     status: str
     ram: int
     vcpus: int
+    vgpus: Optional[int]
+    vgpu_type: Optional[str]
     hypervisor: Optional[str]
     created_at: str
     updated_at: str

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/src/osi_dump.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: osi-dump
-Version: 0.1.2.9
+Version: 0.1.3
 Summary: OpenStack information dump tool
 Author: TVKain
 License: Apache-2.0

{osi_dump-0.1.2.9 → osi_dump-0.1.3}/src/osi_dump.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ src/osi_dump.egg-info/entry_points.txt
 src/osi_dump.egg-info/requires.txt
 src/osi_dump.egg-info/top_level.txt
 src/osi_dump/api/__init__.py
+src/osi_dump/api/neutron.py
 src/osi_dump/api/octavia.py
 src/osi_dump/api/placement.py
 src/osi_dump/batch_handler/__init__.py