hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-09 13:29:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make generate_sequence() thread-safe and move it to Session

Browse Source
This commit is contained in:
Lonami Exo
2017-06-07 12:00:01 +01:00
parent 49ca5c00c7
commit 1860054ec0
2 changed files with 36 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
telethon/network/mtproto_sender.py
View File

@@ -39,16 +39,6 @@ class MtProtoSender:
"""Disconnects from the server"""
self._transport.close()
def _generate_sequence(self, confirmed):
"""Generates the next sequence number, based on whether it
was confirmed yet or not"""
if confirmed:
result = self.session.sequence * 2 + 1
self.session.sequence += 1
return result
else:
return self.session.sequence * 2
# region Send and receive
def send(self, request):
@@ -142,7 +132,9 @@ class MtProtoSender:
plain_writer.write_long(self.session.salt, signed=False)
plain_writer.write_long(self.session.id, signed=False)
plain_writer.write_long(request.msg_id)
plain_writer.write_int(self._generate_sequence(request.confirmed))
plain_writer.write_int(
self.session.generate_sequence(request.confirmed))
plain_writer.write_int(len(packet))
plain_writer.write(packet)