### Python 实现接口自动化测试中的验证码登录 #### 使用 `requests` 库处理验证码登录 对于通过 API 进行的自动化测试，通常可以通过发送 HTTP 请求来模拟用户的交互行为。下面是一个具体的例子，展示如何利用 `LoginAPI` 类完成验证码获取和登录操作。 ```python import requests class LoginAPI: url_verify = 'http://kdtx-test.itheima.net/api/captchaImage' url_login = "http://kdtx-test.itheima.net/api/login" def __init__(self): pass # 获取验证码图片并保存到本地文件 def save_captcha_image(self, file_path='captcha.png'): response = requests.get(url=self.url_verify) with open(file_path, 'wb') as f: f.write(response.content) # 用户手动输入验证码后尝试登录 def manual_login_with_captcha(self, username, password, captcha_text): login_data = { "username": username, "password": password, "code": captcha_text } return requests.post(url=self.url_login, json=login_data).json() ``` 此代码片段展示了两个主要的功能：一是下载验证码图像以便人工识别；二是提交包含用户名、密码以及由用户提供的验证码文本在内的表单数据来进行身份验证[^1]。 #### 结合Selenium绕过验证码的方法 当面对图形化的验证码时，另一种常见的做法是借助浏览器自动化工具如 Selenium 来捕获页面上的 Cookies 或者直接操纵网页元素以避开复杂的视觉挑战。这里给出一个简单的流程说明： 1. 启动 WebDriver 并导航至目标网站。 2. 手动或自动解析出必要的 cookies。 3. 将这些 cookies 添加到后续所有的请求头中去访问受保护资源。 ```python from selenium import webdriver from time import sleep driver = webdriver.Chrome() # 假设使用 Chrome 浏览器驱动程序 try: driver.get('https://example.com') # 等待加载完毕并且执行任何需要的人工干预（比如点击按钮） sleep(5) # 提取所有 cookie 数据 all_cookies = driver.get_cookies() finally: driver.quit() session = requests.Session() for cookie in all_cookies: session.cookies.set(cookie['name'], cookie['value']) response = session.get('https://protected-resource.example.com') print(response.text) ``` 这段脚本先启动了一个无界面模式下的 Chromium 浏览器实例，并打开了指定的目标站点，在适当的时间间隔之后收集到了当前会话所持有的全部 cookies 。接着创建一个新的 Requests Session 对象并将刚才获得的每一个 cookie 设置进去，从而使得接下来发出的所有 GET/POST 请求都能够携带有效的认证信息而无需再次经历繁琐的身份验证过程[^2]。