Python Selenium高效登录X (Twitter) 教程

import time
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException, NoSuchElementException

class XLoginAutomation:
    def __init__(self, username, password, driver_path=None):
        """
        初始化WebDriver。
        :param username: X（Twitter）账号（邮箱或用户名）。
        :param password: X（Twitter）密码。
        :param driver_path: ChromeDriver的路径，如果已添加到系统PATH则无需提供。
        """
        options = webdriver.ChromeOptions()
        # 可以添加无头模式以在后台运行浏览器
        # options.add_argument('--headless')
        # options.add_argument('--disable-gpu')
        # options.add_argument('--no-sandbox')

        if driver_path:
            self.driver = webdriver.Chrome(executable_path=driver_path, options=options)
        else:
            self.driver = webdriver.Chrome(options=options)

        self.wait = WebDriverWait(self.driver, 30) # 增加等待时间，以应对网络延迟或页面加载慢
        self.username = username
        self.password = password

    def login(self):
        """
        执行X（Twitter）的自动化登录流程。
        """
        try:
            print("导航到X（Twitter）的登录页面...")
            self.driver.get('https://twitter.com/i/flow/login')

            # 1. 定位并输入用户名
            print("正在定位用户名输入框...")
            username_field = self.wait.until(
                EC.presence_of_element_located((By.CSS_SELECTOR, 'input[autocomplete="username"]'))
            )
            username_field.send_keys(self.username)
            print("用户名输入成功。")
            time.sleep(1) # 短暂等待，模拟用户操作

            # 2. 定位并点击“下一步”按钮
            print("正在定位并点击'下一步'按钮...")
            # 这里的选择器可能需要根据X（Twitter）页面的最新HTML结构进行调整
            # 尝试使用data-testid属性，它通常更稳定
            try:
                next_button = self.wait.until(
                    EC.element_to_be_clickable((By.CSS_SELECTOR, '[role="button"][data-testid*="nextButton"], [role="button"].r-13qz1uu'))
                )
            except TimeoutException:
                # 如果没有“下一步”按钮（例如直接进入密码输入），则尝试跳过
                print("未找到'下一步'按钮，尝试直接查找密码输入框。")
                pass # 可能会直接进入密码输入阶段，取决于X的登录流程
            else:
                next_button.click()
                print("'下一步'按钮点击成功。")
                time.sleep(2) # 等待页面跳转或元素加载

            # 3. 定位并输入密码
            print("正在定位密码输入框...")
            password_field = self.wait.until(
                EC.presence_of_element_located((By.CSS_SELECTOR, 'input[type="password"]'))
            )
            password_field.send_keys(self.password)
            print("密码输入成功。")
            time.sleep(1)

            # 4. 定位并点击“登录”按钮
            print("正在定位并点击'登录'按钮...")
            login_button = self.wait.until(
                EC.element_to_be_clickable((By.CSS_SELECTOR, '[data-testid*="Login_Button"]'))
            )
            login_button.click()
            print("'登录'按钮点击成功。")
            time.sleep(5) # 等待登录完成和页面跳转

            # 5. 验证登录是否成功
            print("正在验证登录状态...")
            # 检查登录后页面上特有的元素，例如私信链接或主页时间线
            try:
                self.wait.until(
                    EC.presence_of_element_located((By.CSS_SELECTOR, '[data-testid="AppTabBar_DirectMessage_Link"], [aria-label="时间线：主页"]'))
                )
                print("登录成功！")
                return True
            except TimeoutException:
                print("登录失败或验证元素未找到。")
                return False

        except TimeoutException as e:
            print(f"操作超时：{e}")
            print("请检查网络连接、元素选择器是否正确，或X（Twitter）页面结构是否发生变化。")
            return False
        except NoSuchElementException as e:
            print(f"未找到元素：{e}")
            print("请检查元素选择器是否正确。")
            return False
        except Exception as e:
            print(f"发生未知错误：{e}")
            return False
        finally:
            # 登录成功后，可以保持浏览器打开一段时间或手动关闭
            # self.driver.quit() # 如果不需要后续操作，可以关闭浏览器
            pass

    def close_browser(self):
        """关闭WebDriver实例。"""
        if self.driver:
            self.driver.quit()

# 示例用法
if __name__ == "__main__":
    # 请替换为你的X（Twitter）账号和密码
    # 强烈建议将敏感信息从代码中分离，例如通过环境变量或配置文件读取
    X_EMAIL = "your_email_or_username"
    X_PASSWORD = "your_password"

    # 如果ChromeDriver不在系统PATH中，请指定其路径
    # CHROME_DRIVER_PATH = "/path/to/your/chromedriver" 
    CHROME_DRIVER_PATH = None # 如果已添加到PATH，则设置为None

    if X_EMAIL == "your_email_or_username" or X_PASSWORD == "your_password":
        print("警告：请在代码中替换为您的X（Twitter）账号和密码，或通过环境变量设置。")
    else:
        bot = XLoginAutomation(X_EMAIL, X_PASSWORD, driver_path=CHROME_DRIVER_PATH)
        try:
            if bot.login():
                print("自动化登录流程执行完毕。您已成功登录X（Twitter）。")
                # 可以在这里添加其他自动化操作，例如发推文、浏览等
                # 例如：time.sleep(10) # 保持浏览器打开10秒以便观察
            else:
                print("自动化登录流程未能成功完成。")
        finally:
            # 确保在脚本结束时关闭浏览器
            # input("按Enter键关闭浏览器...") # 如果想手动关闭
            bot.close_browser()
            print("浏览器已关闭。")