diff --git a/src/music_kraken/utils/support_classes/hacking.py b/src/music_kraken/utils/support_classes/hacking.py new file mode 100644 index 0000000..1cb1d78 --- /dev/null +++ b/src/music_kraken/utils/support_classes/hacking.py @@ -0,0 +1,69 @@ +from types import FunctionType +from functools import wraps + +from typing import Dict + +class Lake: + def __init__(self): + self.redirects: Dict[int, int] = {} + self.id_to_object: Dict[int, object] = {} + + def get_real_object(self, db_object: object) -> object: + def _get_real_id(_id: int) -> int: + if _id in self.redirects: + return _get_real_id(self.redirects[_id]) + return _id + + _id = _get_real_id(id(db_object)) + if _id not in self.id_to_object: + self.add(db_object) + + return self.id_to_object[_id] + + def add(self, db_object: object): + self.id_to_object[id(db_object)] = db_object + + def override(self, to_override: object, new_db_object: object): + self.redirects[id(to_override)] = id(new_db_object) + del self.id_to_object[id(to_override)] + + +lake = Lake() + + +def wrapper(method): + @wraps(method) + def wrapped(*args, **kwargs): + if len(args) >= 0 and method.__name__ != "__init__": + _self = lake.get_real_object(args[0]) + args = (_self, *args[1:]) + + return method(*args, **kwargs) + return wrapped + + + +class BaseClass: + def merge(self, to_replace): + lake.override(to_replace, self) + + +class MetaClass(type): + def __new__(meta, classname, bases, classDict): + bases = (*bases, BaseClass) + newClassDict = {} + + for attributeName, attribute in classDict.items(): + if isinstance(attribute, FunctionType) and attributeName not in ("__new__", "__init__"): + attribute = wrapper(attribute) + newClassDict[attributeName] = attribute + + for key, value in object.__dict__.items( ): + if hasattr( value, '__call__' ) and value not in newClassDict and key not in ("__new__", "__repr__", "__init__"): + newClassDict[key] = wrapper(value) + + new_instance = type.__new__(meta, classname, bases, newClassDict) + + lake.add(new_instance) + + return new_instance \ No newline at end of file