update the logic to distribute service from sdc.
[modeling/etsicatalog.git] / genericparser / pub / msapi / sdc.py
1 # Copyright 2017 ZTE Corporation.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 import json
16 import logging
17 import os
18
19 from genericparser.pub.config.config import SDC_BASE_URL, SDC_USER, SDC_PASSWD
20 from genericparser.pub.exceptions import GenericparserException
21 from genericparser.pub.utils import fileutil
22 from genericparser.pub.utils import restcall
23
24 logger = logging.getLogger(__name__)
25
26 ASSETTYPE_RESOURCES = "resources"
27 ASSETTYPE_SERVICES = "services"
28 DISTRIBUTED = "DISTRIBUTED"
29
30
31 def call_sdc(resource, method, content=''):
32     additional_headers = {
33         'X-ECOMP-InstanceID': 'Modeling',
34     }
35     return restcall.call_req(base_url=SDC_BASE_URL,
36                              user=SDC_USER,
37                              passwd=SDC_PASSWD,
38                              auth_type=restcall.rest_no_auth,
39                              resource=resource,
40                              method=method,
41                              content=content,
42                              additional_headers=additional_headers)
43
44
45 """
46 sample of return value
47 [
48     {
49         "uuid": "c94490a0-f7ef-48be-b3f8-8d8662a37236",
50         "invariantUUID": "63eaec39-ffbe-411c-a838-448f2c73f7eb",
51         "name": "underlayvpn",
52         "version": "2.0",
53         "toscaModelURL": "/sdc/v1/catalog/resources/c94490a0-f7ef-48be-b3f8-8d8662a37236/toscaModel",
54         "category": "Volte",
55         "subCategory": "VolteVF",
56         "resourceType": "VF",
57         "lifecycleState": "CERTIFIED",
58         "lastUpdaterUserId": "jh0003"
59     }
60 ]
61 """
62
63
64 def get_artifacts(asset_type):
65     resource = "/sdc/v1/catalog/{assetType}"
66     resource = resource.format(assetType=asset_type)
67     ret = call_sdc(resource, "GET")
68     if ret[0] != 0:
69         logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
70         raise GenericparserException("Failed to query artifacts(%s) from sdc." % asset_type)
71     return json.JSONDecoder().decode(ret[1])
72
73
74 def get_artifact(asset_type, csar_id):
75     artifacts = get_artifacts(asset_type)
76     for artifact in artifacts:
77         if artifact["uuid"] == csar_id:
78             if asset_type == ASSETTYPE_SERVICES and \
79                     artifact.get("distributionStatus", None) != DISTRIBUTED:
80                 raise GenericparserException(
81                     "The artifact (%s,%s) is not distributed from sdc." % (asset_type, csar_id))
82             else:
83                 return artifact
84     raise GenericparserException("Failed to query artifact(%s,%s) from sdc." % (asset_type, csar_id))
85
86
87 def get_asset(asset_type, uuid):
88     resource = "/sdc/v1/catalog/{assetType}/{uuid}/metadata".format(assetType=asset_type, uuid=uuid)
89     ret = call_sdc(resource, "GET")
90     if ret[0] != 0:
91         logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
92         raise GenericparserException("Failed to get asset(%s, %s) from sdc." % (asset_type, uuid))
93     asset = json.JSONDecoder().decode(ret[1])
94     if asset.get("distributionStatus", None) != DISTRIBUTED:
95         raise GenericparserException("The asset (%s,%s) is not distributed from sdc." % (asset_type, uuid))
96     else:
97         return asset
98
99
100 def delete_artifact(asset_type, asset_id, artifact_id):
101     resource = "/sdc/v1/catalog/{assetType}/{uuid}/artifacts/{artifactUUID}"
102     resource = resource.format(assetType=asset_type, uuid=asset_id, artifactUUID=artifact_id)
103     ret = call_sdc(resource, "DELETE")
104     if ret[0] != 0:
105         logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
106         raise GenericparserException("Failed to delete artifacts(%s) from sdc." % artifact_id)
107     return json.JSONDecoder().decode(ret[1])
108
109
110 def download_artifacts(download_url, local_path, file_name):
111     additional_headers = {
112         'X-ECOMP-InstanceID': 'Modeling',
113         'accept': 'application/octet-stream'
114     }
115     ret = restcall.call_req(base_url=SDC_BASE_URL,
116                             user=SDC_USER,
117                             passwd=SDC_PASSWD,
118                             auth_type=restcall.rest_no_auth,
119                             resource=download_url,
120                             method="GET",
121                             additional_headers=additional_headers)
122     if ret[0] != 0:
123         logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
124         raise GenericparserException("Failed to download %s from sdc." % download_url)
125     fileutil.make_dirs(local_path)
126     local_file_name = os.path.join(local_path, file_name)
127     local_file = open(local_file_name, 'wb')
128     local_file.write(ret[1])
129     local_file.close()
130     return local_file_name