模拟登陆汽车之家

from io import BytesIO
from PIL import Image
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
import requests
import os,time

from hashlib import md5

class Chaojiying_Client(object):
    # password = password.encode('utf8')
    def __init__(self, username, password, soft_id):
        self.username = username
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ("aaa.jpg", im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()

# driver = webdriver.Chrome()
# driver.get("https://account.autohome.com.cn/login")

class qiCheZhiJia(object):

    driver = webdriver.Chrome()
    driver.get("https://account.autohome.com.cn/login")
    wait = WebDriverWait(driver, 10)

    def search(self):
        style = self.wait.until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, "#tab-nav > div > div.tab-account"))
        )
        style.click()

        phone = self.wait.until(
            EC.presence_of_element_located((By.CSS_SELECTOR, "#UserName"))
        )
        passwd = self.wait.until(
            EC.presence_of_element_located((By.CSS_SELECTOR, "#PassWord"))
        )
        phone.send_keys("username")
        passwd.send_keys("passwd")

        pic_click = self.wait.until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, "#embed-captcha2 > div > div.geetest_btn > div.geetest_radar_btn > div.geetest_radar_tip > span.geetest_radar_tip_content"))
        )
        pic_click.click()
        img = self.wait.until(
            EC.presence_of_element_located((By.CSS_SELECTOR, "body > div.geetest_fullpage_click.geetest_float.geetest_wind.geetest_click > div.geetest_fullpage_click_wrap > div.geetest_fullpage_click_box > div > div > div.geetest_table_box > "
                                                             "div.geetest_window > div > div.geetest_item_wrap > img"))
        )
        time.sleep(1)
        location = img.location
        size = img.size
        top,bottom,left,right = \
            location['y']-size["height"]/5,location['y'] + size['height'],location['x'],location['x'] + size['width']

        screenshot = self.driver.get_screenshot_as_png()
        screenshot = Image.open(BytesIO(screenshot))

        captcha = screenshot.crop((left,top,right,bottom))
        bytes_array = BytesIO()
        captcha.save(bytes_array,format='PNG')

        with open("aaa.png",'wb') as f:
            f.write(bytes_array.getvalue())

        chaojiying = Chaojiying_Client(b'username', b'passwd', b'id')  # 用户中心>>软件ID 生成一个替换 96001
        # img_url = os.path.dirname(__file__)+"/"+"ccd.jpg"  # 本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
        try:
            res = chaojiying.PostPic(bytes_array.getvalue(), 9004)
            print(chaojiying.PostPic(bytes_array.getvalue(), 9004))
            groups = res.get('pic_str').split('|')
            locations = [[int(number) for number in group.split(',')] for group in groups]
            for location in locations:
                ActionChains(self.driver).move_to_element_with_offset(img,location[0],location[1]-size["height"]/5).click().perform()
                time.sleep(1)

            sure = self.wait.until(
                EC.element_to_be_clickable((By.CSS_SELECTOR, "body > div.geetest_fullpage_click.geetest_float.geetest_wind.geetest_click > div.geetest_fullpage_click_wrap > div.geetest_fullpage_click_box > div > div > div.geetest_panel > a > div")))

            sure.click()
            time.sleep(2)
            login = self.wait.until(
                EC.element_to_be_clickable((By.CSS_SELECTOR, "#SubmitLogin"))
            )
            login.click()
        except Exception as e:
            print(e)
            chaojiying.ReportError(res.get('pic_id'))
            exit()

if __name__ == "__main__":
    qczj = qiCheZhiJia()
    qczj.search()