X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=o2ims%2Fdomain%2Focloud_repo.py;h=cf2d97999cc9b169c8f9496515a2f026ec745f32;hb=7e21b8d5ceabc857812194f7a865066e4f13ad85;hp=811e85cffc6ba8d3aeb2faaa748d062e34e9ad0e;hpb=defe8209b3628593c186487857fe02586d7e1503;p=pti%2Fo2.git

diff --git a/o2ims/domain/ocloud_repo.py b/o2ims/domain/ocloud_repo.py
index 811e85c..cf2d979 100644
--- a/o2ims/domain/ocloud_repo.py
+++ b/o2ims/domain/ocloud_repo.py
@@ -13,7 +13,7 @@
 #  limitations under the License.
 
 import abc
-from typing import List, Set
+from typing import List, Set, Tuple
 from o2ims.domain import ocloud
 
 
@@ -31,8 +31,8 @@ class OcloudRepository(abc.ABC):
             self.seen.add(ocloud)
         return ocloud
 
-    def list(self) -> List[ocloud.Ocloud]:
-        return self._list()
+    def list(self, *args) -> List[ocloud.Ocloud]:
+        return self._list(*args)
 
     def update(self, ocloud: ocloud.Ocloud):
         self._update(ocloud)
@@ -49,6 +49,10 @@ class OcloudRepository(abc.ABC):
     def _get(self, ocloud_id) -> ocloud.Ocloud:
         raise NotImplementedError
 
+    @abc.abstractmethod
+    def _list(self, *args) -> List[ocloud.Ocloud]:
+        raise NotImplementedError
+
     @abc.abstractmethod
     def _update(self, ocloud: ocloud.Ocloud):
         raise NotImplementedError
@@ -74,8 +78,12 @@ class ResourceTypeRepository(abc.ABC):
             self.seen.add(resource_type)
         return resource_type
 
-    def list(self) -> List[ocloud.ResourceType]:
-        return self._list()
+    def list(self, *args) -> List[ocloud.ResourceType]:
+        return self._list(*args)[1]
+
+    def list_with_count(self, *args, **kwargs) -> \
+            Tuple[int, List[ocloud.ResourceType]]:
+        return self._list(*args, **kwargs)
 
     def update(self, resource_type: ocloud.ResourceType):
         self._update(resource_type)
@@ -93,6 +101,10 @@ class ResourceTypeRepository(abc.ABC):
     def _get_by_name(self, resource_type_name) -> ocloud.ResourceType:
         raise NotImplementedError
 
+    @abc.abstractmethod
+    def _list(self, **kwargs) -> Tuple[int, List[ocloud.ResourceType]]:
+        raise NotImplementedError
+
     @abc.abstractmethod
     def _update(self, resource_type: ocloud.ResourceType):
         raise NotImplementedError
@@ -112,8 +124,12 @@ class ResourcePoolRepository(abc.ABC):
             self.seen.add(resource_pool)
         return resource_pool
 
-    def list(self) -> List[ocloud.ResourcePool]:
-        return self._list()
+    def list(self, *args) -> List[ocloud.ResourcePool]:
+        return self._list(*args)[1]
+
+    def list_with_count(self, *args, **kwargs) -> \
+            Tuple[int, List[ocloud.ResourcePool]]:
+        return self._list(*args, **kwargs)
 
     def update(self, resource_pool: ocloud.ResourcePool):
         self._update(resource_pool)
@@ -127,6 +143,10 @@ class ResourcePoolRepository(abc.ABC):
     def _get(self, resource_pool_id) -> ocloud.ResourcePool:
         raise NotImplementedError
 
+    @abc.abstractmethod
+    def _list(self, **kwargs) -> Tuple[int, List[ocloud.ResourcePool]]:
+        raise NotImplementedError
+
     @abc.abstractmethod
     def _update(self, resource_pool: ocloud.ResourcePool):
         raise NotImplementedError
@@ -146,8 +166,12 @@ class ResourceRepository(abc.ABC):
             self.seen.add(resource)
         return resource
 
-    def list(self, resourcepool_id, **kwargs) -> List[ocloud.Resource]:
-        return self._list(resourcepool_id, **kwargs)
+    def list(self, resourcepool_id, *args) -> List[ocloud.Resource]:
+        return self._list(resourcepool_id, *args)[1]
+
+    def list_with_count(self, resourcepool_id, *args, **kwargs) -> \
+            Tuple[int, List[ocloud.Resource]]:
+        return self._list(resourcepool_id, *args, **kwargs)
 
     def update(self, resource: ocloud.Resource):
         self._update(resource)
@@ -162,7 +186,8 @@ class ResourceRepository(abc.ABC):
         raise NotImplementedError
 
     @abc.abstractmethod
-    def _list(self, resourcepool_id, **kwargs) -> ocloud.Resource:
+    def _list(self, resourcepool_id, **kwargs) -> \
+            Tuple[int, List[ocloud.Resource]]:
         raise NotImplementedError
 
     @abc.abstractmethod
@@ -184,8 +209,12 @@ class DeploymentManagerRepository(abc.ABC):
             self.seen.add(deployment_manager)
         return deployment_manager
 
-    def list(self) -> List[ocloud.DeploymentManager]:
-        return self._list()
+    def list(self, *args) -> List[ocloud.DeploymentManager]:
+        return self._list(*args)[1]
+
+    def list_with_count(self, *args, **kwargs) -> \
+            Tuple[int, List[ocloud.DeploymentManager]]:
+        return self._list(*args, **kwargs)
 
     def update(self, deployment_manager: ocloud.DeploymentManager):
         self._update(deployment_manager)
@@ -198,6 +227,10 @@ class DeploymentManagerRepository(abc.ABC):
     def _get(self, deployment_manager_id) -> ocloud.DeploymentManager:
         raise NotImplementedError
 
+    @abc.abstractmethod
+    def _list(self, **kwargs) -> Tuple[int, List[ocloud.DeploymentManager]]:
+        raise NotImplementedError
+
     @abc.abstractmethod
     def _update(self, deployment_manager: ocloud.DeploymentManager):
         raise NotImplementedError