summaryrefslogtreecommitdiffstats
path: root/onap_data_provider/resources/vsp_resource.py
blob: 17a4d5b21b549b0f8fbfc2b68a2c0e30505dcdc0 (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
"""VSP resource module."""
"""
   Copyright 2021 Deutsche Telekom AG

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file 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.
"""
from typing import Any, Dict, Optional
from onapsdk.sdc.vendor import Vendor  # type: ignore
from onapsdk.sdc.vsp import Vsp  # type: ignore

from .resource import Resource


class VspResource(Resource):
    """VSP resource class."""

    def __init__(self, data: Dict[str, Any]) -> None:
        """Initialize VSP resource.

        Args:
            data (Dict[str, Any]): Data needed to create resource.

        """
        super().__init__(data)
        self._vsp: Optional[Vsp] = None

    def create(self) -> None:
        """Create VSP resource."""
        if not self.exists:
            with open(self.data["package"], "rb") as package:
                self._vsp = Vsp(
                    name=self.data["name"],
                    vendor=Vendor(self.data["vendor"]),
                    package=package,
                )
                self._vsp.onboard()

    @property
    def exists(self) -> bool:
        """Check if VSP exists.

        Returns:
            bool: True if VSP exists, False otherwise

        """
        return self.vsp is not None

    @property
    def vsp(self) -> Vsp:
        """VSP property.

        Returns:
            Vsp: VSP object which is describer by provided data. None if does not exist yet.

        """
        if not self._vsp:
            vsp: Vsp = Vsp(name=self.data["name"])
            if not vsp.created():
                return None
            self._vsp = vsp
        return self._vsp