# See the License for the specific language governing permissions and\r
# limitations under the License.\r
\r
-from o2ims.service import unit_of_work\r
+import uuid\r
\r
+from o2common.service import unit_of_work\r
+from o2ims.views.ocloud_dto import SubscriptionDTO\r
+from o2ims.domain.subscription_obj import Subscription\r
\r
-def ocloud_one(ocloudid: str, uow: unit_of_work.SqlAlchemyUnitOfWork):\r
+from o2common.helper import o2logging\r
+logger = o2logging.get_logger(__name__)\r
+\r
+\r
+def oclouds(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.oclouds.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def ocloud_one(ocloudid: str, uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.oclouds.get(ocloudid)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def resource_types(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.resource_types.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def resource_type_one(resourceTypeId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.resource_types.get(resourceTypeId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def resource_pools(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.resource_pools.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def resource_pool_one(resourcePoolId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.resource_pools.get(resourcePoolId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def resources(resourcePoolId: str, uow: unit_of_work.AbstractUnitOfWork,\r
+ **kwargs):\r
+\r
+ filter_kwargs = {} # filter key should be the same with database name\r
+ if 'resourceTypeName' in kwargs:\r
+ resource_type_name = kwargs['resourceTypeName']\r
+ with uow:\r
+ # res_types = uow.resource_types.list()\r
+ # restype_ids = [\r
+ # restype.resourceTypeId for restype in res_types\r
+ # if resourceTypeName == restype.name]\r
+ # restype_id = '' if len(restype_ids) == 0 else restype_ids[0]\r
+ res_type = uow.resource_types.get_by_name(resource_type_name)\r
+ restype_id = '' if res_type is None else res_type.resourceTypeId\r
+ filter_kwargs['resourceTypeId'] = restype_id\r
+\r
+ # li = uow.resources.list(resourcePoolId)\r
+ # return [r.serialize() for r in li if r.resourceTypeId == restype_id]\r
+ if 'parentId' in kwargs:\r
+ filter_kwargs['parentId'] = kwargs['parentId']\r
+\r
+ with uow:\r
+ li = uow.resources.list(resourcePoolId, **filter_kwargs)\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def resource_one(resourceId: str, uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.resources.get(resourceId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def deployment_managers(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.deployment_managers.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def deployment_manager_one(deploymentManagerId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.deployment_managers.get(deploymentManagerId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def subscriptions(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.subscriptions.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def subscription_one(subscriptionId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.subscriptions.get(subscriptionId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def subscription_create(subscriptionDto: SubscriptionDTO.subscription,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+\r
+ sub_uuid = str(uuid.uuid4())\r
+ subscription = Subscription(\r
+ sub_uuid, subscriptionDto['callback'],\r
+ subscriptionDto['consumerSubscriptionId'],\r
+ subscriptionDto['filter'])\r
with uow:\r
- results = uow.session.execute(\r
- """\r
- SELECT oCloudId, name FROM ocloud WHERE oCloudId = :ocloudid\r
- """,\r
- dict(ocloudid=ocloudid),\r
- )\r
- return dict(results[0]) if len(results) > 0 else None\r
+ uow.subscriptions.add(subscription)\r
+ uow.commit()\r
+ return {"subscriptionId": sub_uuid}\r
\r
\r
-def oclouds(uow: unit_of_work.SqlAlchemyUnitOfWork):\r
+def subscription_delete(subscriptionId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
with uow:\r
- results = uow.session.execute(\r
- """\r
- SELECT oCloudId, name FROM ocloud\r
- """,\r
- )\r
- return [dict(r) for r in results]\r
+ uow.subscriptions.delete(subscriptionId)\r
+ uow.commit()\r
+ return True\r