-
-
Notifications
You must be signed in to change notification settings - Fork 48
Expand file tree
/
Copy pathobject.py
More file actions
123 lines (97 loc) · 3.89 KB
/
object.py
File metadata and controls
123 lines (97 loc) · 3.89 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# Hydrogram - Telegram MTProto API Client Library for Python
# Copyright (C) 2017-2023 Dan <https://github.com/delivrance>
# Copyright (C) 2023-present Hydrogram <https://hydrogram.org>
#
# This file is part of Hydrogram.
#
# Hydrogram 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.
#
# Hydrogram 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 Hydrogram. If not, see <http://www.gnu.org/licenses/>.
import typing
from datetime import datetime
from enum import Enum
from json import dumps
if typing.TYPE_CHECKING:
import hydrogram
class Object:
def __init__(self, client: "hydrogram.Client" = None):
self._client = client
def bind(self, client: "hydrogram.Client"):
"""Bind a Client instance to this and to all nested Hydrogram objects.
Parameters:
client (:obj:`~hydrogram.types.Client`):
The Client instance to bind this object with. Useful to re-enable bound methods after serializing and
deserializing Hydrogram objects with ``repr`` and ``eval``.
"""
self._client = client
for i in self.__dict__:
o = getattr(self, i)
if isinstance(o, Object):
o.bind(client)
@staticmethod
def default(obj: "Object"):
if isinstance(obj, bytes):
return repr(obj)
# https://t.me/hydrogramchat/167281
# Instead of re.Match, which breaks for python <=3.6
if isinstance(obj, typing.Match):
return repr(obj)
if isinstance(obj, Enum):
return str(obj)
if isinstance(obj, datetime):
return str(obj)
return {
"_": obj.__class__.__name__,
**{
attr: ("*" * 9 if attr == "phone_number" else getattr(obj, attr))
for attr in filter(lambda x: not x.startswith("_"), obj.__dict__)
if getattr(obj, attr) is not None
},
}
def __str__(self) -> str:
return dumps(self, indent=4, default=Object.default, ensure_ascii=False)
def __repr__(self) -> str:
return "hydrogram.types.{}({})".format(
self.__class__.__name__,
", ".join(
f"{attr}={getattr(self, attr)!r}"
for attr in filter(lambda x: not x.startswith("_"), self.__dict__)
if getattr(self, attr) is not None
),
)
def __eq__(self, other: "Object") -> bool:
for attr in self.__dict__:
try:
if attr.startswith("_"):
continue
if getattr(self, attr) != getattr(other, attr):
return False
except AttributeError:
return False
return True
def __hash__(self):
return hash(tuple(sorted(self.__dict__.items())))
def __setstate__(self, state):
for attr in state:
obj = state[attr]
# Maybe a better alternative would be https://docs.python.org/3/library/inspect.html#inspect.signature
if isinstance(obj, tuple) and len(obj) == 2 and obj[0] == "dt":
state[attr] = datetime.fromtimestamp(obj[1])
self.__dict__ = state
def __getstate__(self):
state = self.__dict__.copy()
state.pop("_client", None)
for attr in state:
obj = state[attr]
if isinstance(obj, datetime):
state[attr] = ("dt", obj.timestamp())
return state