Base class for get model names activity

Change-Id: Ia7eeb2a89eab5c3c4364de7bb6b8f2313d12163f
Signed-off-by: Daniel Arndt <daniel.arndt@canonical.com>

diff --git a/osm_common/temporal/activities/vnf.py b/osm_common/temporal/activities/vnf.py
index 0c5b22e..9aca0e3 100644 (file)
--- a/osm_common/temporal/activities/vnf.py
+++ b/osm_common/temporal/activities/vnf.py
@@ -417,3 +417,51 @@ class SetVnfModel(BaseActivity):
 
     async def __call__(self, activity_input: Input) -> None:
         raise NotImplementedError()
+
+
+class GetModelNames(BaseActivity):
+    """Gets the models of VNFs associated with the NS.
+
+    Collaborators:
+        DB read:           vnfrs
+
+    Raises (retryable):
+        DbException: If DB read operations fail
+
+    Activity Lifecycle:
+        This activity should complete relatively quickly (less than a second).
+
+        This activity will not report a heartbeat due to its
+        short-running nature.
+
+        This is an idempotent activity.
+    """
+
+    @dataclass
+    class Input:
+        """
+        Attributes:
+        -----------
+        ns_uuid : str
+            The UUID of the NS from which we are looking for the model names.
+        """
+
+        ns_uuid: str
+
+    @dataclass
+    class Output:
+        """
+        Attributes:
+        -----------
+        model_names: set[str]
+            The set of model names associated with the NS.
+        """
+
+        model_names: set[str]
+
+    def __init__(self, db: DbBase):
+        super().__init__()
+        self.db: DbBase = db
+
+    async def __call__(self, activity_input: Input) -> Output:
+        raise NotImplementedError()