pinhook/pinhook/plugin.py

cmds = []

class Output:
    def __init__(self, msg_type, msg):
        self.msg_type = msg_type
        self.msg = self.sanitize(msg)

    def sanitize(self, msg):
        return msg.splitlines()


def action(msg):
    return Output('action', msg)


def message(msg):
    return Output('message', msg)


def add_plugin(command, func):
    cmds.append({'cmd': command, 'func': func})


def clear_plugins():
    cmds.clear()


def register(command):
    def register_for_command(func):
        add_plugin(command, func)
        return func
    return register_for_command