TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 20:59:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Log unknown constructors

Browse Source
This commit is contained in:
Dan 2018-08-24 17:39:55 +02:00
parent caecbef2f6
commit 0b6b598059
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pyrogram/api/core/object.py
View File

@ -16,6 +16,7 @@
# You should have received a copy of the GNU Lesser General Public License
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
import logging
from collections import OrderedDict
from datetime import datetime
from io import BytesIO
@ -23,13 +24,23 @@ from json import JSONEncoder, dumps
from ..all import objects
log = logging.getLogger(__name__)
class Object:
all = {}
@staticmethod
def read(b: BytesIO, *args):
return Object.all[int.from_bytes(b.read(4), "little")].read(b, *args)
constructor_id = int.from_bytes(b.read(4), "little")
try:
return Object.all[constructor_id].read(b, *args)
except KeyError:
log.error("Unknown constructor found: {}. Full data: {}".format(
hex(constructor_id),
b.getvalue().hex())
)
def write(self, *args) -> bytes:
pass