PaiGram/utils/baseobject.py

104 lines
3.3 KiB
Python
Raw Normal View History

2022-07-26 10:07:31 +00:00
import json
from copy import deepcopy
from typing import Dict, Union, Optional, List
from utils.typedefs import JSONDict
2022-07-26 10:07:31 +00:00
class BaseObject:
"""
大多数数据对象的基础类型
"""
def __new__(cls, *args: object, **kwargs: object) -> "BaseObject":
instance = super().__new__(cls)
return instance
def __str__(self) -> str:
return str(self.to_dict())
def __getitem__(self, item: str) -> object:
try:
return getattr(self, item)
except AttributeError as exc:
raise KeyError(
f"Objects of type {self.__class__.__name__} don't have an attribute called "
f"`{item}`."
) from exc
def __getstate__(self) -> Dict[str, Union[str, object]]:
return self._get_attrs(include_private=True, recursive=False)
def __setstate__(self, state: dict) -> None:
for key, val in state.items():
setattr(self, key, val)
def __deepcopy__(self, memodict: dict = None):
if memodict is None:
memodict = {}
cls = self.__class__
result = cls.__new__(cls) # 创建新实例
attrs = self._get_attrs(include_private=True) # 获取其所有属性
for k in attrs: # 在DeepCopy对象中设置属性
setattr(result, k, deepcopy(getattr(self, k), memodict))
return result
# 添加插槽可减少内存使用,并允许更快的属性访问
__slots__ = ()
def _get_attrs(self, include_private: bool = False, recursive: bool = False, ) -> Dict[str, Union[str, object]]:
data = {}
if not recursive:
try:
# __dict__ 具有来自超类的属性因此无需在下面的for循环中输入
data.update(self.__dict__)
except AttributeError:
pass
# 我们希望使用self获取类的所有属性但如果使用 self.__slots__ ,仅包括该类本身使用的属性,而不是它的超类
# 因此我们得到它的MRO然后再得到它们的属性
# 使用“[-1]”切片排除了“object”类
for cls in self.__class__.__mro__[:-1]:
for key in cls.__slots__: # 忽略 属性已定义
if not include_private and key.startswith("_"):
continue
value = getattr(self, key, None)
if value is not None:
if recursive and hasattr(value, "to_dict"):
data[key] = value.to_dict()
else:
data[key] = value
elif not recursive:
data[key] = value
return data
@staticmethod
def _parse_data(data: Optional[JSONDict]) -> Optional[JSONDict]:
return None if data is None else data.copy()
@classmethod
def de_json(cls, data: Optional[JSONDict]):
data = cls._parse_data(data)
if data is None:
return None
if cls == BaseObject:
return cls()
return cls(**data)
@classmethod
def de_list(cls, data: Optional[List[JSONDict]]) -> List:
if not data:
return []
return [cls.de_json(d) for d in data]
def to_json(self) -> str:
return json.dumps(self.to_dict())
def to_dict(self) -> JSONDict:
return self._get_attrs(recursive=True)