From 091b03c635e16d9cc5b22e2da494a31a85d859ef Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Thu, 14 Jun 2018 23:33:56 +0200 Subject: [PATCH] Use create_task instead ensure_future --- telethon/client/updates.py | 5 ++--- telethon/network/mtprotosender.py | 12 +++++------- 2 files changed, 7 insertions(+), 10 deletions(-) diff --git a/telethon/client/updates.py b/telethon/client/updates.py index 8d7adef9..d2e5dbde 100644 --- a/telethon/client/updates.py +++ b/telethon/client/updates.py @@ -150,13 +150,12 @@ class UpdateMethods(UserMethods): itertools.chain(update.users, update.chats)} for u in update.updates: u._entities = entities - asyncio.ensure_future( - self._dispatch_update(u), loop=self._loop) + self._loop.create_task(self._dispatch_update(u)) return if isinstance(update, types.UpdateShort): update = update.update update._entities = {} - asyncio.ensure_future(self._dispatch_update(update), loop=self._loop) + self._loop.create_task(self._dispatch_update(update)) async def _dispatch_update(self, update): if self._events_pending_resolve: diff --git a/telethon/network/mtprotosender.py b/telethon/network/mtprotosender.py index 2abe4c8d..58fa77a0 100644 --- a/telethon/network/mtprotosender.py +++ b/telethon/network/mtprotosender.py @@ -232,12 +232,10 @@ class MTProtoSender: raise _last_error __log__.debug('Starting send loop') - self._send_loop_handle = asyncio.ensure_future( - self._send_loop(), loop=self._loop) + self._send_loop_handle = self._loop.create_task(self._send_loop()) __log__.debug('Starting receive loop') - self._recv_loop_handle = asyncio.ensure_future( - self._recv_loop(), loop=self._loop) + self._recv_loop_handle = self._loop.create_task(self._recv_loop()) if self._is_first_query: __log__.debug('Running first query') @@ -352,11 +350,11 @@ class MTProtoSender: continue except ConnectionError as e: __log__.info('Connection reset while receiving %s', e) - asyncio.ensure_future(self._reconnect(), loop=self._loop) + self._loop.create_task(self._reconnect()) break except OSError as e: __log__.warning('OSError while receiving %s', e) - asyncio.ensure_future(self._reconnect(), loop=self._loop) + self._loop.create_task(self._reconnect()) break # TODO Check salt, session_id and sequence_number @@ -375,7 +373,7 @@ class MTProtoSender: # an actually broken authkey? __log__.warning('Broken authorization key?: {}'.format(e)) self.state.auth_key = None - asyncio.ensure_future(self._reconnect(), loop=self._loop) + self._loop.create_task(self._reconnect()) break except SecurityError as e: # A step while decoding had the incorrect data. This message