osi-dump 0.1.3.3.2.6__py3-none-any.whl → 0.1.3.3.2.8__py3-none-any.whl

osi_dump/importer/image/openstack_image_importer.py

@@ -50,12 +50,15 @@ class OpenStackImageImporter(ImageImporter):
 
     def _get_image_info(self, os_image: OSImage) -> Image:
 
-        properties: dict = os_image.properties
-
-        properties.pop("owner_specified.openstack.md5", None)
-        properties.pop("owner_specified.openstack.sha256", None)
-        properties.pop("owner_specified.openstack.object", None)
-        properties.pop("stores", None)
+        try:
+            properties: dict = os_image.properties
+
+            properties.pop("owner_specified.openstack.md5", None)
+            properties.pop("owner_specified.openstack.sha256", None)
+            properties.pop("owner_specified.openstack.object", None)
+            properties.pop("stores", None)
+        except Exception as e: 
+            logger.warn(f"properties for {os_image.id} is None")
 
         image = Image(
             image_id=os_image.id,

osi_dump/importer/role_assignment/openstack_role_assignment_importer.py

@@ -15,13 +15,28 @@ from osi_dump.api.keystone import get_role_assignments
 logger = logging.getLogger(__name__)
 
 
-class OpenStackRoleAssignmentImporter(RoleAssignmentImporter):
-    user_names = {}
-    role_names = {}
-    
-    
+class OpenStackRoleAssignmentImporter(RoleAssignmentImporter):    
     def __init__(self, connection: Connection):
         self.connection = connection
+        
+        self.users = {}
+        self.roles = {}
+        
+    def _get_users(self): 
+        os_users = self.connection.identity.users()
+        
+        for os_user in os_users: 
+            print(os_user)
+            self.users[os_user.id] = os_user.name 
+    
+    def _get_roles(self): 
+        os_roles = self.connection.identity.roles()
+        
+        for os_role in os_roles: 
+            
+            print(os_role)
+            self.roles[os_role.id] = os_role.name 
+        
 
     def import_role_assignments(self) -> list[RoleAssignment]:
         """Import role_assignments information from Openstack
@@ -36,11 +51,22 @@ class OpenStackRoleAssignmentImporter(RoleAssignmentImporter):
         logger.info(
             f"Importing role_assignments for {self.connection.auth['auth_url']}"
         )
+        
+        try: 
+            self._get_users()
+        except Exception as e: 
+            logger.info(f"Getting user list failed {e}")
+            
+        try: 
+            self._get_roles() 
+        except Exception as e: 
+            logger.info(f"Getting role list failed {e}")
 
         try:
             osrole_assignments: list[OSRoleAssignment] = list(
                 self.connection.identity.role_assignments()
             )
+
         except Exception as e:
             raise Exception(
                 f"Can not fetch role_assignments for {self.connection.auth['auth_url']} {e}"
@@ -81,25 +107,12 @@ class OpenStackRoleAssignmentImporter(RoleAssignmentImporter):
         role_name = None
 
         try:
-            if user_id in OpenStackRoleAssignmentImporter.user_names:
-                user_name = OpenStackRoleAssignmentImporter.user_names[user_id]
-            else: 
-                user_name = self.connection.identity.get_user(
-                    role_assignment.user["id"]
-                ).name
-                OpenStackRoleAssignmentImporter.user_names[user_id] = user_name
+            user_name = self.users[user_id]
         except Exception as e:
             logger.warning(f"Can not get user name: {e}")
 
         try:
-            if role_id in OpenStackRoleAssignmentImporter.role_names: 
-                role_name = OpenStackRoleAssignmentImporter.role_names[role_id]
-            else: 
-                role_name = self.connection.identity.get_role(
-                    role_assignment.role["id"]
-                ).name
-                OpenStackRoleAssignmentImporter.role_names[role_id] = role_name 
-                
+            role_name = self.roles[role_id]
         except Exception as e:
             logger.warning(f"Can not get role name: {e}")
 

{osi_dump-0.1.3.3.2.6.dist-info → osi_dump-0.1.3.3.2.8.dist-info}/METADATA

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

{osi_dump-0.1.3.3.2.6.dist-info → osi_dump-0.1.3.3.2.8.dist-info}/RECORD

@@ -66,7 +66,7 @@ osi_dump/importer/hypervisor/hypervisor_importer.py,sha256=JuoJBltqFYhH-Ql9TLUPH
 osi_dump/importer/hypervisor/openstack_hypervisor_importer.py,sha256=lVxVGWKNlNAPhEHC_nWg9cxIGz8Lv6AVQXwIgQsWnP0,3923
 osi_dump/importer/image/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osi_dump/importer/image/image_importer.py,sha256=6zwAMnS58S2HGwARwebqf4VE9IBQBv5Ot6nmcn9H4fY,181
-osi_dump/importer/image/openstack_image_importer.py,sha256=ffc13qnIGu07OXnIPaKiVxGdufep82RfQoKqLb1bwHc,2455
+osi_dump/importer/image/openstack_image_importer.py,sha256=dqQAsNDPljTDN82YPEjBbPDjvLOCtad_hgRWhIpCWXw,2585
 osi_dump/importer/instance/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osi_dump/importer/instance/instance_importer.py,sha256=TaaPgEFVVMRT3mSXrzQkSaSIBPBwwANpEaWCgRsgImc,196
 osi_dump/importer/instance/openstack_instance_importer.py,sha256=D-Q9AIN81g2GW-SRWpKxBeagxvJvhN_AumSt5WGQrEc,4244
@@ -77,7 +77,7 @@ osi_dump/importer/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 osi_dump/importer/project/openstack_project_importer.py,sha256=ylcm3jEFqh4gW8vhzhIaBq5v4A-PV3BaaogC8uHlSkg,3295
 osi_dump/importer/project/project_importer.py,sha256=jwEvxteFbSwyWMV8hKAtf5Lo2bZysWkQ1L_NVUE8XOQ,191
 osi_dump/importer/role_assignment/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-osi_dump/importer/role_assignment/openstack_role_assignment_importer.py,sha256=i_3fx86y_UotOkQ9I9oLiXQ3UlXtokYxW4XreodAIW0,3678
+osi_dump/importer/role_assignment/openstack_role_assignment_importer.py,sha256=iJYSY8VWNZ8K7xz8wotGOqbvS5fjJsA6WQcmyPLh3mU,3720
 osi_dump/importer/role_assignment/role_assignment_importer.py,sha256=Faorx0qsgdXjv5y2t-7obpV3wFfbmByx_-8b9yBr7L8,228
 osi_dump/importer/router/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osi_dump/importer/router/openstack_router_importer.py,sha256=T_Wr45nJrCcp2OOdRawmdQkNBDzczA-3DPx9dlewd4g,2781
@@ -109,8 +109,8 @@ osi_dump/util/extract_hostname.py,sha256=IpdklGHevmtRu67xeSRE_5n2mvWGI1sDsnJwExo
 osi_dump/util/openstack_util.py,sha256=_o8LiRmQKtxQ8hU3rbrMNwIAxt0-aWD2rjq-NvVdK9E,1299
 osi_dump/util/panda_excel.py,sha256=owr8AffIxqN3yY8kIv55Ox4j2dRz7kXZvSZFZeft8dw,734
 osi_dump/util/validate_dir_path.py,sha256=pL_OrY8JnNwk3vj6Zp6bsZtgHXhszSGRoqIt-1G5S90,507
-osi_dump-0.1.3.3.2.6.dist-info/METADATA,sha256=YP_-NefdJ8q6kpPwWq5d752llU6lf5SMQ9PERl2Pzus,679
-osi_dump-0.1.3.3.2.6.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-osi_dump-0.1.3.3.2.6.dist-info/entry_points.txt,sha256=ozm5sIBtXzLv6_FiUe26v1BgA3_xUReGLPhKQKZ56wQ,46
-osi_dump-0.1.3.3.2.6.dist-info/top_level.txt,sha256=OtAAwmJfcoPvlw_Cemo_H1aXIGV_7w0O2941KQt6faQ,9
-osi_dump-0.1.3.3.2.6.dist-info/RECORD,,
+osi_dump-0.1.3.3.2.8.dist-info/METADATA,sha256=NWKgobveDL7jT3LbI4NVIH4Ui3lleOP9yMXKN9tjnyg,679
+osi_dump-0.1.3.3.2.8.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+osi_dump-0.1.3.3.2.8.dist-info/entry_points.txt,sha256=ozm5sIBtXzLv6_FiUe26v1BgA3_xUReGLPhKQKZ56wQ,46
+osi_dump-0.1.3.3.2.8.dist-info/top_level.txt,sha256=OtAAwmJfcoPvlw_Cemo_H1aXIGV_7w0O2941KQt6faQ,9
+osi_dump-0.1.3.3.2.8.dist-info/RECORD,,