X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fadapter%2Focloud_repository.py;h=bc3557d4eed40dfacde7b5a4dd139816842f732d;hb=f73c8e3b01b8f5b7438ba544870e06d8f30cdea0;hp=5fa9b7e05863229af30a1927265b804c455f97d3;hpb=38646c684a13536146ecf16a35e6e33d614b29e9;p=pti%2Fo2.git diff --git a/o2ims/adapter/ocloud_repository.py b/o2ims/adapter/ocloud_repository.py index 5fa9b7e..bc3557d 100644 --- a/o2ims/adapter/ocloud_repository.py +++ b/o2ims/adapter/ocloud_repository.py @@ -16,7 +16,10 @@ 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 +from o2common.helper import o2logging +logger = o2logging.get_logger(__name__) class OcloudSqlAlchemyRepository(OcloudRepository): @@ -95,3 +98,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()