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

# -*- coding: utf-8 -*-

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')