import requests
from hashlib import md5
class ChaojiyingClient(object):
    def __init__(self, username, password, soft_id):
        self.username = username
        password = password.encode('utf8')
        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: 圖片字節(jié)
        codetype: 題目類型 參考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files,
                          headers=self.headers)
        return r.json()
    def PostPic_base64(self, base64_str, codetype):
        """
        im: 圖片字節(jié)
        codetype: 題目類型 參考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
            'file_base64': base64_str
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, headers=self.headers)
        return r.json()
    def ReportError(self, im_id):
        """
        im_id:報(bào)錯(cuò)題目的圖片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()
if __name__ == '__main__':
    chaojiying = ChaojiyingClient('超級(jí)鷹用戶名', '超級(jí)鷹用戶名的密碼', '96001')  # 用戶中心>>軟件ID 生成一個(gè)替換 96001
    im = open('a.jpg', 'rb').read()  # 本地圖片文件路徑 來(lái)替換 a.jpg 有時(shí)WIN系統(tǒng)須要//
    print(chaojiying.PostPic(im, 1902))  # 1902 驗(yàn)證碼類型  官方網(wǎng)站>>價(jià)格體系 3.4+版 print 后要加()
    # print chaojiying.PostPic(base64_str, 1902)  #此處為傳入 base64代碼

from selenium import webdriver
from selenium.webdriver.common.by import By
from PIL import Image
from selenium.webdriver.chrome.options import Options
from chaojiying import chaojiying_Python
chrome_options = Options()
chrome_options.add_argument('window-size=1920x1080')  # 指定瀏覽器分辨率
chrome_options.add_argument('--disable-gpu')  # 谷歌文檔提到需要加上這個(gè)屬性來(lái)規(guī)避bug
chrome_options.add_argument('--hide-scrollbars')  # 隱藏滾動(dòng)條, 應(yīng)對(duì)一些特殊頁(yè)面
# chrome_options.add_argument('blink-settings=imagesEnabled=false')  # 不加載圖片, 提升速度
chrome_options.add_argument('--headless')  # 瀏覽器不提供可視化頁(yè)面. linux下如果系統(tǒng)不支持可視化不加這條會(huì)啟動(dòng)失敗
# chrome = webdriver.Chrome(executable_path='../chromedriver.exe')
chrome = webdriver.Chrome(executable_path='../chromedriver.exe', options=chrome_options)
chrome.implicitly_wait(10)
chrome.maximize_window()
try:
    chrome.get('http://www.aa7a.cn/user.php?')
    username = chrome.find_element(By.ID, 'username')
    password = chrome.find_element(By.ID, 'password')
    captcha = chrome.find_element(By.ID, 'captcha')
    # 保存大圖
    chrome.save_screenshot('main.png')
    img = chrome.find_element(By.ID, 'login_img_checkcode')
    img_location = img.location
    img_size = img.size
    # 使用pillow扣除大圖中的驗(yàn)證碼
    img_tu = (
        int(img_location['x']),
        int(img_location['y']),
        int(img_location['x'] + img_size['width']),
        int(img_location['y'] + img_size['height']),
    )
    # 打開(kāi)頁(yè)面大圖
    im = Image.open('./main.png')
    # 剪切驗(yàn)證碼圖片
    fram = im.crop(img_tu)
    # 保存驗(yàn)證碼圖片
    fram.save('code.png')
    # 打開(kāi)驗(yàn)證碼圖片
    code_img = open('code.png', 'rb').read()
    # 調(diào)用超級(jí)鷹識(shí)別
    res = chaojiying_Python.chaojiying.PostPic(code_img, 1902)
    code = res.get('pic_str')
    username.send_keys('username')
    password.send_keys('123')
    captcha.send_keys(code)
    print(code)
except Exception as e:
    print(e)
finally:
    chrome.quit()