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

# 设置Chrome选项
chrome_options = webdriver.ChromeOptions()
# 添加无头模式选项（可选）
# chrome_options.add_argument('--headless')
# 禁用自动化扩展提示
chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
# 隐藏自动化软件的指纹
chrome_options.add_experimental_option('useAutomationExtension', False)

# 初始化WebDriver
driver = webdriver.Chrome(options=chrome_options)
driver.execute_cdp_cmd("Page.addScriptToEvaluateOnNewDocument", {
    "source": """
    Object.defineProperty(navigator, 'webdriver', {
      get: () => undefined
    })
    """
})

# 打开登录页面
driver.get("https://fxg.jinritemai.com/login/common")

# 等待用户手动输入账号和验证码
print("请手动输入账号和验证码，完成登录后按Enter键继续...")
input()

# 用户登录后，打开对话页面
driver.get("https://im.jinritemai.com/pc_seller_v2/main/workspace")

try:
    # 等待特定元素加载并获取元素
    WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.CLASS_NAME, "messageList"))  # 替换为实际的类名
    )

    # 获取所有具有相同类名的元素
    elements = driver.find_elements(By.CLASS_NAME, "messageList")

    # 打印每个元素的HTML内容
    for element in elements:
        print(element.get_attribute('outerHTML'))

finally:
    # 关闭浏览器
    driver.quit()