Add compass-tasks

Registered tasks and MQ modules for Compass

Change-Id: Id1569a61fe53357d53448478d5ba42cb1f386bc6
Signed-off-by: Harry Huang <huangxiangyu5@huawei.com>

1 files changed, 151 insertions, 0 deletions

diff --git a/compass-tasks/actions/health_check/check_os_installer.py b/compass-tasks/actions/health_check/check_os_installer.py
new file mode 100644
index 0000000..6ef9818
--- /dev/null
+++ b/compass-tasks/actions/health_check/check_os_installer.py
@@ -0,0 +1,151 @@
+# Copyright 2014 Huawei Technologies Co. Ltd
+#
+# 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.
+
+"""Compass Health Check module for OS Installer."""
+
+import os
+import xmlrpclib
+
+from compass.actions.health_check import base
+
+
+class OsInstallerCheck(base.BaseCheck):
+    """os installer health check."""
+    NAME = "OS Installer Check"
+
+    def run(self):
+        """do health check."""
+        method_name = 'self.' + self.os_installer['name'] + '_check()'
+        return eval(method_name)
+
+    def cobbler_check(self):
+        """Runs cobbler check from xmlrpc client."""
+        try:
+            remote = xmlrpclib.Server(
+                self.os_installer['cobbler_url'],
+                allow_none=True)
+            credentials = self.os_installer['credentials']
+            token = remote.login(
+                credentials['username'], credentials['password'])
+        except Exception:
+            self.code = 0
+            self.messages.append(
+                "[%s]Error: Cannot login to Cobbler with "
+                "the tokens provided in the config file"
+                % self.NAME)
+            self.messages.append(
+                "[%s]Error: Failed to connect to Cobbler "
+                "API, please check if /etc/cobbler/setting "
+                "is properly configured" % self.NAME)
+            return (self.code, self.messages)
+
+        check_result = remote.check(token)
+
+        for index, message in enumerate(check_result):
+            if "SELinux" in message:
+                check_result.pop(index)
+
+        if len(check_result) != 0:
+            self.code = 0
+            for error_msg in check_result:
+                self.messages.append("[%s]Error: " % self.NAME + error_msg)
+
+        if len(remote.get_distros()) == 0:
+            self._set_status(0,
+                             "[%s]Error: No Cobbler distros found" % self.NAME)
+
+        if len(remote.get_profiles()) == 0:
+            self._set_status(0,
+                             "[%s]Error: No Cobbler profiles found"
+                             % self.NAME)
+
+        found_ppa = False
+        if len(remote.get_repos()) != 0:
+            for repo in remote.get_repos():
+                if 'ppa_repo' in repo['mirror']:
+                    found_ppa = True
+                    break
+
+        if found_ppa is False:
+            self._set_status(0,
+                             "[%s]Error: No repository ppa_repo found"
+                             % self.NAME)
+
+        path_map = {
+            'match_kickstart': (
+                '/var/lib/cobbler/kickstarts/',
+                ['default.ks', 'default.seed']
+            ),
+            'match_snippets': (
+                '/var/lib/cobbler/snippets/',
+                [
+                    'kickstart_done',
+                    'kickstart_start',
+                    'kickstart_pre_partition_disks',
+                    'kickstart_partition_disks',
+                    'kickstart_pre_anamon',
+                    'kickstart_post_anamon',
+                    'kickstart_pre_install_network_config',
+                    'kickstart_network_config',
+                    'kickstart_post_install_network_config',
+                    'kickstart_chef',
+                    'kickstart_ntp',
+                    'kickstart_yum_repo_config',
+                    'preseed_pre_partition_disks',
+                    'preseed_partition_disks',
+                    'preseed_pre_anamon',
+                    'preseed_post_anamon',
+                    'preseed_pre_install_network_config',
+                    'preseed_network_config',
+                    'preseed_post_install_network_config',
+                    'preseed_chef',
+                    'preseed_ntp',
+                    'preseed_apt_repo_config',
+                ]
+            ),
+            'match_ks_mirror': (
+                '/var/www/cobbler/',
+                ['ks_mirror']
+            ),
+            'match_repo_mirror': (
+                '/var/www/cobbler/',
+                ['repo_mirror']
+            ),
+            'match_iso': (
+                '/var/lib/cobbler/',
+                ['iso']
+            ),
+        }
+        not_exists = []
+        for key in path_map.keys():
+            for path in path_map[key][1]:
+                if not os.path.exists(path_map[key][0] + path):
+                    not_exists.append(path_map[key][0] + path)
+
+        if len(not_exists) != 0:
+            self._set_status(
+                0,
+                "[%s]Error: These locations do not exist: "
+                "%s" % (
+                    self.NAME,
+                    ', '.join(item for item in not_exists)
+                )
+            )
+
+        if self.code == 1:
+            self.messages.append(
+                "[%s]Info: OS Installer health check has completed."
+                " No problems found, all systems go." % self.NAME)
+
+        return (self.code, self.messages)