Set quotas on every execution and bugfix the lookup

Signed-off-by: Marc Schöchlin <[email protected]>

Author: Marc Schöchlin

src/openstack_workload_generator/entities/domain.py

@@ -97,6 +97,7 @@ def create_and_get_projects(self, create_projects: list[str]):
 
         for project_name in create_projects:
             if project_name in self.workload_projects:
+                self.workload_projects[project_name].adapt_quota()
                 continue
             project = WorkloadGeneratorProject(
                 self.conn, project_name, self.obj, self.workload_user

src/openstack_workload_generator/entities/helpers.py

@@ -62,10 +62,12 @@ def get(key: str, regex: str = ".+", multi_line: bool = False) -> str:
 
     @staticmethod
     def load_config(config_file: str):
-        potential_profile_file = str(
+
+        profile_path = str(
             os.path.realpath(os.path.dirname(os.path.realpath(__file__)))
-            + f"/../../../profiles/{config_file}"
+            + f"/../../../profiles/"
         )
+        potential_profile_file = str(Path(os.getenv("OPENSTACK_WORKLOAD_MANAGER_PROFILES", profile_path)) / Path(config_file))
 
         if os.getenv("OPENSTACK_WORKLOAD_MANAGER_PROFILES", None):
             potential_profile_file = str(
@@ -192,7 +194,7 @@ def configured_quota_names(quota_category: str) -> list[str]:
     @staticmethod
     def quota(quota_name: str, quota_category: str, default_value: int) -> int:
         if quota_category in Config._config:
-            value = Config._config.get(quota_name, default_value)
+            value = Config._config[quota_category].get(quota_name, default_value)
             if isinstance(value, int):
                 return value
             else:

src/openstack_workload_generator/entities/project.py

@@ -169,8 +169,6 @@ def _set_quota(self, quota_category: str):
         else:
             raise RuntimeError(f"Not implemented: {quota_category}")
 
-        # service_obj = getattr(self._admin_conn, api_area)
-        # current_quota = service_obj.get_quota_set(self.obj.id)
         LOGGER.debug(f"current quotas for {quota_category} : {current_quota}")
 
         new_quota = {}
@@ -193,7 +191,7 @@ def _set_quota(self, quota_category: str):
                 )
                 new_quota[key_name] = new_value
 
-        if len(new_quota):
+        if len(new_quota.keys()) > 0:
             set_quota_method = getattr(self._admin_conn, f"set_{api_area}_quotas")
             set_quota_method(self.obj.id, **new_quota)
             LOGGER.info(