galicea-odoo-addons-ecosystem/galicea_environment_checkup/environment_checkup/core.py

import logging

_logger = logging.getLogger(__name__)


class CheckResult(object):
    SUCCESS = "success"
    WARNING = "warning"
    FAIL = "fail"

    def __init__(self, result, message, details=None):
        super(CheckResult, self).__init__()

        self.result = result
        self.message = message
        self.details = details


class CheckSuccess(CheckResult):
    def __init__(self, message, **kwargs):
        super(CheckSuccess, self).__init__(CheckResult.SUCCESS, message, **kwargs)


class CheckIssue(CheckResult, Exception):
    def __init__(self, result, message, **kwargs):
        Exception.__init__(self, message)
        CheckResult.__init__(self, result, message, **kwargs)


class CheckFail(CheckIssue):
    def __init__(self, message, **kwargs):
        super(CheckFail, self).__init__(CheckResult.FAIL, message, **kwargs)


class CheckWarning(CheckIssue):
    def __init__(self, message, **kwargs):
        super(CheckWarning, self).__init__(CheckResult.WARNING, message, **kwargs)


class Check(object):
    def __init__(self, module):
        self.module = module

    def run(self, env):
        try:
            return self._run(env)
        except CheckIssue as issue:
            return issue
        except Exception as ex:
            _logger.exception(ex)
            return CheckFail("Check failed when processing: {}".format(ex))

    def _run(self, env):
        raise NotImplementedError("Should be overriden by the subclass")