python web自动化 selenium 登录

今天终于开始了web自动化入门，按照系统操作的先后顺序，先从登录开始写起，话不多说，附代码：

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


# 定义登录函数
def login(url, username, password):
    # 创建 Chrome 浏览器驱动对象
    driver = webdriver.Chrome()
    driver.maximize_window()  # 是浏览器全屏

    # 访问登录页面
    driver.get(url)

    # 使用find_element_by_xpath方法通过 XPath 定位具有特定 placeholder 的用户名输入框元素
    username_field = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.CSS_SELECTOR, "input[autocomplete='off'][placeholder='请输入']"))
    )
    print(username_field)

    # 使用find_element_by_xpath方法通过 XPath 定位具有特定 placeholder 的密码输入框元素
    password_field = driver.find_element(By.XPATH, "//input[@placeholder='请输入密码']")

    # 定位登录按钮元素
    login_button = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.CSS_SELECTOR, "button[data-v-f96ef2e4].el-button.login-btn.el-button--primary.el-button--small")))
    print(login_button)

    # 找到用户名输入框并填入用户名
    username_field.send_keys(username)

    # 找到密码输入框并填入密码
    password_field.send_keys(password)

    # 找到登录按钮并点击
    login_button.click()

    # 等待页面加载完成
    # WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "success-message")))
    time.sleep(20)
    # # 检查是否显示登录成功的消息
    # success_message = driver.find_element(By.ID, "success-message")
    # if success_message.is_displayed():
    #     print("登录成功！")
    # else:
    #     print("登录失败或出现异常。")
    #     driver.quit()


# 调用URL
url = "1yourUrl"
username = "yourusername"
password = "yourpassword"
login(url, username, password)

看似简单的几行代码，其实在选择元素定位方法时，把我绕晕了，对页面元素还是不熟，继续加油。