[APACHE] Add Apache CNF use case files
[demo.git] / tutorials / ApacheCNF / automation / create_cloud_regions.py
1 # ============LICENSE_START=======================================================
2 # Copyright (C) 2021 Orange
3 # ================================================================================
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
7 #
8 #      http://www.apache.org/licenses/LICENSE-2.0
9 #
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15 #
16 # ============LICENSE_END=========================================================
17
18 import logging
19 import os
20 from uuid import uuid4
21
22 from onapsdk.so.so_db_adapter import SoDbAdapter, IdentityService
23
24 from config import Config, VariablesDict
25 from onapsdk.exceptions import ResourceNotFound, APIError
26 from onapsdk.aai.cloud_infrastructure import Complex, CloudRegion
27 from onapsdk.msb.k8s import ConnectivityInfo
28
29 logger = logging.getLogger()
30 logger.setLevel(logging.DEBUG)
31
32
33 def create_complex(complex_id):
34     logger.info("******** Complex *******")
35     try:
36         region_complex = next(Complex.get_all(physical_location_id=complex_id))
37         logger.info("Complex exists")
38         return region_complex
39     except ResourceNotFound:
40         logger.info("Complex does not exist")
41         region_complex = Complex.create(physical_location_id=complex_id,
42                                         name=complex_id,
43                                         physical_location_type="office",
44                                         street1="DummyStreet 1",
45                                         city="DummyCity",
46                                         postal_code="00-000",
47                                         country="DummyCountry",
48                                         region="DummyRegion")
49         logger.info("Complex created")
50         return region_complex
51
52
53 def create_cloud_region(cloud_region):
54     logger.info("******** Cloud Region *******")
55     region_id = cloud_region["name"]
56     cloud_owner = cloud_region["cloud_owner"]
57     cloud_type = cloud_region["cloud_type"]
58     complex_id = cloud_region["complex_id"]
59     cloud_region_version = "1.0" if cloud_type == "k8s" else "v2.5"
60     try:
61         region = next(CloudRegion.get_all(cloud_owner=cloud_owner, cloud_region_id=region_id))
62         logger.info("Cloud region exists")
63         return region
64     except ResourceNotFound:
65         logger.info("Cloud region does not exist")
66         region = CloudRegion.create(cloud_owner=cloud_owner,
67                                     cloud_region_id=region_id,
68                                     cloud_type=cloud_type,
69                                     owner_defined_type="t1",
70                                     cloud_region_version=cloud_region_version,
71                                     complex_name=complex_id,
72                                     cloud_zone="CloudZone",
73                                     sriov_automation="false",
74                                     orchestration_disabled=False,
75                                     in_maint=False)
76         logger.info("Cloud region created")
77         return region
78
79
80 def link_region_to_complex(cloud_region, complx):
81     logger.info("******** Cloud region <-> Complex *******")
82     cloud_region.link_to_complex(complex_object=complx)
83
84
85 def add_tenant(cloud_region, tenant_id, tenant_name):
86     logger.info("Tenant does not exist")
87     cloud_region.add_tenant(tenant_id=tenant_id,
88                             tenant_name=tenant_name)
89     logger.info(f"Tenant {tenant_name} added to region")
90
91
92 def add_tenants(cloud_region, k8s_region, tenants):
93     logger.info("******** Tenants *******")
94     for tenant in tenants:
95         tenant_name = tenant["name"]
96         if k8s_region:
97             try:
98                 next(_tenant for _tenant in cloud_region.tenants if _tenant.name == tenant_name)
99                 logger.info("Tenant exists")
100             except (StopIteration, ResourceNotFound):
101                 tenant_id = str(uuid4())
102                 add_tenant(cloud_region=cloud_region, tenant_id=tenant_id, tenant_name=tenant_name)
103         else:
104             tenant_id = tenant["id"]
105             try:
106                 cloud_region.get_tenant(tenant_id)
107                 logger.info("Tenant exists")
108             except ResourceNotFound:
109                 add_tenant(cloud_region=cloud_region, tenant_id=tenant_id, tenant_name=tenant_name)
110
111
112 def update_connectivity_info(region):
113     logger.info("******** Connectivity Info *******")
114     kubeconfig_path = region["kubeconfig_file"]
115     cloud_owner = region["cloud_owner"]
116     region_id = region["name"]
117     try:
118         with open(os.path.join(os.path.dirname(os.path.realpath(__file__)), kubeconfig_path), 'rb') as kubeconfig_file:
119             kubeconfig = kubeconfig_file.read()
120
121         connectivity_info = ConnectivityInfo.get_connectivity_info_by_region_id(cloud_region_id=region_id)
122         logger.info("Connectivity Info exists ")
123         logger.info("Delete Connectivity Info ")
124         connectivity_info.delete()
125         ConnectivityInfo.create(cloud_region_id=region_id,
126                                 cloud_owner=cloud_owner,
127                                 kubeconfig=kubeconfig)
128         logger.info("Connectivity Info created ")
129     except (APIError, ResourceNotFound):
130         logger.info("Connectivity Info does not exists ")
131         ConnectivityInfo.create(cloud_region_id=region_id,
132                                 cloud_owner=cloud_owner,
133                                 kubeconfig=kubeconfig)
134         logger.info("Connectivity Info created ")
135     except FileNotFoundError:
136         logger.error("Error - File Not Found")
137         logger.info("Please check if kubeconfig file exists")
138         exit(1)
139
140
141 def add_region_to_so_db(region):
142     logger.info("******** SO Database *******")
143     if is_k8s_region(region):
144         identity_service = IdentityService(identity_id="Keystone_K8s",
145                                            url="http://test:5000/v3",
146                                            mso_id="onapsdk_user",
147                                            mso_pass="mso_pass_onapsdk",
148                                            project_domain_name="NULL",
149                                            user_domain_name="NULL",
150                                            identity_server_type="KEYSTONE")
151
152         SoDbAdapter.add_cloud_site(cloud_region_id=region["name"],
153                                    complex_id=region["complex_id"],
154                                    identity_service=identity_service,
155                                    orchestrator="multicloud")
156     else:
157         identity_url = region["identity_url"]
158         mso_id = region["mso_id"]
159         mso_pass = region["mso_pass"]
160         identity_server_type = region["identity_server_type"]
161         identity_service = IdentityService(identity_id=region["name"] + "_KEYSTONE",
162                                            url=identity_url,
163                                            mso_id=mso_id,
164                                            mso_pass=mso_pass,
165                                            project_domain_name="Default",
166                                            user_domain_name="Default",
167                                            identity_server_type=identity_server_type)
168
169         SoDbAdapter.add_cloud_site(cloud_region_id=region["name"],
170                                    complex_id=region["complex_id"],
171                                    identity_service=identity_service,
172                                    orchestrator="NULL")
173
174
175 def is_k8s_region(region):
176     is_k8s = False
177     if region["cloud_type"] == "k8s":
178         is_k8s = True
179     return is_k8s
180
181
182 ########################################################################################################################
183 def main():
184     config = Config(env_dict=VariablesDict.env_variable)
185
186     for region in config.cloud_regions:
187         complx = create_complex(region["complex_id"])
188         cloud_region = create_cloud_region(region)
189         link_region_to_complex(cloud_region, complx)
190         add_tenants(cloud_region, is_k8s_region(region), region.get("tenants"))
191         if is_k8s_region(region):
192             update_connectivity_info(region)
193         add_region_to_so_db(region)
194
195
196 if __name__ == "__main__":
197     sh = logging.StreamHandler()
198     sh_formatter = logging.Formatter('%(asctime)s %(levelname)s %(lineno)d:%(filename)s(%(process)d) - %(message)s')
199     sh.setFormatter(sh_formatter)
200     logger.addHandler(sh)
201
202     main()