fix: merging and replacing instances
This commit is contained in:
parent
cae2ecffcb
commit
0db00f934b
69
src/music_kraken/utils/support_classes/hacking.py
Normal file
69
src/music_kraken/utils/support_classes/hacking.py
Normal file
@ -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
|
Loading…
Reference in New Issue
Block a user