feat(session_string_generator): implement QR code and phone number login methods for Telegram session generation

session_string_generator.py

@@ -19,26 +19,89 @@ parameters support integer IDs, string representations of IDs (e.g., "123456"),
 and usernames (e.g., "@mychannel").
 """
 
+import asyncio
+import io
 import os
-from telethon.sync import TelegramClient
-from telethon.sessions import StringSession
-from dotenv import load_dotenv
 import sys
 
-# Load environment variables from .env file
+from dotenv import load_dotenv
+from telethon import errors
+from telethon.sessions import StringSession
+from telethon.sync import TelegramClient
+
 load_dotenv()
 
 
+def _qr_login(client: TelegramClient) -> None:
+    import qrcode
+
+    qr = client.qr_login()
+
+    print("\n----- QR Code Login -----\n")
+
+    qr_obj = qrcode.QRCode(border=1)
+    qr_obj.add_data(qr.url)
+    qr_obj.make(fit=True)
+    f = io.StringIO()
+    qr_obj.print_ascii(out=f, invert=True)
+    print(f.getvalue())
+
+    print("Scan the QR code above with your Telegram app:")
+    print("  Open Telegram > Settings > Devices > Link Desktop Device\n")
+    print(f"Or open this link on a device where you're logged in:\n  {qr.url}\n")
+    print(f"Expires at: {qr.expires.strftime('%H:%M:%S')}")
+    print("Waiting for you to scan...")
+
+    try:
+        client.loop.run_until_complete(qr.wait(timeout=120))
+    except asyncio.TimeoutError:
+        print("\nQR code expired. Please try again.")
+        client.disconnect()
+        sys.exit(1)
+    except errors.SessionPasswordNeededError:
+        pw = input("\nTwo-factor authentication enabled. Please enter your password: ")
+        client.sign_in(password=pw)
+
+
+def _phone_login(client: TelegramClient) -> None:
+    phone = input("Please enter your phone (or bot token): ")
+
+    try:
+        client.send_code_request(phone)
+    except errors.FloodWaitError as e:
+        print(
+            f"\nFlood wait error; you must wait {e.seconds} seconds before trying again."
+        )
+        client.disconnect()
+        sys.exit(1)
+    except errors.PhoneNumberInvalidError:
+        print("\nThe phone number is invalid.")
+        client.disconnect()
+        sys.exit(1)
+    except Exception as e:
+        print(f"\nError sending code: {e}")
+        client.disconnect()
+        sys.exit(1)
+
+    code = input("\nPlease enter the code you received: ")
+    try:
+        client.sign_in(phone, code)
+    except errors.SessionPasswordNeededError:
+        pw = input("Two-factor authentication enabled. Please enter your password: ")
+        client.sign_in(password=pw)
+
+
 def main() -> None:
     API_ID = os.getenv("TELEGRAM_API_ID")
     API_HASH = os.getenv("TELEGRAM_API_HASH")
 
     if not API_ID or not API_HASH:
         print("Error: TELEGRAM_API_ID and TELEGRAM_API_HASH must be set in .env file")
-        print("Create an .env file with your credentials from https://my.telegram.org/apps")
+        print(
+            "Create an .env file with your credentials from https://my.telegram.org/apps"
+        )
         sys.exit(1)
 
-    # Convert API_ID to integer
     try:
         API_ID = int(API_ID)
     except ValueError:
@@ -47,56 +110,62 @@ def main() -> None:
 
     print("\n----- Telegram Session String Generator -----\n")
     print("This script will generate a session string for your Telegram account.")
-    print(
-        "You will be asked to enter your phone number and the verification code sent to your Telegram app."
-    )
     print("The generated session string can be added to your .env file.")
     print(
         "\nYour credentials will NOT be stored on any server and are only used for local authentication.\n"
     )
 
+    print("Choose login method:")
+    print("  1) QR code login (recommended -- scan from your Telegram app)")
+    print("  2) Phone number + verification code")
+    method = input("\nEnter 1 or 2 [default: 1]: ").strip() or "1"
+
     try:
-        # Connect to Telegram and generate the session string
+        client = TelegramClient(StringSession(), API_ID, API_HASH)
-        with TelegramClient(StringSession(), API_ID, API_HASH) as client:
+        client.connect()
-            # The client.session.save() function from StringSession returns the session string
-            session_string = StringSession.save(client.session)
 
-            print("\nAuthentication successful!")
+        if not client.is_user_authorized():
-            print("\n----- Your Session String -----")
+            if method == "1":
-            print(f"\n{session_string}\n")
+                _qr_login(client)
-            print("Add this to your .env file as:")
+            else:
-            print(f"TELEGRAM_SESSION_STRING={session_string}")
+                _phone_login(client)
-            print("\nIMPORTANT: Keep this string private and never share it with anyone!")
 
-            # Optional: auto-update the .env file
+        session_string = StringSession.save(client.session)
-            choice = input(
-                "\nWould you like to automatically update your .env file with this session string? (y/N): "
-            )
-            if choice.lower() == "y":
-                try:
-                    # Read the current .env file
-                    with open(".env", "r") as file:
-                        env_contents = file.readlines()
 
-                    # Update or add the SESSION_STRING line
+        print("\nAuthentication successful!")
-                    session_string_line_found = False
+        print("\n----- Your Session String -----")
-                    for i, line in enumerate(env_contents):
+        print(f"\n{session_string}\n")
-                        if line.startswith("TELEGRAM_SESSION_STRING="):
+        print("Add this to your .env file as:")
-                            env_contents[i] = f"TELEGRAM_SESSION_STRING={session_string}\n"
+        print(f"TELEGRAM_SESSION_STRING={session_string}")
-                            session_string_line_found = True
+        print("\nIMPORTANT: Keep this string private and never share it with anyone!")
-                            break
 
-                    if not session_string_line_found:
+        choice = input(
-                        env_contents.append(f"TELEGRAM_SESSION_STRING={session_string}\n")
+            "\nWould you like to automatically update your .env file with this session string? (y/N): "
+        )
+        if choice.lower() == "y":
+            try:
+                with open(".env", "r") as file:
+                    env_contents = file.readlines()
 
-                    # Write back to the .env file
+                session_string_line_found = False
-                    with open(".env", "w") as file:
+                for i, line in enumerate(env_contents):
-                        file.writelines(env_contents)
+                    if line.startswith("TELEGRAM_SESSION_STRING="):
+                        env_contents[i] = f"TELEGRAM_SESSION_STRING={session_string}\n"
+                        session_string_line_found = True
+                        break
 
-                    print("\n.env file updated successfully!")
+                if not session_string_line_found:
-                except Exception as e:
+                    env_contents.append(f"TELEGRAM_SESSION_STRING={session_string}\n")
-                    print(f"\nError updating .env file: {e}")
+
-                    print("Please manually add the session string to your .env file.")
+                with open(".env", "w") as file:
+                    file.writelines(env_contents)
+
+                print("\n.env file updated successfully!")
+            except Exception as e:
+                print(f"\nError updating .env file: {e}")
+                print("Please manually add the session string to your .env file.")
+
+        client.disconnect()
 
     except Exception as e:
         print(f"\nError: {e}")