You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.5 KiB
49 lines
1.5 KiB
# -*- 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')
|