From fa1028560cc77c55bb4379e92602bec513f525e5 Mon Sep 17 00:00:00 2001 From: strannik-j Date: Wed, 19 Jun 2019 10:49:08 +0300 Subject: [PATCH] Add Two-Step Verification --- download_service.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/download_service.py b/download_service.py index feb4b90..f565798 100644 --- a/download_service.py +++ b/download_service.py @@ -12,11 +12,13 @@ from telethon.tl.types import DocumentAttributeFilename from telethon.tl.types import Document from telethon.utils import get_input_media from telethon.errors.rpc_error_list import LocationInvalidError +from telethon.errors import SessionPasswordNeededError # from telegram_client_x import TelegramClientX from telethon.telegram_client import TelegramClient from telethon.tl.types import Message from tg_access import * from io import BytesIO +from getpass import getpass import sys @@ -68,7 +70,13 @@ client.connect() if not client.is_user_authorized(): phone = input('Enter phone: ') client.send_code_request(phone) - client.sign_in(phone, input('Enter code: ')) + try: + client.sign_in(phone, input('Enter code: ')) + except SessionPasswordNeededError: + pw = getpass('Two step verification is enabled. ' + 'Please enter your password: ') + client.sign_in(password=pw) + def on_download_progress(recv_bytes, total_bytes): global last_call_time_receive if time.time() - last_call_time_receive < 1: @@ -153,4 +161,4 @@ def main(argv): if __name__ == '__main__': import sys - main(sys.argv[0:]) \ No newline at end of file + main(sys.argv[0:])