Selenium中精确操作组合框元素的实践指南

"//*[contains(@class, 'Ax4B8 ZAGvjd')]"

'//div[@class="M52nVb ytPNkd"]//input[@class="Ax4B8 ZAGvjd"]'

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import undetected_chromedriver as uc # 假设使用undetected_chromedriver
from time import sleep
import argparse

class Google:
    def __init__(self) -> None:
        # 使用undetected_chromedriver以避免被检测
        self.driver = uc.Chrome()
        self.driver.delete_all_cookies()
        self.wait_time = 20 # 显式等待时间

    def login_and_goto_google_finance(self, email, password):
        self.driver.get("https://accounts.google.com")
        WebDriverWait(self.driver, self.wait_time) \
            .until(EC.visibility_of_element_located((By.NAME, 'identifier'))) \
            .send_keys(f'{email}' + Keys.ENTER)
        WebDriverWait(self.driver, self.wait_time) \
            .until(EC.visibility_of_element_located((By.NAME, 'Passwd'))) \
            .send_keys(f'{password}' + Keys.ENTER)

    def navigate_to_site(self, url):
        self.driver.get(url)
        # 等待页面加载并点击“投资”或其他相关元素
        WebDriverWait(self.driver, self.wait_time).until(
            EC.element_to_be_clickable((By.XPATH, '//span[text()="Investment"]'))
        ).click()

        # 调用enter_symbol方法
        self.enter_symbol("BP", 1, "20240201", 100) # 示例数据
        sleep(5) # 适当的等待，以便观察结果

    def enter_symbol(self, symbol_name, qty, date, price):
        try:
            # 使用更精确的XPath定位器
            stock_input_element = WebDriverWait(self.driver, self.wait_time).until(
                EC.element_to_be_clickable((By.XPATH, '//div[@class="M52nVb ytPNkd"]//input[@class="Ax4B8 ZAGvjd"]'))
            )
            print(f"找到元素，aria_role: {stock_input_element.get_attribute('aria-role')}")

            # 清除可能存在的默认值或残留内容
            stock_input_element.clear() 
            # 发送股票代码并回车
            stock_input_element.send_keys(f'{symbol_name}' + Keys.ENTER)

            # 这里可以添加逻辑来输入数量、日期和价格，这通常涉及定位其他输入框
            # 例如：
            # qty_input = self.driver.find_element(By.XPATH, 'xpath_for_qty_input')
            # qty_input.send_keys(str(qty))
            # ...

        except Exception as e:
            print(f"在输入股票代码时发生错误: {e}")
        finally:
            sleep(5) # 确保有足够时间观察结果或进行下一步操作

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-u", "--username", type=str, help="Email Id for logging in to Google", required=True)
    parser.add_argument("-p", "--password", type=str, help="Password for logging in to Google", required=True)
    args = parser.parse_args()

    google = Google()
    try:
        google.login_and_goto_google_finance(args.username, args.password)
        # 替换为您的Google Finance投资组合链接
        google.navigate_to_site("https://www.google.com/finance/portfolio/view?id=") 
    finally:
        sleep(10) # 保持浏览器打开一段时间以便检查
        google.driver.quit()