mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2025-06-19 11:36:41 +00:00
148 lines
3.5 KiB
Python
148 lines
3.5 KiB
Python
from abc import ABC, abstractmethod
|
|
import time
|
|
import struct
|
|
import os
|
|
|
|
|
|
class Session(ABC):
|
|
def __init__(self):
|
|
self.id = struct.unpack('q', os.urandom(8))[0]
|
|
|
|
self._sequence = 0
|
|
self._last_msg_id = 0
|
|
self._time_offset = 0
|
|
self._salt = 0
|
|
self._report_errors = True
|
|
self._flood_sleep_threshold = 60
|
|
|
|
def clone(self, to_instance=None):
|
|
cloned = to_instance or self.__class__()
|
|
cloned._report_errors = self.report_errors
|
|
cloned._flood_sleep_threshold = self.flood_sleep_threshold
|
|
return cloned
|
|
|
|
@abstractmethod
|
|
def set_dc(self, dc_id, server_address, port):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
@abstractmethod
|
|
def server_address(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
@abstractmethod
|
|
def port(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
@abstractmethod
|
|
def auth_key(self):
|
|
raise NotImplementedError
|
|
|
|
@auth_key.setter
|
|
@abstractmethod
|
|
def auth_key(self, value):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def close(self):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def save(self):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def delete(self):
|
|
raise NotImplementedError
|
|
|
|
@classmethod
|
|
@abstractmethod
|
|
def list_sessions(cls):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def process_entities(self, tlo):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def get_input_entity(self, key):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def cache_file(self, md5_digest, file_size, instance):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def get_file(self, md5_digest, file_size, cls):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def salt(self):
|
|
return self._salt
|
|
|
|
@salt.setter
|
|
def salt(self, value):
|
|
self._salt = value
|
|
|
|
@property
|
|
def report_errors(self):
|
|
return self._report_errors
|
|
|
|
@report_errors.setter
|
|
def report_errors(self, value):
|
|
self._report_errors = value
|
|
|
|
@property
|
|
def time_offset(self):
|
|
return self._time_offset
|
|
|
|
@time_offset.setter
|
|
def time_offset(self, value):
|
|
self._time_offset = value
|
|
|
|
@property
|
|
def flood_sleep_threshold(self):
|
|
return self._flood_sleep_threshold
|
|
|
|
@flood_sleep_threshold.setter
|
|
def flood_sleep_threshold(self, value):
|
|
self._flood_sleep_threshold = value
|
|
|
|
@property
|
|
def sequence(self):
|
|
return self._sequence
|
|
|
|
@sequence.setter
|
|
def sequence(self, value):
|
|
self._sequence = value
|
|
|
|
def get_new_msg_id(self):
|
|
"""Generates a new unique message ID based on the current
|
|
time (in ms) since epoch"""
|
|
now = time.time() + self._time_offset
|
|
nanoseconds = int((now - int(now)) * 1e+9)
|
|
new_msg_id = (int(now) << 32) | (nanoseconds << 2)
|
|
|
|
if self._last_msg_id >= new_msg_id:
|
|
new_msg_id = self._last_msg_id + 4
|
|
|
|
self._last_msg_id = new_msg_id
|
|
|
|
return new_msg_id
|
|
|
|
def update_time_offset(self, correct_msg_id):
|
|
now = int(time.time())
|
|
correct = correct_msg_id >> 32
|
|
self._time_offset = correct - now
|
|
self._last_msg_id = 0
|
|
|
|
def generate_sequence(self, content_related):
|
|
if content_related:
|
|
result = self._sequence * 2 + 1
|
|
self._sequence += 1
|
|
return result
|
|
else:
|
|
return self._sequence * 2
|