o2ims/adapter/orm.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.\r
   2 #\r
   3 #  Licensed under the Apache License, Version 2.0 (the "License");\r
   4 #  you may not use this file except in compliance with the License.\r
   5 #  You may obtain a copy of the License at\r
   6 #\r
   7 #      http://www.apache.org/licenses/LICENSE-2.0\r
   8 #\r
   9 #  Unless required by applicable law or agreed to in writing, software\r
  10 #  distributed under the License is distributed on an "AS IS" BASIS,\r
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  12 #  See the License for the specific language governing permissions and\r
  13 #  limitations under the License.\r
  14 \r
  15 import logging\r
  16 \r
  17 from sqlalchemy import (\r
  18     Table,\r
  19     MetaData,\r
  20     Column,\r
  21     # Integer,\r
  22     String,\r
  23     # Date,\r
  24     ForeignKey,\r
  25     # engine,\r
  26     # event,\r
  27 )\r
  28 \r
  29 from sqlalchemy.orm import mapper, relationship\r
  30 \r
  31 from o2ims.domain import ocloud as ocloudModel\r
  32 \r
  33 logger = logging.getLogger(__name__)\r
  34 \r
  35 metadata = MetaData()\r
  36 \r
  37 ocloud = Table(\r
  38     "ocloud",\r
  39     metadata,\r
  40     Column("oCloudId", String(255), primary_key=True),\r
  41     Column("globalcloudId", String(255)),\r
  42     Column("name", String(255)),\r
  43     Column("description", String(255)),\r
  44     Column("infrastructureManagementServiceEndpoint", String(255))\r
  45     # Column("extensions", String(1024))\r
  46 )\r
  47 \r
  48 resourcetype = Table(\r
  49     "resourcetype",\r
  50     metadata,\r
  51     Column("resourceTypeId", String(255), primary_key=True),\r
  52     Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
  53     Column("name", String(255)),\r
  54     Column("vendor", String(255)),\r
  55     Column("model", String(255)),\r
  56     Column("version", String(255)),\r
  57     Column("description", String(255)),\r
  58     # Column("extensions", String(1024))\r
  59 )\r
  60 \r
  61 resourcepool = Table(\r
  62     "resourcepool",\r
  63     metadata,\r
  64     Column("resourcePoolId", String(255), primary_key=True),\r
  65     Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
  66     Column("globalLocationId", String(255)),\r
  67     Column("name", String(255)),\r
  68     Column("location", String(255)),\r
  69     Column("description", String(255)),\r
  70     # Column("resources", String(1024))\r
  71     # Column("extensions", String(1024))\r
  72 )\r
  73 \r
  74 resource = Table(\r
  75     "resource",\r
  76     metadata,\r
  77     Column("resourceId", String(255), primary_key=True),\r
  78     Column("resourceTypeId", ForeignKey("resourcetype.resourceTypeId")),\r
  79     Column("resourcePoolId", ForeignKey("resourcepool.resourcePoolId")),\r
  80     # Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
  81     Column("parentId", String(255)),\r
  82     Column("description", String(255)),\r
  83     # Column("elements", String(255))\r
  84     # Column("extensions", String(1024))\r
  85 )\r
  86 \r
  87 deploymentmanager = Table(\r
  88     "deploymentmanager",\r
  89     metadata,\r
  90     Column("deploymentManagerId", String(255), primary_key=True),\r
  91     Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
  92     Column("name", String(255)),\r
  93     Column("description", String(255)),\r
  94     Column("deploymentManagementServiceEndpoint", String(255)),\r
  95     Column("supportedLocations", String(255)),\r
  96     Column("capabilities", String(255)),\r
  97     Column("capacity", String(255)),\r
  98     # Column("extensions", String(1024))\r
  99 )\r
 100 \r
 101 subscription = Table(\r
 102     "subscription",\r
 103     metadata,\r
 104     Column("subscriptionId", String(255), primary_key=True),\r
 105     Column("callback", String(255)),\r
 106     Column("consumerSubscriptionId", String(255)),\r
 107     Column("filter", String(255)),\r
 108 )\r
 109 \r
 110 \r
 111 def start_o2ims_mappers(engine=None):\r
 112     logger.info("Starting O2 IMS mappers")\r
 113 \r
 114     dm_mapper = mapper(ocloudModel.DeploymentManager, deploymentmanager)\r
 115     resourcepool_mapper = mapper(ocloudModel.ResourcePool, resourcepool)\r
 116     resourcetype_mapper = mapper(ocloudModel.ResourceType, resourcetype)\r
 117     mapper(\r
 118         ocloudModel.Ocloud,\r
 119         ocloud,\r
 120         properties={\r
 121             "deploymentManagers": relationship(dm_mapper),\r
 122             "resourceTypes": relationship(resourcetype_mapper),\r
 123             "resourcePools": relationship(resourcepool_mapper)\r
 124         })\r
 125     mapper(\r
 126         ocloudModel.Resource,\r
 127         resource,\r
 128         properties={\r
 129             "resourceTypes": relationship(resourcetype_mapper),\r
 130             "resourcePools": relationship(resourcepool_mapper)\r
 131         }\r
 132     )\r
 133     mapper(ocloudModel.Subscription, subscription)\r
 134 \r
 135     if engine is not None:\r
 136         metadata.create_all(engine)\r