Selenium模拟登陆简书

from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from PIL import Image
from io import BytesIO
import time


USERNAME = '15xxx'
PASSWORD = '111111'
BORDER = 6

class Login(object):
    def __init__(self):
        self.browser = webdriver.Chrome()
        self.username = USERNAME
        self.password = PASSWORD
        self.wait = WebDriverWait(self.browser, 10)

    def __del__(self):
        self.browser.close()

    def open(self):
        self.browser.get('https://www.jianshu.com/')
        login = self.wait.until(EC.element_to_be_clickable((By.XPATH, '//a[@class="btn log-in"]')))
        login.click()
        time.sleep(10)

        username = self.wait.until(EC.presence_of_element_located((By.ID, 'session_email_or_mobile_number')))
        password = self.wait.until(EC.presence_of_element_located((By.ID, 'session_password')))
        username.clear()
        password.clear()
        username.send_keys(self.username)
        password.send_keys(self.password)

    def click_button(self):
        '''
        点击按钮呼出验证码图片
        '''
        button = self.wait.until(EC.element_to_be_clickable((By.CLASS_NAME, 'geetest_radar_tip')))
        button.click()


    def get_captcha(self,name='captcha.png'):
        '''
        截图并且按照验证码图片的位置进行抠图
        '''
        img = self.wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'geetest_canvas_img')))
        time.sleep(2)
        location = img.location
        size = img.size
        print('验证码图片位置是：', location)
        print('验证码图片尺寸是：', size)
        top,bottom,left,right = location['y'], location['y']+size['height'], location['x'], location['x']+size['width']

        screenshot = self.browser.get_screenshot_as_png()
        screenshot = Image.open(BytesIO(screenshot))
        captcha = screenshot.crop((left, top, right, bottom))
        captcha.save(name)


    def get_slider(self):
        slider = self.wait.until(EC.element_to_be_clickable((By.CLASS_NAME, 'geetest_slider_button')))
        return slider


    def get_gap(self, captcha1, captcha2):
        '''
        左边部分一定存在差异，因此直接忽略
        '''
        left = 60
        for i in range(left,captcha1.size[0]):  # captcha1.size宽度和高度的元组
            for j in range(captcha1.size[1]):
                if not self.is_pixel_equal(captcha1, captcha2, i, j):
                    left = i
                    return left
        return left

    def is_pixel_equal(self, captcha1, captcha2, x, y):
        '''
        对每个像素的RGB值进行比较
        '''
        pixel1 = captcha1.load()[x, y]
        pixel2 = captcha2.load()[x, y]
        threshold = 60
        if abs(pixel1[0] - pixel2[0]) < threshold and abs(pixel2[1] - pixel2[1]) < threshold and 
           abs(pixel1[2] - pixel2[2]) < threshold:
            return True
        else:
            return False

    def get_track(self, distance):
        track = []
        current = 0
        mid = distance*4/5
        t = 0.2
        v = 0
        while current < distance:
            if current < mid:
                a = 10
            else:
                a = -18
            v0 = v
            v = v0 + a * t
            move = v0 * t + 1 / 2 * a * t * t
            current += move
            track.append(round(move))
        return track

    def move_to_gap(self, slider, track):
        ActionChains(self.browser).click_and_hold(slider).perform()
        for x in track:
            ActionChains(self.browser).move_by_offset(xoffset=x, yoffset=0)
        time.sleep(0.5)
        ActionChains(self.browser).release().perform()

    def loginin(self):
        submit = self.wait.until(EC.element_to_be_clickable((By.CLASS_NAME, 'commit')))
        submit.click()
        time.sleep(5)
        print('登陆成功')

    def crack(self):
        self.open()
        self.click_button()  # 点击按钮呼出验证码图片
        captcha1 = self.get_captcha('captcha1.png')  # 截图，获取原始验证码位置，抠图

        slider = self.get_slider()   # 获取验证码滑块位置
        slider.click()  # 点击位置
        captcha2 = self.get_captcha('captcha2.png')  # 截图，获取凹凸验证码位置，抠图

        gap = self.get_gap(captcha1, captcha2)       # 图片对比，距离
        gap -= BORDER                                # 减去左边的边框距离
        track = self.get_track(gap)                  # 获取移动轨迹

        print('滑动轨迹', track)

        self.move_to_gap(slider, track)
        success = self.wait.until(
            EC.text_to_be_present_in_element((By.CLASS_NAME, 'geetest_success_radar_tip_content'), '验证成功'))
        print(success)

        time.sleep(0.5)
        self.loginin()
        # if not success:
        #     self.crack()
        # else:
        #     self.login()

if __name__ == '__main__':
    crack = Login()
    crack.crack()