Enhance: Enable O2 IMS for distributed cloud

[pti/o2.git] / o2common / service / client / base_client.py

diff --git a/o2common/service/client/base_client.py b/o2common/service/client/base_client.py
index 96ff988..c10cf07 100644 (file)
--- a/o2common/service/client/base_client.py
+++ b/o2common/service/client/base_client.py
@@ -19,7 +19,7 @@ import abc
 
 class BaseClient(abc.ABC):
     def __init__(self):
-        pass
+        self._pool_id = None
 
     def list(self, **filters):
         return self._list(**filters)
@@ -27,6 +27,10 @@ class BaseClient(abc.ABC):
     def get(self, id):
         return self._get(id)
 
+    def set_pool_driver(self, pool_id):
+        self._pool_id = pool_id
+        self._set_stx_client()
+
     @abc.abstractmethod
     def _get(self, id):
         raise NotImplementedError
@@ -34,3 +38,7 @@ class BaseClient(abc.ABC):
     @abc.abstractmethod
     def _list(self, **filters):
         raise NotImplementedError
+
+    @abc.abstractmethod
+    def _set_stx_client(self):
+        raise NotImplementedError