eventhandler.py

import imp
import json
import os

_warnings = []

class EventHandler:
    def on_pr_opened(self, api, payload):
        pass

    def on_pr_updated(self, api, payload):
        pass

    def on_new_comment(self, api, payload):
        pass

    def warn(self, msg):
        global _warnings
        _warnings += [msg]

    def is_open_pr(self, payload):
        return payload['issue']['state'] == 'open' and 'pull_request' in payload['issue']

    def register_tests(self, path):
        from test import create_test
        tests_location = os.path.join(path, 'tests')
        if not os.path.isdir(tests_location):
            return
        tests = [os.path.join(tests_location, f) for f in os.listdir(tests_location) if f.endswith('.json')]
        for testfile in tests:
            with open(testfile) as f:
                contents = json.load(f)
                if not isinstance(contents['initial'], list):
                    assert not isinstance(contents['expected'], list)
                    contents['initial'] = [contents['initial']]
                    contents['expected'] = [contents['expected']]
                for initial, expected in zip(contents['initial'], contents['expected']):
                    yield create_test(testfile, initial, expected)

def reset_test_state():
    global _warnings
    _warnings = []

def get_warnings():
    global _warnings
    return _warnings

def get_handlers():
    modules = []
    handlers = []
    possiblehandlers = os.listdir('handlers')
    for i in possiblehandlers:
        location = os.path.join('handlers', i)
        module = '__init__'
        if not os.path.isdir(location) or not module + ".py" in os.listdir(location):
            continue
        try:
            (file, pathname, description) = imp.find_module(module, [location])
            module = imp.load_module(module, file, pathname, description)
            handlers.append(module.handler_interface())
            modules.append((module, location))
        finally:
            file.close()
    return (modules, handlers)