blob: 5407a8576ee017c525b64e42152ad689b17ea0a1 (
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
|
import logging
import logging.config
from abc import ABC, abstractmethod
from typing import List
from onapsdk.configuration import settings
from onapsdk.aai.business import Customer
class BaseStep(ABC):
"""Base step class."""
_logger: logging.Logger = logging.getLogger("")
def __init_subclass__(cls):
"""Subclass initialization.
Add _logger property for any BaseStep subclass
"""
super().__init_subclass__()
cls._logger: logging.Logger = logging.getLogger("")
logging.config.dictConfig(settings.LOG_CONFIG)
# Setup Proxy if SOCK_HTTP is defined in settings
try:
cls.set_proxy(settings.SOCK_HTTP)
except AttributeError:
pass
def __init__(self, cleanup: bool = False) -> None:
"""Step initialization.
Args:
cleanup(bool, optional): Determines if cleanup action should be called.
"""
self._steps: List["BaseStep"] = []
self._cleanup: bool = cleanup
self._parent: "BaseStep" = None
def add_step(self, step: "BaseStep") -> None:
"""Add substep.
Add substep and mark step as a substep parent.
Args:
step (BaseStep): Step object
"""
self._steps.append(step)
step._parent: "BaseStep" = self
@property
def parent(self) -> "BaseStep":
"""Step parent.
If parent is not set the step is a root one.
"""
return self._parent
@property
def is_root(self) -> bool:
"""Is a root step.
Step is a root if has no parent
Returns:
bool: True if step is a root step, False otherwise
"""
return self._parent is None
def execute(self) -> None:
"""Step's action.
Run all substeps action before it's own action.
Override this method and remember to call `super().action()` before.
"""
for step in self._steps:
step.execute()
def cleanup(self) -> None:
"""Step's cleanup.
Not all steps has to have cleanup method
"""
if self._cleanup:
for step in self._steps:
step.cleanup()
@classmethod
def set_proxy(cls, sock_http):
"""Set sock proxy."""
onap_proxy = {}
onap_proxy['http'] = sock_http
onap_proxy['https'] = sock_http
Customer.set_proxy(onap_proxy)
class YamlTemplateBaseStep(BaseStep, ABC):
"""Base YAML template step."""
@property
@abstractmethod
def yaml_template(self) -> dict:
"""YAML template abstract property.
Every YAML template step need to implement that property.
Returns:
dict: YAML template
"""
|
