Selenium自动化测试 让浏览器自己动起来好神奇呀

今天我要和小伙伴们分享一个超级有趣的Python技能——Selenium自动化测试！想象一下，你的浏览器可以自己打开网页、点击按钮、填写表单，就像有了魔法一样~这个强大的工具不仅可以帮我们测试网页，还能完成很多自动化的任务，比如自动签到、自动抢票等等。让我们一起来探索这个神奇的世界吧！

准备工作

首先要做好准备工作呢，我们需要：

1. 安装Selenium库：

pip install selenium

2. 安装浏览器驱动（这里以Chrome为例）：

# 现在新版本的selenium已经不需要手动下载驱动啦

from selenium import webdriver

from selenium.webdriver.chrome.service import Service as ChromeService

from selenium.webdriver.common.by import By

from selenium.webdriver.common.keys import Keys

启动浏览器

让我们先来打开一个浏览器窗口吧~

# 创建浏览器实例

browser = webdriver.Chrome()

# 访问网页

browser.get("https://www.python.org")

小贴士：第一次运行时可能会稍微慢一些，因为要启动浏览器哦！不要着急呢~

寻找元素

浏览器打开后，我们要学会和页面元素互动。就像在寻宝一样，我们有很多方式可以找到想要的元素：

# 通过ID找元素

search_bar = browser.find_element(By.ID, "id-search-field")



# 通过类名找元素

submit_button = browser.find_element(By.CLASS_NAME, "search-button")



# 通过CSS选择器找元素

menu_items = browser.find_elements(By.CSS_SELECTOR, "li.tier-1")

注意事项：如果元素没找到，selenium会抛出NoSuchElementException异常，我们可以这样处理：

from selenium.common.exceptions import NoSuchElementException



try:

    element = browser.find_element(By.ID, "not-exist")

except NoSuchElementException:

    print("哎呀，没找到这个元素呢~")

模拟用户操作

接下来是最有趣的部分啦！我们可以模拟各种用户操作：

# 在搜索框输入文字

search_bar = browser.find_element(By.ID, "id-search-field")

search_bar.send_keys("python")

search_bar.send_keys(Keys.RETURN)  # 按回车键



# 点击按钮

button = browser.find_element(By.CLASS_NAME, "button")

button.click()



# 等待一下下~

from selenium.webdriver.support.ui import WebDriverWait

from selenium.webdriver.support import expected_conditions as EC



# 最多等待10秒，直到元素出现

element = WebDriverWait(browser, 10).until(

    EC.presence_of_element_located((By.ID, "result"))

)

实战小例子

让我们来做一个可爱的小例子，自动登录一个网站：

def login_website(username, password):

    browser = webdriver.Chrome()

    try:

        # 打开登录页面

        browser.get("https://example.com/login")

        

        # 输入用户名和密码

        username_input = browser.find_element(By.NAME, "username")

        password_input = browser.find_element(By.NAME, "password")

        

        username_input.send_keys(username)

        password_input.send_keys(password)

        

        # 点击登录按钮

        login_button = browser.find_element(By.ID, "login-button")

        login_button.click()

        

        # 等待登录成功

        WebDriverWait(browser, 10).until(

            EC.presence_of_element_located((By.CLASS_NAME, "welcome-message"))

        )

        print("登录成功啦！")

        

    except Exception as e:

        print(f"哎呀，出错了呢：{str(e)}")

    finally:

        browser.quit()  # 记得关闭浏览器哦~

小贴士：在实际项目中，记得把用户名和密码保存在配置文件中，不要直接写在代码里呢~

进阶小技巧

1. 截图功能：

browser.save_screenshot("screenshot.png")

2. 切换窗口：

# 切换到新打开的窗口

browser.switch_to.window(browser.window_handles[-1])

3. 处理弹窗：

# 接受弹窗

alert = browser.switch_to.alert

alert.accept()

小伙伴们，今天的Python学习之旅就到这里啦！记得动手敲代码，有问题随时在评论区问我哦。祝大家学习愉快，Python学习节节高！记住一定要多练习，把这些可爱的代码都试一试，这样才能真正掌握呢！下次见啦~