最近看到了selenium自动化操作就想实现一下模拟登录
前面那里没什么难度,看到验证码就犯难,后面看到别人的思路通过打码平台获取验证码识别结果,通过图片的二维坐标依次点击图片。
通过动作链执行操作,代码如下:
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver import ActionChains
import content
import base64
import requests
import re
import time
class Login():
def __init__(self):
self.url = 'https://kyfw.12306.cn/otn/resources/login.html'
self.driver = webdriver.Chrome(r'C:/chromedriver.exe')
self.user = content.USER_NAME
self.pwd = content.PASSWORD
def open_page(self):
'''打开登录界面并自动输入账号和密码'''
self.driver.get(self.url)#打开界面
self.driver.maximize_window()
self.driver.implicitly_wait(10)#隐式等待10s 等浏览器加载完毕
self.driver.find_element_by_xpath('/html/body/div[2]/div[2]/ul/li[2]/a').click()#点击账号密码登录
self.driver.implicitly_wait(5)
self.driver.find_element_by_xpath('//*[@id="J-userName"]').send_keys(self.user)
self.driver.find_element_by_xpath('//*[@id="J-password"]').send_keys(self.pwd)
image_url = self.driver.find_element_by_xpath('//*[@id="J-loginImg"]').get_attribute('src')#通过get_attribute方法获取属性值
#print(image_url)#data:image/jpg;base64,图片base64的数据
return image_url
def get_yzm(self, image_url):
'''下载验证码到本地'''
base_url = image_url.split(',')[-1]#最后一个
image_data = base64.b64decode(base_url)#将base64转化为二进制文件
with open('yzm.jpg','wb') as f:
f.write(image_data)
def resolution_yzm(self):
'''主要为了识别验证码并输出结果'''
pojie_url = 'http://littlebigluo.qicp.net:47720/'
with open('yzm.jpg', 'rb') as f:
headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.106 Safari/537.36'}
data = {'pic_xxfile': f}
response = requests.post(url=pojie_url, headers=headers, files=data).content.decode('utf-8')
#print(response)
result = []#用来存储结果的列表
#通过正则匹配结果,通过空格分隔每个结果
for i in re.findall('<B>(.*?)</B>', response)[0].split(' '):
result.append(int(i))
#print(result)
return result
def click_code(self, result):
'''通过动作链点击验证码并登录'''
img_element = self.driver.find_element_by_xpath('//*[@id="J-loginImg"]')
action = ActionChains(driver=self.driver)#实例化一个动作链对象
for one_code in result:
MAP = [[40,43],[118,42],[196,46],[271,38],[44,113],[102.120],[184,102],[265,114]]#创建图片像素点的二位列表
#索引每个登录的的坐标前一个为X轴,后一个为Y轴
action.move_to_element(img_element).move_to_element_with_offset(img_element,MAP[one_code - 1][0],MAP[one_code - 1][1],).click()
time.sleep(1)#点击延迟
action.perform()#执行操作
time.sleep(1)
self.driver.find_element_by_xpath('//*[@id="J-login"]').click()
def main():
login = Login()
img_base64 = login.open_page()
login.get_yzm(img_base64)
yzm_result = login.resolution_yzm()
login.click_code(yzm_result)
main()