X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fadapter%2Focloud_repository.py;h=4d4361c52ab62ecea7aa5540356cfdd4714ae256;hb=5c501cb5e57a1ac3e0e7d38e22a3759e7958777a;hp=32dced90b0f665d76507e58fb395ee1748292638;hpb=8be81dfad35b08c4de77168e885bb18253069771;p=pti%2Fo2.git diff --git a/o2ims/adapter/ocloud_repository.py b/o2ims/adapter/ocloud_repository.py index 32dced9..4d4361c 100644 --- a/o2ims/adapter/ocloud_repository.py +++ b/o2ims/adapter/ocloud_repository.py @@ -15,7 +15,9 @@ from typing import List # from o2ims.adapter import orm from o2ims.domain import ocloud -from o2ims.domain.ocloud_repo import OcloudRepository +from o2ims.domain.ocloud_repo import OcloudRepository, ResourceTypeRepository,\ + ResourcePoolRepository, ResourceRepository, DeploymentManagerRepository,\ + SubscriptionRepository class OcloudSqlAlchemyRepository(OcloudRepository): @@ -27,9 +29,9 @@ class OcloudSqlAlchemyRepository(OcloudRepository): self.session.add(ocloud) # self.session.add_all(ocloud.deploymentManagers) - def _get(self, ocloudid) -> ocloud.Ocloud: + def _get(self, ocloud_id) -> ocloud.Ocloud: return self.session.query(ocloud.Ocloud).filter_by( - oCloudId=ocloudid).first() + oCloudId=ocloud_id).first() def _list(self) -> List[ocloud.Ocloud]: return self.session.query(ocloud.Ocloud).order_by( @@ -37,3 +39,98 @@ class OcloudSqlAlchemyRepository(OcloudRepository): def _update(self, ocloud: ocloud.Ocloud): self.session.add(ocloud) + + +class ResouceTypeSqlAlchemyRepository(ResourceTypeRepository): + def __init__(self, session): + super().__init__() + self.session = session + + def _add(self, resourceType: ocloud.ResourceType): + self.session.add(resourceType) + + def _get(self, resource_type_id) -> ocloud.ResourceType: + return self.session.query(ocloud.ResourceType).filter_by( + resourceTypeId=resource_type_id).first() + + def _list(self) -> List[ocloud.ResourceType]: + return self.session.query() + + def _update(self, resourceType: ocloud.ResourceType): + self.session.add(resourceType) + + +class ResourcePoolSqlAlchemyRepository(ResourcePoolRepository): + def __init__(self, session): + super().__init__() + self.session = session + + def _add(self, resourcePool: ocloud.ResourcePool): + self.session.add(resourcePool) + + def _get(self, resource_pool_id) -> ocloud.ResourcePool: + return self.session.query(ocloud.ResourcePool).filter_by( + resourcePoolId=resource_pool_id).first() + + def _list(self) -> List[ocloud.ResourcePool]: + return self.session.query() + + def _update(self, resourcePool: ocloud.ResourcePool): + self.session.add(resourcePool) + + +class ResourceSqlAlchemyRepository(ResourceRepository): + def __init__(self, session): + super().__init__() + self.session = session + + def _add(self, resource: ocloud.Resource): + self.session.add(resource) + + def _get(self, resource_id) -> ocloud.Resource: + return self.session.query(ocloud.Resource).filter_by( + resourceId=resource_id).first() + + def _list(self) -> List[ocloud.Resource]: + return self.session.query() + + 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)