MTPyroger/pyrogram/raw/core/tl_object.py
2021-01-01 22:58:48 +01:00

95 lines
2.9 KiB
Python

# Pyrogram - Telegram MTProto API Client Library for Python
# Copyright (C) 2017-2021 Dan <https://github.com/delivrance>
#
# This file is part of Pyrogram.
#
# Pyrogram is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pyrogram is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
from io import BytesIO
from json import dumps
from typing import cast, List, Any, Union, Dict
from ..all import objects
class TLObject:
__slots__: List[str] = []
QUALNAME = "Base"
@classmethod
def read(cls, data: BytesIO, *args: Any) -> Any:
try:
return cast(TLObject, objects[int.from_bytes(data.read(4), "little")]).read(data, *args)
except KeyError as e:
left = data.read().hex()
left = [left[i:i + 64] for i in range(0, len(left), 64)]
left = [[left[i:i + 8] for i in range(0, len(left), 8)] for left in left]
left = "\n".join(" ".join(x for x in left) for left in left)
raise ValueError(f"Unknown constructor found: {hex(e.args[0])}\n{left}")
def write(self, *args: Any) -> bytes:
pass
@staticmethod
def default(obj: "TLObject") -> Union[str, Dict[str, str]]:
if isinstance(obj, bytes):
return repr(obj)
return {
"_": obj.QUALNAME,
**{
attr: getattr(obj, attr)
for attr in obj.__slots__
if getattr(obj, attr) is not None
}
}
def __str__(self) -> str:
return dumps(self, indent=4, default=TLObject.default, ensure_ascii=False)
def __repr__(self) -> str:
return "pyrogram.raw.{}({})".format(
self.QUALNAME,
", ".join(
f"{attr}={repr(getattr(self, attr))}"
for attr in self.__slots__
if getattr(self, attr) is not None
)
)
def __eq__(self, other: Any) -> bool:
for attr in self.__slots__:
try:
if getattr(self, attr) != getattr(other, attr):
return False
except AttributeError:
return False
return True
def __len__(self) -> int:
return len(self.write())
def __getitem__(self, item: Any) -> Any:
return getattr(self, item)
def __setitem__(self, key: Any, value: Any) -> Any:
setattr(self, key, value)
def __call__(self, *args: Any, **kwargs: Any) -> Any:
pass