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
|
from onapsdk.configuration import settings
from onapsdk.sdc.vendor import Vendor
from onapsdk.sdc.vsp import Vsp
from onaptests.utils.resources import get_resource_location
from ..base import BaseStep, YamlTemplateBaseStep
from .vendor import VendorOnboardStep
class VspOnboardStep(BaseStep):
"""Vsp onboard step."""
def __init__(self, cleanup=False):
"""Initialize step.
Substeps:
- VendorOnboardStep.
"""
super().__init__(cleanup=cleanup)
self.add_step(VendorOnboardStep(cleanup=cleanup))
@property
def description(self) -> str:
"""Step description."""
return "Onboard vsp in SDC."
@property
def component(self) -> str:
"""Component name."""
return "SDC"
@BaseStep.store_state
def execute(self):
"""Onboard Vsp.
Use settings values:
- VSP_NAME,
- VSP_FILE_PATH,
- VENDOR_NAME.
"""
super().execute()
vendor: Vendor = Vendor(name=settings.VENDOR_NAME)
vsp: Vsp = Vsp(name=settings.VSP_NAME, vendor=vendor, package=open(settings.VSP_FILE_PATH, "rb"))
vsp.onboard()
class YamlTemplateVspOnboardStep(YamlTemplateBaseStep):
"""Vsp onboard using YAML template step."""
def __init__(self, cleanup=False):
"""Initialize step.
Substeps:
- VendorOnboardStep.
"""
super().__init__(cleanup=cleanup)
self.add_step(VendorOnboardStep(cleanup=cleanup))
@property
def description(self) -> str:
"""Step description."""
return "Onboard vsp described in YAML file in SDC."
@property
def component(self) -> str:
"""Component name."""
return "SDC"
@property
def yaml_template(self) -> dict:
"""YAML template.
Get YAML template from parent.
Returns:
dict: YAML template
"""
if settings.MODEL_YAML_TEMPLATE:
return self.model_yaml_template
else:
return self.parent.yaml_template
@property
def model_yaml_template(self) -> dict:
"""Model YAML template.
Get model YAML template from parent.
Returns:
dict: YAML template
"""
return self.parent.model_yaml_template
@YamlTemplateBaseStep.store_state
def execute(self):
"""Onboard Vsps from YAML template.
Use settings values:
- VENDOR_NAME.
"""
super().execute()
vendor: Vendor = Vendor(name=settings.VENDOR_NAME)
if "vnfs" in self.yaml_template:
for vnf in self.yaml_template["vnfs"]:
with open(get_resource_location(vnf["heat_files_to_upload"]), "rb") as package:
vsp: Vsp = Vsp(name=f"{vnf['vnf_name']}_VSP",
vendor=vendor,
package=package)
vsp.onboard()
elif "pnfs" in self.yaml_template:
for pnf in self.yaml_template["pnfs"]:
if "heat_files_to_upload" in pnf:
with open(get_resource_location(pnf["heat_files_to_upload"]), "rb") as package:
vsp: Vsp = Vsp(name=f"{pnf['pnf_name']}_VSP",
vendor=vendor,
package=package)
vsp.onboard()
|