Update o2dms lcm api

[pti/o2.git] / o2dms / adapter / dms_repository.py

diff --git a/o2dms/adapter/dms_repository.py b/o2dms/adapter/dms_repository.py
index 5dd2bb1..2c7c3ca 100644 (file)
--- a/o2dms/adapter/dms_repository.py
+++ b/o2dms/adapter/dms_repository.py
@@ -41,3 +41,57 @@ class NfDeploymentDescSqlAlchemyRepository(dms_repo
         self.session.query(dms.NfDeploymentDesc).filter_by(
             id=nfdeployment_desc_id
         ).delete()
+
+
+class NfDeploymentSqlAlchemyRepository(
+        dms_repo.NfDeploymentRepository):
+
+    def __init__(self, session):
+        super().__init__()
+        self.session = session
+
+    def _add(self, nfdeployment: dms.NfDeployment):
+        self.session.add(nfdeployment)
+
+    def _get(self, nfdeployment_id) -> dms.NfDeployment:
+        return self.session.query(dms.NfDeployment).filter_by(
+            id=nfdeployment_id).first()
+
+    def _list(self) -> List[dms.NfDeployment]:
+        return self.session.query()
+
+    def _update(self, nfdeployment_id, **kwargs):
+        self.session.query(dms.NfDeployment).filter_by(
+            id=nfdeployment_id).update(**kwargs)
+
+    def _delete(self, nfdeployment_id):
+        self.session.query(dms.NfDeployment).filter_by(
+            id=nfdeployment_id
+        ).delete()
+
+
+class NfOCloudVResourceSqlAlchemyRepository(
+        dms_repo.NfOCloudVResourceRepository):
+
+    def __init__(self, session):
+        super().__init__()
+        self.session = session
+
+    def _add(self, nfocloudvres: dms.NfOCloudVResource):
+        self.session.add(nfocloudvres)
+
+    def _get(self, nfocloudvres_id) -> dms.NfOCloudVResource:
+        return self.session.query(dms.NfOCloudVResource).filter_by(
+            id=nfocloudvres_id).first()
+
+    def _list(self) -> List[dms.NfOCloudVResource]:
+        return self.session.query()
+
+    def _update(self, nfocloudvres, **kwargs):
+        self.session.query(dms.NfOCloudVResource).filter_by(
+            id=nfocloudvres).update(**kwargs)
+
+    def _delete(self, nfocloudvres):
+        self.session.query(dms.NfOCloudVResource).filter_by(
+            id=nfocloudvres
+        ).delete()