Add Subscription get and list API; Provide swagger openAPI doc

[pti/o2.git] / o2ims / adapter / unit_of_work.py

diff --git a/o2ims/adapter/unit_of_work.py b/o2ims/adapter/unit_of_work.py
index c958ce2..d966a87 100644 (file)
--- a/o2ims/adapter/unit_of_work.py
+++ b/o2ims/adapter/unit_of_work.py
@@ -19,7 +19,7 @@ from sqlalchemy.orm import sessionmaker
 from sqlalchemy.orm.session import Session\r
 \r
 from o2ims import config\r
-from o2ims.adapter.ocloud_repository import OcloudSqlAlchemyRepository\r
+from o2ims.adapter import ocloud_repository\r
 from o2ims.adapter.stx_repository import StxObjectSqlAlchemyRepository\r
 from o2ims.service.unit_of_work import AbstractUnitOfWork\r
 \r
@@ -38,7 +38,16 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
 \r
     def __enter__(self):\r
         self.session = self.session_factory()  # type: Session\r
-        self.oclouds = OcloudSqlAlchemyRepository(self.session)\r
+        self.oclouds = ocloud_repository\\r
+            .OcloudSqlAlchemyRepository(self.session)\r
+        self.resource_types = ocloud_repository\\r
+            .ResouceTypeSqlAlchemyRepository(self.session)\r
+        self.resource_pools = ocloud_repository\\r
+            .ResourcePoolSqlAlchemyRepository(self.session)\r
+        self.resources = ocloud_repository\\r
+            .ResourceSqlAlchemyRepository(self.session)\r
+        self.deployment_managers = ocloud_repository\\r
+            .DeploymentManagerSqlAlchemyRepository(self.session)\r
         self.stxobjects = StxObjectSqlAlchemyRepository(self.session)\r
         return super().__enter__()\r
 \r