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