From a91109c9faefec74bf38aa212420a9f04bcac5a9 Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Wed, 13 Jun 2018 10:55:37 +0200 Subject: [PATCH] Retry send_code_request on AuthRestartError --- telethon/client/auth.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/telethon/client/auth.py b/telethon/client/auth.py index 10913ad6..0aba1d5a 100644 --- a/telethon/client/auth.py +++ b/telethon/client/auth.py @@ -311,8 +311,12 @@ class AuthMethods(MessageParseMethods, UserMethods): phone_hash = self._phone_code_hash.get(phone) if not phone_hash: - result = await self(functions.auth.SendCodeRequest( - phone, self.api_id, self.api_hash)) + try: + result = await self(functions.auth.SendCodeRequest( + phone, self.api_id, self.api_hash)) + except errors.AuthRestartError: + return self.send_code_request(phone, force_sms=force_sms) + self._tos = result.terms_of_service self._phone_code_hash[phone] = phone_hash = result.phone_code_hash else: