selenium登录豆瓣

import time
import requests
from selenium import webdriver
from selenium.webdriver.common.by import By
import cv2
from selenium.webdriver import ActionChains
from selenium.webdriver.chrome.options import Options


def pageRes(url):
    """
    获取验证码图片响应
    :param url:
    :return:
    """
    res = requests.get(url=url)
    if res.status_code == 200:
        return res.content

def writeImg(filepath, imgData):
    """
    将验证码图片保存本地
    :param filepath:
    :return:
    """
    with open(filepath, 'wb') as f:
        f.write(imgData)

def getImg(x_path, string, imgPath):
    """
    :param x_path:
    :param string:
    :param imgPath:
    :return:
    """
    try:
        time.sleep(2)
        src = driver.find_element(By.XPATH, x_path).get_attribute('src').split('*')
        link = src[0] + string
        Img = pageRes(link)
        writeImg(imgPath, Img)

    except Exception as e:
        print('---异常---')

def func_verify():
    '''
    bg: 背景图片
    tp: 缺口图片
    out:输出图片
    '''
    # 读取背景图片和缺口图片
    bg = './verifyImgOne.jpg'
    tp = './verifyImgTwo.jpg'
    #out = "output.png"
    bg_img = cv2.imread(bg, 0)  # 背景图片
    tp_img = cv2.imread(tp, 0)  # 缺口图片
    height, long = bg_img.shape[::-1]

    # 识别图片边缘
    bg_edge = cv2.Canny(bg_img, 100, 200)
    tp_edge = cv2.Canny(tp_img, 100, 200)

    # 转换图片格式
    bg_pic = cv2.cvtColor(bg_edge, cv2.COLOR_GRAY2RGB)
    tp_pic = cv2.cvtColor(tp_edge, cv2.COLOR_GRAY2RGB)

    # 缺口匹配
    res = cv2.matchTemplate(bg_pic, tp_pic, cv2.TM_CCOEFF_NORMED)
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)  # 寻找最优匹配

    # 绘制方框
    th, tw = tp_pic.shape[:2]
    tl = max_loc  # 左上角点的坐标
    br = (tl[0] + tw, tl[1] + th)  # 右下角点的坐标
    #cv2.rectangle(bg_img, tl, br, (0, 0, 255), 2)  # 绘制矩形
    #cv2.imwrite(out, bg_img)  # 保存在本地
    #print(tl[0])
    # 返回缺口的X坐标
    return tl[0] * 280 / height

def get_tracks(distance):
    """
    拿到移动轨迹，模仿人的滑动行为，先匀加速，后匀减速
    匀变速运动公式：
    v = v0 + at
    s = v0t + 1/2a*t*t
    """

    # 初速度
    v = 0
    # 单位时间为0.3秒统计移动轨迹，即0.3秒内的移动位移
    t = 0.3
    # 位置/移动轨迹列表，列表内的一个元素代表0.3秒内的位移
    tracks = []
    # 当前的位移
    current = 0

    # 当距离 < mid值时开始匀加速，当距离 > mid值时开始匀减速
    mid = distance * 4/5

    # current移动位移，小于滑块的距离
    while current < distance:
        if current < mid:
            # 匀加速
            a = 2

        else:
            # 匀减速
            a = -3

        # 初速度
        v0 = v
        # 0.3秒内的位移
        s = v0*t + 1/2 * a * (t**2)
        # 当前位置
        current += s
        # 添加到轨迹列表
        tracks.append(round(s))
        # 速度已经到达v， 该速度作为下次的初速度
        v = v0 + a*t

        # 每0.3秒移动的位移所有元素移动轨迹
    return tracks


# 设置浏览器
url = 'https://movie.douban.com/'
chrome_options = Options()
chrome_options.add_argument('–disable-gpu')                                       # 谷歌文档提到需要加上这个属性来规避bug
chrome_options.add_argument('lang=zh_CN.UTF-8')                                  # 设置默认编码为utf-8
chrome_options.add_experimental_option('excludeSwitches', ['enable-automation']) # 取消chrome受自动控制提示
driver = webdriver.Chrome(options=chrome_options)
driver.get(url=url)

# 登录
driver.find_element(By.XPATH, '//*[@id="db-global-nav"]/div/div[1]/a').click()
driver.find_element(By.XPATH, '//*[@id="account"]/div[2]/div[2]/div/div[1]/ul[1]/li[2]').click()
driver.find_element(By.XPATH, '//*[@id="username"]').send_keys('账号')
driver.find_element(By.XPATH, '//*[@id="password"]').send_keys('密码')
driver.find_element(By.XPATH, '//*[@id="account"]/div[2]/div[2]/div/div[2]/div[1]/div[4]/a').click()
time.sleep(3)
element = driver.find_element(By.XPATH, '/html/body/div[8]/iframe')
driver.switch_to.frame(element)

# 获取验证码背景图
filepathOne = './verifyImgOne.jpg'
stringOne = '*&sid=6919143215671857152&img_index=1&subsid=3'
getImg('//*[@id="slideBg"]', stringOne, filepathOne)

# 获取验证码缺口图
filepathTwo  = './verifyImgTwo.jpg'
stringTwo = '*&sid=6919137859361099776&img_index=2&subsid=4'
getImg('//*[@id="slideBlock"]', stringTwo, filepathTwo)

# 移动滑块完成验证
value = func_verify()
element = driver.find_element(By.XPATH, '//*[@id="tcaptcha_drag_button"]')
ActionChains(driver).click_and_hold(element).perform()
ActionChains(driver).move_to_element_with_offset(to_element=element, xoffset=0, yoffset=0).perform()
tracks = get_tracks(int(value)+12)
for track in tracks:
    ActionChains(driver).move_by_offset(xoffset=track, yoffset=0).perform()

time.sleep(1)
ActionChains(driver).release().perform()