Add CaaS networks related validations

[ta/cm-plugins.git] / recuserconfighandlers / recnetworkprofileshandler / recnetworkprofileshandler.py

diff --git a/recuserconfighandlers/recnetworkprofileshandler/recnetworkprofileshandler.py b/recuserconfighandlers/recnetworkprofileshandler/recnetworkprofileshandler.py
new file mode 100644 (file)
index 0000000..9bcbf4f
--- /dev/null
+++ b/recuserconfighandlers/recnetworkprofileshandler/recnetworkprofileshandler.py
@@ -0,0 +1,64 @@
+#! /usr/bin/python
+# Copyright 2019 Nokia
+#
+# 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 cmframework.apis import cmuserconfig
+from cmframework.apis import cmerror
+from cmdatahandlers.api import configerror
+from cmdatahandlers.network_profiles.config import config as net_prof_config
+
+"""
+This plugin is used to handle REC specific network profiles configs. Currently
+its sole purpuse is to set default type for any provider networks.
+"""
+
+
+class recnetworkprofileshandler(cmuserconfig.CMUserConfigPlugin):
+    defaul_sriov_net_type = 'caas'
+    defaul_provider_interface_type = 'caas'
+
+    def __init__(self):
+        super(recnetworkprofileshandler, self).__init__()
+
+    def handle(self, confman):
+        try:
+            self._set_default_type_for_provider_networks(confman)
+        except configerror.ConfigError as exp:
+            raise cmerror.CMError(str(exp))
+
+    def _set_default_type_for_provider_networks(self, confman):
+        netprofconf = confman.get_network_profiles_config_handler()
+        network_profiles = netprofconf.get_network_profiles()
+        for profile in network_profiles:
+            try:
+                self._set_default_provider_network_type(netprofconf, profile)
+                self._set_default_sriov_provider_network_type(netprofconf, profile)
+            except (net_prof_config.MissingProviderNetworkInterfaces, net_prof_config.MissingSriovProviderNetworks):
+                pass
+
+    def _set_default_provider_network_type(self, netprofconf, profile):
+        for interface in netprofconf.get_profile_provider_network_interfaces(profile):
+            try:
+                netprofconf.get_profile_provider_network_interface_type(
+                    profile, interface)
+            except net_prof_config.MissingProviderNetworkInterfaceType:
+                netprofconf.set_profile_provider_network_interface_type(
+                        profile, interface, self.defaul_provider_interface_type)
+
+    def _set_default_sriov_provider_network_type(self, netprofconf, profile):
+        for sriov_net in netprofconf.get_profile_sriov_provider_networks(profile):
+            if not netprofconf.get_profile_sriov_provider_network_type(
+                    profile, sriov_net):
+                netprofconf.set_profile_sriov_provider_network_type(
+                        profile, sriov_net, self.defaul_sriov_net_type)