aboutsummaryrefslogtreecommitdiffstats
path: root/onap-client/onap_client/sdc/license_model.py
blob: 591509ea2d06566e68c1edc006405465e35fd4e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
# -*- coding: utf8 -*-
# ============LICENSE_START=======================================================
# org.onap.vvp/validation-scripts
# ===================================================================
# Copyright © 2020 AT&T Intellectual Property. All rights reserved.
# ===================================================================
#
# Unless otherwise specified, all software contained herein is licensed
# under the Apache License, Version 2.0 (the "License");
# you may not use this software except in compliance with the License.
# You may obtain a copy of the License at
#
#             http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#
#
# Unless otherwise specified, all documentation contained herein is licensed
# under the Creative Commons License, Attribution 4.0 Intl. (the "License");
# you may not use this documentation except in compliance with the License.
# You may obtain a copy of the License at
#
#             https://creativecommons.org/licenses/by/4.0/
#
# Unless required by applicable law or agreed to in writing, documentation
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# ============LICENSE_END============================================

from onap_client.lib import generate_dummy_string, generate_dummy_date
from onap_client.resource import Resource
from onap_client.client.clients import Client


class LicenseModel(Resource):
    resource_name = "LICENSE_MODEL"
    spec = {
        "vendor_name": {
            "type": str,
            "required": False,
            "default": generate_dummy_string("test_vendor_"),
        },
        "mfr_ref_number": {
            "type": str,
            "required": False,
            "default": generate_dummy_string("mfref"),
        },
        "entitlement_pool_name": {
            "type": str,
            "required": False,
            "default": generate_dummy_string("test_kg_"),
        },
        "key_group_name": {
            "type": str,
            "required": False,
            "default": generate_dummy_string("test_ep_"),
        },
        "feature_group_name": {
            "type": str,
            "required": False,
            "default": generate_dummy_string("test_fg_"),
        },
        "license_agreement_name": {
            "type": str,
            "required": False,
            "default": generate_dummy_string("test_la_"),
        },
        "license_start_date": {
            "type": str,
            "required": False,
            "default": generate_dummy_date(days=1),
        },
        "license_end_date": {
            "type": str,
            "required": False,
            "default": generate_dummy_date(days=365),
        },
    }

    def __init__(
        self,
        vendor_name,
        mfr_ref_number,
        entitlement_pool_name,
        key_group_name,
        feature_group_name,
        license_agreement_name,
        license_start_date,
        license_end_date,
    ):
        self.oc = Client()

        license_input = {}
        license_input["vendor_name"] = vendor_name
        license_input["manufacturer_reference_number"] = mfr_ref_number
        license_input["entitlement_pool_name"] = entitlement_pool_name
        license_input["key_group_name"] = key_group_name
        license_input["feature_group_name"] = feature_group_name
        license_input["license_agreement_name"] = license_agreement_name
        license_input["license_start_date"] = license_start_date
        license_input["license_end_date"] = license_end_date

        super().__init__(license_input)

    def _create(self, license_input):
        """Creates a license model object in SDC"""
        return create_license_model(license_input)

    def _post_create(self):
        pass

    def _submit(self):
        """Submits the license model in SDC"""

        self.oc.sdc.license_model.submit_license_model(**self.attributes, action="Submit")

        license_model = self.oc.sdc.license_model.get_license_model(**self.attributes)
        self.attributes["tosca"] = license_model.response_data


# TODO
# Break this up into class funcs?
def create_license_model(license_input):
    """Creates a license model object in SDC

    :license_input: dictionary with values to input for lm creation

    :return: dictionary of updated values for created lm
    """
    oc = Client()

    kwargs = license_input
    license_model = oc.sdc.license_model.add_license_model(**kwargs)

    kwargs["license_model_id"] = license_model.license_model_id
    kwargs["license_model_version_id"] = license_model.license_model_version_id

    key_group = oc.sdc.license_model.add_key_group(**kwargs)
    key_group_id = key_group.key_group_id

    entitlement_pool = oc.sdc.license_model.add_entitlement_pool(**kwargs)
    entitlement_pool_id = entitlement_pool.entitlement_pool_id

    kwargs["entitlement_pool_id"] = entitlement_pool_id
    kwargs["key_group_id"] = key_group_id

    feature_group = oc.sdc.license_model.add_feature_group(**kwargs)
    feature_group_id = feature_group.feature_group_id

    kwargs["feature_group_id"] = feature_group_id

    license_agreement = oc.sdc.license_model.add_license_agreement(**kwargs)
    kwargs["license_agreement_id"] = license_agreement.license_agreement_id

    license_model = oc.sdc.license_model.get_license_model(**kwargs)
    kwargs["tosca"] = license_model.response_data

    return kwargs


def get_license_model_id(license_model_name):
    """GETs license model UUID from SDC

    :license_model_name: name of license model in SDC

    :return: uuid of lm or None
    """
    oc = Client()

    response = oc.sdc.license_model.get_license_models()
    results = response.response_data.get("results")
    for license_model in results:
        if license_model.get("name") == license_model_name:
            return license_model.get("id")
    return None


def get_license_model_version_id(license_model_id):
    """GETs license model version UUID from SDC

    :license_model_id: uuid of license model in SDC

    :return: uuid of lm version id or None
    """
    oc = Client()

    license_model_version_id = None
    creation_time = -1
    response = oc.sdc.license_model.get_license_model_versions(
        license_model_id=license_model_id
    )
    results = response.response_data.get("results")
    for version in results:
        if version.get("creationTime", 0) > creation_time:
            creation_time = version.get("creationTime")
            license_model_version_id = version.get("id")

    return license_model_version_id


def get_license_model_attribute(license_model_id, license_model_version_id, attribute):
    """GETs license model attribute from SDC

    :license_model_id: uuid of license model in SDC
    :license_model_version_id: uuid of license model version in SDC
    :attribute: attribute to GET (license-agreements, feature-groups, entitlement-pools, license-key-groups)

    :return: uuid of attribute of license-model
    """
    oc = Client()

    response = oc.sdc.license_model.get_license_model_version_attribute(
        license_model_id=license_model_id,
        license_model_version_id=license_model_version_id,
        attribute=attribute,
    )
    return response.response_data.get("results")[0]