Create a centralized Connection class, replaces TcpTransport (#112)

This commit is contained in:
Lonami Exo
2017-08-28 21:23:31 +02:00
parent bc72e52834
commit fa22a3f848
8 changed files with 143 additions and 138 deletions

View File

@@ -10,7 +10,7 @@ from . import helpers as utils
from .errors import (
RPCError, FloodWaitError, FileMigrateError, TypeNotFoundError
)
from .network import authenticator, MtProtoSender, TcpTransport
from .network import authenticator, MtProtoSender, Connection
from .utils import get_appropriated_part_size
from .crypto import rsa, CdnDecrypter
@@ -117,19 +117,19 @@ class TelegramBareClient:
# If ping failed, ensure we're disconnected first
self.disconnect()
transport = TcpTransport(self.session.server_address,
self.session.port,
proxy=self.proxy,
timeout=self._timeout)
connection = Connection(
self.session.server_address, self.session.port,
proxy=self.proxy, timeout=self._timeout
)
try:
if not self.session.auth_key:
self.session.auth_key, self.session.time_offset = \
authenticator.do_authentication(transport)
authenticator.do_authentication(connection)
self.session.save()
self._sender = MtProtoSender(transport, self.session)
self._sender = MtProtoSender(connection, self.session)
self._sender.connect()
# Now it's time to send an InitConnectionRequest