12306自动登录

最近看到了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('(.*?)', 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()