Move registration API to configuration
[pti/o2.git] / 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 # from typing_extensions import Required\r
16 from sqlalchemy import (\r
17     Table,\r
18     MetaData,\r
19     Column,\r
20     Integer,\r
21     String,\r
22     Text,\r
23     Enum,\r
24     # Date,\r
25     DateTime,\r
26     ForeignKey,\r
27     # Boolean,\r
28     # engine,\r
29     # event,\r
30 )\r
31 \r
32 from sqlalchemy.orm import mapper, relationship\r
33 # from sqlalchemy.sql.sqltypes import Integer\r
34 \r
35 from o2ims.domain import ocloud as ocloudModel\r
36 from o2ims.domain import subscription_obj as subModel\r
37 from o2ims.domain import configuration_obj as confModel\r
38 from o2ims.domain.resource_type import ResourceTypeEnum\r
39 \r
40 from o2common.helper import o2logging\r
41 logger = o2logging.get_logger(__name__)\r
42 \r
43 metadata = MetaData()\r
44 \r
45 ocloud = Table(\r
46     "ocloud",\r
47     metadata,\r
48     Column("updatetime", DateTime),\r
49     Column("createtime", DateTime),\r
50     Column("hash", String(255)),\r
51     Column("version_number", Integer),\r
52 \r
53     Column("oCloudId", String(255), primary_key=True),\r
54     Column("globalcloudId", String(255)),\r
55     Column("name", String(255)),\r
56     Column("description", String(255)),\r
57     Column("infrastructureManagementServiceEndpoint", String(255))\r
58     # Column("extensions", String(1024))\r
59 )\r
60 \r
61 resourcetype = Table(\r
62     "resourcetype",\r
63     metadata,\r
64     Column("updatetime", DateTime),\r
65     Column("createtime", DateTime),\r
66     Column("hash", String(255)),\r
67 \r
68     Column("resourceTypeId", String(255), primary_key=True),\r
69     Column("resourceTypeEnum", Enum(ResourceTypeEnum), nullable=False),\r
70     Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
71     Column("name", String(255)),\r
72     Column("vendor", String(255)),\r
73     Column("model", String(255)),\r
74     Column("version", String(255)),\r
75     Column("description", String(255)),\r
76     # Column("extensions", String(1024))\r
77 )\r
78 \r
79 resourcepool = Table(\r
80     "resourcepool",\r
81     metadata,\r
82     Column("updatetime", DateTime),\r
83     Column("createtime", DateTime),\r
84     Column("hash", String(255)),\r
85     Column("version_number", Integer),\r
86 \r
87     Column("resourcePoolId", String(255), primary_key=True),\r
88     Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
89     Column("globalLocationId", String(255)),\r
90     Column("name", String(255)),\r
91     Column("location", String(255)),\r
92     Column("description", String(255)),\r
93     # Column("resources", String(1024))\r
94     # Column("extensions", String(1024))\r
95 )\r
96 \r
97 resource = Table(\r
98     "resource",\r
99     metadata,\r
100     Column("updatetime", DateTime),\r
101     Column("createtime", DateTime),\r
102     Column("hash", String(255)),\r
103     Column("version_number", Integer),\r
104 \r
105     Column("resourceId", String(255), primary_key=True),\r
106     Column("resourceTypeId", ForeignKey("resourcetype.resourceTypeId")),\r
107     Column("resourcePoolId", ForeignKey("resourcepool.resourcePoolId")),\r
108     Column("name", String(255)),\r
109     # Column("globalAssetId", String(255)),\r
110     Column("parentId", String(255)),\r
111     Column("description", String(255)),\r
112     Column("elements", Text())\r
113     # Column("extensions", String(1024))\r
114 )\r
115 \r
116 deploymentmanager = Table(\r
117     "deploymentmanager",\r
118     metadata,\r
119     Column("updatetime", DateTime),\r
120     Column("createtime", DateTime),\r
121     Column("hash", String(255)),\r
122     Column("version_number", Integer),\r
123 \r
124     Column("deploymentManagerId", String(255), primary_key=True),\r
125     Column("oCloudId", ForeignKey("ocloud.oCloudId")),\r
126     Column("name", String(255)),\r
127     Column("description", String(255)),\r
128     Column("deploymentManagementServiceEndpoint", String(255)),\r
129     Column("supportedLocations", String(255)),\r
130     Column("capabilities", String(255)),\r
131     Column("capacity", String(255)),\r
132     # Column("extensions", String(1024))\r
133 )\r
134 \r
135 subscription = Table(\r
136     "subscription",\r
137     metadata,\r
138     Column("updatetime", DateTime),\r
139     Column("createtime", DateTime),\r
140     Column("hash", String(255)),\r
141     Column("version_number", Integer),\r
142 \r
143     Column("subscriptionId", String(255), primary_key=True),\r
144     Column("callback", String(255)),\r
145     Column("consumerSubscriptionId", String(255)),\r
146     Column("filter", String(255)),\r
147 )\r
148 \r
149 configuration = Table(\r
150     "configuration",\r
151     metadata,\r
152     Column("updatetime", DateTime),\r
153     Column("createtime", DateTime),\r
154 \r
155     Column("configurationId", String(255), primary_key=True),\r
156     Column("conftype", String(255)),\r
157     Column("callback", String(255)),\r
158     Column("status", String(255)),\r
159     Column("comments", String(255)),\r
160 )\r
161 \r
162 \r
163 def start_o2ims_mappers(engine=None):\r
164     logger.info("Starting O2 IMS mappers")\r
165 \r
166     dm_mapper = mapper(ocloudModel.DeploymentManager, deploymentmanager)\r
167     resourcepool_mapper = mapper(ocloudModel.ResourcePool, resourcepool)\r
168     resourcetype_mapper = mapper(ocloudModel.ResourceType, resourcetype)\r
169     mapper(\r
170         ocloudModel.Ocloud,\r
171         ocloud,\r
172         properties={\r
173             "deploymentManagers": relationship(dm_mapper),\r
174             "resourceTypes": relationship(resourcetype_mapper),\r
175             "resourcePools": relationship(resourcepool_mapper)\r
176         })\r
177     mapper(\r
178         ocloudModel.Resource,\r
179         resource,\r
180         properties={\r
181             "resourceTypes": relationship(resourcetype_mapper),\r
182             "resourcePools": relationship(resourcepool_mapper)\r
183         }\r
184     )\r
185     mapper(subModel.Subscription, subscription)\r
186     mapper(confModel.Configuration, configuration)\r
187 \r
188     if engine is not None:\r
189         metadata.create_all(engine)\r