hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2026-09-03 07:30:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed tiny bugs with authentication, added more unit tests

Browse Source
This commit is contained in:
Lonami
2016-09-04 21:07:09 +02:00
parent 7c8c65560e
commit b027dd2c8f
11 changed files with 98 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tl/__init__.py
View File

@@ -1,5 +1,9 @@
from .all_tlobjects import tlobjects
from .session import Session
from .mtproto_request import MTProtoRequest
from .telegram_client import TelegramClient
import os
# Only import most stuff if the TLObjects were generated
if os.path.isfile('tl/all_tlobjects.py'):
from .all_tlobjects import tlobjects
from .session import Session
from .mtproto_request import MTProtoRequest
from .telegram_client import TelegramClient
del os
from .tlobject import TLObject, TLArg

13
tl/session.py
View File

@@ -4,6 +4,7 @@ from os.path import isfile as file_exists
import time
import pickle
import utils
import random
class Session:
@@ -38,9 +39,13 @@ class Session:
def get_new_msg_id(self):
"""Generates a new message ID based on the current time (in ms) since epoch"""
# Refer to mtproto_plain_sender.py for the original method, this is a simple copy
new_msg_id = int(self.time_offset + time.time() * 1000)
if self.last_message_id >= new_msg_id:
new_msg_id = self._last_msg_id + 4
ms_time = int(time.time() * 1000)
new_msg_id = (((ms_time // 1000 + self.time_offset) << 32) | # "must approximately equal unixtime*2^32"
((ms_time % 1000) << 22) | # "approximate moment in time the message was created"
random.randint(0, 524288) << 2) # "message identifiers are divisible by 4"
self._last_msg_id = new_msg_id
if self.last_message_id >= new_msg_id:
new_msg_id = self.last_message_id + 4
self.last_message_id = new_msg_id
return new_msg_id