X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2dms%2Fadapter%2Fdms_repository.py;h=d0116d038f54c4857fe225189363db9dfeb9f140;hb=refs%2Fheads%2Fmaster;hp=c69acff2a652f20a4ce5acc71620851e882d0610;hpb=5e02e76308e1677fb106572b885a366eb6c0fbec;p=pti%2Fo2.git

diff --git a/o2dms/adapter/dms_repository.py b/o2dms/adapter/dms_repository.py
index c69acff..d0116d0 100644
--- a/o2dms/adapter/dms_repository.py
+++ b/o2dms/adapter/dms_repository.py
@@ -28,10 +28,78 @@ class NfDeploymentDescSqlAlchemyRepository(dms_repo
 
     def _get(self, nfdeployment_desc_id) -> dms.NfDeploymentDesc:
         return self.session.query(dms.NfDeploymentDesc).filter_by(
-            nfDeploymentDescId=nfdeployment_desc_id).first()
+            id=nfdeployment_desc_id).first()
 
     def _list(self) -> List[dms.NfDeploymentDesc]:
         return self.session.query()
 
-    def _update(self, nfdeployment_desc: dms.NfDeploymentDesc):
-        self.session.add(nfdeployment_desc)
+    def _update(self, nfdeployment_desc_id, **kwargs):
+        self.session.query(dms.NfDeploymentDesc).filter_by(
+            id=nfdeployment_desc_id).update(**kwargs)
+
+    def _delete(self, nfdeployment_desc_id):
+        self.session.query(dms.NfDeploymentDesc).filter_by(
+            id=nfdeployment_desc_id
+        ).delete()
+
+    def _count(self, **kwargs):
+        return self.session.query(
+            dms.NfDeploymentDesc).filter_by(**kwargs).count()
+
+
+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()
+
+    def _count(self, **kwargs):
+        return self.session.query(
+            dms.NfDeployment).filter_by(**kwargs).count()
+
+
+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()