external/tgcloud
1
0
Fork 0
mirror of https://github.com/SlavikMIPT/tgcloud.git synced 2025-03-09 15:40:14 +00:00
Code Issues Releases Wiki Activity

Add Two-Step Verification

Browse source
This commit is contained in:
strannik-j 2019-06-19 10:49:08 +03:00 • committed by GitHub
parent c59a982d03
commit fa1028560c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
download_service.py
View file

@ -12,11 +12,13 @@ from telethon.tl.types import DocumentAttributeFilename
from telethon.tl.types import Document from telethon.tl.types import Document
from telethon.utils import get_input_media from telethon.utils import get_input_media
from telethon.errors.rpc_error_list import LocationInvalidError from telethon.errors.rpc_error_list import LocationInvalidError
from telethon.errors import SessionPasswordNeededError
# from telegram_client_x import TelegramClientX # from telegram_client_x import TelegramClientX
from telethon.telegram_client import TelegramClient from telethon.telegram_client import TelegramClient
from telethon.tl.types import Message from telethon.tl.types import Message
from tg_access import * from tg_access import *
from io import BytesIO from io import BytesIO
from getpass import getpass
import sys import sys
@ -68,7 +70,13 @@ client.connect()
if not client.is_user_authorized(): if not client.is_user_authorized():
phone = input('Enter phone: ') phone = input('Enter phone: ')
client.send_code_request(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): def on_download_progress(recv_bytes, total_bytes):
global last_call_time_receive global last_call_time_receive
if time.time() - last_call_time_receive < 1: if time.time() - last_call_time_receive < 1:
@ -153,4 +161,4 @@ def main(argv):
if __name__ == '__main__': if __name__ == '__main__':
import sys import sys
main(sys.argv[0:]) main(sys.argv[0:])