X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fadapter%2Focloud_repository.py;h=0826cf0dc2d9a4533d137ecff214a1dec5e40623;hb=84867b76a65efc4e9add52c86eae7e451c4a038d;hp=5fa9b7e05863229af30a1927265b804c455f97d3;hpb=38646c684a13536146ecf16a35e6e33d614b29e9;p=pti%2Fo2.git diff --git a/o2ims/adapter/ocloud_repository.py b/o2ims/adapter/ocloud_repository.py index 5fa9b7e..0826cf0 100644 --- a/o2ims/adapter/ocloud_repository.py +++ b/o2ims/adapter/ocloud_repository.py @@ -16,7 +16,8 @@ from typing import List # from o2ims.adapter import orm from o2ims.domain import ocloud from o2ims.domain.ocloud_repo import OcloudRepository, ResourceTypeRepository,\ - ResourcePoolRepository, ResourceRepository + ResourcePoolRepository, ResourceRepository, DeploymentManagerRepository,\ + SubscriptionRepository class OcloudSqlAlchemyRepository(OcloudRepository): @@ -95,3 +96,45 @@ class ResourceSqlAlchemyRepository(ResourceRepository): def _update(self, resource: ocloud.Resource): self.session.add(resource) + + +class DeploymentManagerSqlAlchemyRepository(DeploymentManagerRepository): + def __init__(self, session): + super().__init__() + self.session = session + + def _add(self, deployment_manager: ocloud.DeploymentManager): + self.session.add(deployment_manager) + + def _get(self, deployment_manager_id) -> ocloud.DeploymentManager: + return self.session.query(ocloud.DeploymentManager).filter_by( + deploymentManagerId=deployment_manager_id).first() + + def _list(self) -> List[ocloud.DeploymentManager]: + return self.session.query() + + def _update(self, deployment_manager: ocloud.DeploymentManager): + self.session.add(deployment_manager) + + +class SubscriptionSqlAlchemyRepository(SubscriptionRepository): + def __init__(self, session): + super().__init__() + self.session = session + + def _add(self, subscription: ocloud.Subscription): + self.session.add(subscription) + + def _get(self, subscription_id) -> ocloud.Subscription: + return self.session.query(ocloud.Subscription).filter_by( + subscriptionId=subscription_id).first() + + def _list(self) -> List[ocloud.Subscription]: + return self.session.query() + + def _update(self, subscription: ocloud.Subscription): + self.session.add(subscription) + + def _delete(self, subscription_id): + self.session.query(ocloud.Subscription).filter_by( + subscriptionId=subscription_id).delete()