Network Module class structure for Kube support
Related-PROD: PROD-35903
Change-Id: I474a347452b472df3e2272357d1a036b4893e844
diff --git a/cfg_checker/modules/network/mapper.py b/cfg_checker/modules/network/mapper.py
index fd19864..2e925f2 100644
--- a/cfg_checker/modules/network/mapper.py
+++ b/cfg_checker/modules/network/mapper.py
@@ -5,7 +5,7 @@
from cfg_checker.common import logger_cli
from cfg_checker.common.exception import InvalidReturnException
from cfg_checker.modules.network.network_errors import NetworkErrors
-from cfg_checker.nodes import SaltNodes
+from cfg_checker.nodes import SaltNodes, KubeNodes
# TODO: use templated approach
# net interface structure should be the same
@@ -38,16 +38,16 @@
skip_list=None,
skip_list_file=None
):
- self.salt_master = SaltNodes(config)
logger_cli.info("# Initializing mapper")
+ self.env_config = config
# init networks and nodes
self.networks = {}
- self.nodes = self.salt_master.get_nodes(
+ self.nodes = self.master.get_nodes(
skip_list=skip_list,
skip_list_file=skip_list_file
)
- self.cluster = self.salt_master.get_info()
- self.domain = self.salt_master.domain
+ self.cluster = self.master.get_info()
+ self.domain = self.master.domain
# init and pre-populate interfaces
self.interfaces = {k: {} for k in self.nodes}
# Init errors class
@@ -115,14 +115,14 @@
# class uses nodes from self.nodes dict
_reclass = {}
# Get required pillars
- self.salt_master.get_specific_pillar_for_nodes("linux:network")
- for node in self.salt_master.nodes.keys():
+ self.master.get_specific_pillar_for_nodes("linux:network")
+ for node in self.master.nodes.keys():
# check if this node
- if not self.salt_master.is_node_available(node):
+ if not self.master.is_node_available(node):
continue
# get the reclass value
_pillar = \
- self.salt_master.nodes[node]['pillars']['linux']['network']
+ self.master.nodes[node]['pillars']['linux']['network']
# we should be ready if there is no interface in reclass for a node
# for example on APT node
if 'interface' in _pillar:
@@ -172,16 +172,16 @@
# class uses nodes from self.nodes dict
_runtime = {}
logger_cli.info("# Mapping node runtime network data")
- self.salt_master.prepare_script_on_active_nodes("ifs_data.py")
- _result = self.salt_master.execute_script_on_active_nodes(
+ self.master.prepare_script_on_active_nodes("ifs_data.py")
+ _result = self.master.execute_script_on_active_nodes(
"ifs_data.py",
args=["json"]
)
- for key in self.salt_master.nodes.keys():
+ for key in self.master.nodes.keys():
# check if we are to work with this node
- if not self.salt_master.is_node_available(key):
+ if not self.master.is_node_available(key):
continue
- # due to much data to be passed from salt_master,
+ # due to much data to be passed from master,
# it is happening in order
if key in _result:
_text = _result[key]
@@ -194,21 +194,21 @@
)
)
_dict = json.loads(_text[_text.find('{'):])
- self.salt_master.nodes[key]['routes'] = _dict.pop("routes")
- self.salt_master.nodes[key]['networks'] = _dict
+ self.master.nodes[key]['routes'] = _dict.pop("routes")
+ self.master.nodes[key]['networks'] = _dict
else:
- self.salt_master.nodes[key]['networks'] = {}
- self.salt_master.nodes[key]['routes'] = {}
+ self.master.nodes[key]['networks'] = {}
+ self.master.nodes[key]['routes'] = {}
logger_cli.debug("... {} has {} networks".format(
key,
- len(self.salt_master.nodes[key]['networks'].keys())
+ len(self.master.nodes[key]['networks'].keys())
))
logger_cli.info("-> done collecting networks data")
logger_cli.info("-> mapping IPs")
# match interfaces by IP subnets
- for host, node_data in self.salt_master.nodes.items():
- if not self.salt_master.is_node_available(host):
+ for host, node_data in self.master.nodes.items():
+ if not self.master.is_node_available(host):
continue
for net_name, net_data in node_data['networks'].items():
@@ -463,7 +463,7 @@
for hostname in names:
_notes = []
node = hostname.split('.')[0]
- if not self.salt_master.is_node_available(hostname, log=False):
+ if not self.master.is_node_available(hostname, log=False):
logger_cli.info(
" {0:8} {1}".format(node, "node not available")
)
@@ -516,7 +516,7 @@
# get gate and routes if proto is static
if _proto == 'static':
# get the gateway for current net
- _routes = self.salt_master.nodes[hostname]['routes']
+ _routes = self.master.nodes[hostname]['routes']
_route = _routes[_net] if _net in _routes else None
# get the default gateway
if 'default' in _routes:
@@ -732,3 +732,37 @@
# " {0:8} {1}".format(hostname.split('.')[0], _text)
# )
# logger_cli.info("\n")
+
+
+class SaltNetworkMapper(NetworkMapper):
+ def __init__(
+ self,
+ config,
+ errors_class=None,
+ skip_list=None,
+ skip_list_file=None
+ ):
+ self.master = SaltNodes(config)
+ super(SaltNetworkMapper, self).__init__(
+ config,
+ errors_class=errors_class,
+ skip_list=skip_list,
+ skip_list_file=skip_list_file
+ )
+
+
+class KubeNetworkMapper(NetworkMapper):
+ def __init__(
+ self,
+ config,
+ errors_class=None,
+ skip_list=None,
+ skip_list_file=None
+ ):
+ self.master = KubeNodes(config)
+ super(KubeNetworkMapper, self).__init__(
+ config,
+ errors_class=errors_class,
+ skip_list=skip_list,
+ skip_list_file=skip_list_file
+ )