bitbot-3.11-fork/src/utils/decorators.py

import typing
from .parse import argument_spec

BITBOT_MAGIC = "__bitbot"

class BitBotMagic(object):
    def __init__(self):
        self._hooks: typing.List[typing.Tuple[str, dict]] = []
        self._kwargs: typing.List[typing.Tuple[str, typing.Any]] = []
        self._exports: typing.List[typing.Tuple[str, typing.Any]] = []
    def add_hook(self, hook: str, kwargs: dict):
        self._hooks.append((hook, kwargs))
    def add_kwarg(self, key: str, value: typing.Any):
        self._kwargs.append((key, value))

    def get_hooks(self):
        hooks: typing.List[typing.Tuple[str, typing.List[Tuple[str, typing.Any]]]] = []
        for hook, kwargs in self._hooks:
            hooks.append((hook, self._kwargs.copy()+list(kwargs.items())))
        return hooks

    def add_export(self, key: str, value: typing.Any):
        self._exports.append((key, value))
    def get_exports(self):
        return self._exports.copy()

def get_magic(obj: typing.Any):
    if not has_magic(obj):
        setattr(obj, BITBOT_MAGIC, BitBotMagic())
    return getattr(obj, BITBOT_MAGIC)
def has_magic(obj: typing.Any):
    return hasattr(obj, BITBOT_MAGIC)

def hook(event: str, **kwargs):
    def _hook_func(func):
        magic = get_magic(func)
        magic.add_hook(event, kwargs)
        return func
    return _hook_func
def export(setting: str, value: typing.Any):
    def _export_func(module):
        magic = get_magic(module)
        magic.add_export(setting, value)
        return module
    return _export_func

def _kwarg(key: str, value: typing.Any, func: typing.Any):
    magic = get_magic(func)
    magic.add_kwarg(key, value)
    return func

def kwarg(key: str, value: typing.Any):
    def _kwarg_func(func):
        return _kwarg(key, value, func)
    return _kwarg_func

def spec(spec: str):
    def _spec_func(func):
        return _kwarg("spec", argument_spec(spec), func)
    return _spec_func