Use dynamic build for Manager class

Signed-off-by: mulhern <[email protected]>

pyproject.toml

@@ -1,3 +1,6 @@
 [build-system]
 requires = ["setuptools"]
 build-backend = "setuptools.build_meta"
+
+[tool.pylint]
+good-names="Manager"

src/stratis_cli/_actions/_top.py

@@ -34,6 +34,7 @@
 from ._connection import get_object
 from ._constants import TOP_OBJECT
 from ._formatting import print_table
+from ._utils import ClassKey, make_dyn_class
 
 
 def _fetch_keylist(proxy):
@@ -44,8 +45,7 @@ def _fetch_keylist(proxy):
     :rtype: list of str
     :raises StratisCliEngineError:
     """
-    # pylint: disable=import-outside-toplevel
-    from ._data import Manager
+    Manager = make_dyn_class(ClassKey.MANAGER)
 
     (keys, return_code, message) = Manager.Methods.ListKeys(proxy, {})
     if return_code != StratisdErrors.OK:  # pragma: no cover
@@ -68,8 +68,7 @@ def _add_update_key(proxy, key_desc, capture_key, *, keyfile_path):
     """
     assert capture_key == (keyfile_path is None)
 
-    # pylint: disable=import-outside-toplevel
-    from ._data import Manager
+    Manager = make_dyn_class(ClassKey.MANAGER)
 
     if capture_key:
         password = getpass(prompt="Enter key data followed by the return key: ")
@@ -127,7 +126,7 @@ def get_report(namespace):
 
         else:
             if namespace.report_name == ReportKey.ENGINE_STATE.value:
-                from ._data import Manager
+                Manager = make_dyn_class(ClassKey.MANAGER)
 
                 (report, return_code, message) = Manager.Methods.EngineStateReport(
                     get_object(TOP_OBJECT), {}
@@ -242,8 +241,7 @@ def unset_key(namespace):
         :raises StratisCliNoChangeError:
         :raises StratisCliIncoherenceError:
         """
-        # pylint: disable=import-outside-toplevel
-        from ._data import Manager
+        Manager = make_dyn_class(ClassKey.MANAGER)
 
         proxy = get_object(TOP_OBJECT)
 

src/stratis_cli/_actions/_utils.py

@@ -18,10 +18,17 @@
 
 # isort: STDLIB
 import json
+import os
+import xml.etree.ElementTree as ET  # nosec B405
+from enum import Enum
 from uuid import UUID
 
+# isort: FIRSTPARTY
+from dbus_python_client_gen import DPClientGenerationError, make_class
+
 from .._constants import PoolIdType
 from .._errors import (
+    StratisCliGenerationError,
     StratisCliMissingClevisTangURLError,
     StratisCliMissingClevisThumbprintError,
 )
@@ -32,6 +39,36 @@
     CLEVIS_PIN_TANG,
     CLEVIS_PIN_TPM2,
 )
+from ._constants import MANAGER_INTERFACE
+from ._environment import get_timeout
+from ._introspect import SPECS
+
+DBUS_TIMEOUT_SECONDS = 120
+
+
+class ClassKey(Enum):
+    """
+    Keys for dynamically generated classes.
+    """
+
+    MANAGER = ("Manager", MANAGER_INTERFACE)
+
+
+def make_dyn_class(key):
+    """
+    Dynamically generate a class from introspection specification.
+    """
+    timeout = get_timeout(
+        os.environ.get("STRATIS_DBUS_TIMEOUT", DBUS_TIMEOUT_SECONDS * 1000)
+    )
+    try:
+        return make_class(
+            key.value[0], ET.fromstring(SPECS[key.value[1]]), timeout  # nosec B314
+        )
+    except DPClientGenerationError as err:  # pragma: no cover
+        raise StratisCliGenerationError(
+            "Failed to generate some class needed for invoking dbus-python methods"
+        ) from err
 
 
 class ClevisInfo: