import http.client, mimetypes, urllib, json, time, requests


######################################################################

class YDMHttp:
    apiurl = 'http://api.yundama.com/api.php'
    username = ''
    password = ''
    appid = ''
    appkey = ''

    def __init__(self, username, password, appid, appkey)
 
:
        self.username = username
        self.password = password
        self.appid = str(appid)
        self.appkey = appkey

    def request(self, fields, files=[]):
        response = self.post_url(self.apiurl, fields, files)
        response = json.loads(response)
        return response

    def
 
 balance(self):
        data = {'method': 'balance', 'username': self.username, 'password': self.password, 'appid': self.appid,
                'appkey': self.appkey}
        response = self.request(data)
        if (response):
            if (response['ret'] and response['ret'] < 0):
                return
 
 response['ret']
            else:
                return response['balance']
        else:
            return -9001

    def login(self):
        data = {'method': 'login', 'username': self.username, 'password': self.password, 'appid': self.appid,
                'appkey': self.appkey}
        response = self.request(data)
        if (response):
            if (response['ret'] and response['ret'] < 0):
                return response['ret']
            else:
                return response['uid']
        else:
            return -9001

    def upload(self, filename, codetype, timeout):
        data = {'method': 'upload', 'username': self.username, 'password': self.password, 'appid': self.appid,
                'appkey': self.appkey, 'codetype': str(codetype), 'timeout': str(timeout)}
        file = {'file': filename}
        response = self.request(data, file)
        if (response):
            if (response['ret'] and response['ret'] < 0):
                return response['ret']
            else:
                return response['cid']
        else:
            return -9001

    def result(self, cid):
        data = {'method': 'result', 'username': self.username, 'password': self.password, 'appid': self.appid,
                'appkey': self.appkey, 'cid': str(cid)}
        response = self.request(data)
        return response and response['text'] or ''

    def decode(self, filename, codetype, timeout):
        cid = self.upload(filename, codetype, timeout)
        if (cid > 0):
            for i in range(0, timeout):
                result = self.result(cid)
                if (result != ''):
                    return cid, result
                else:
                    time.sleep(1)
            return -3003, ''
        else:
            return cid, ''

    def report(self, cid):
        data = {'method': 'report', 'username': self.username, 'password': self.password, 'appid': self.appid,
                'appkey': self.appkey, 'cid': str(cid), 'flag': '0'}
        response = self.request(data)
        if (response):
            return response['ret']
        else:
            return -9001

    def post_url(self, url, fields, files=[]):
        for key in files:
            files[key] = open(files[key], 'rb');
        res = requests.post(url, files=files, data=fields)
        return res.text


######################################################################
#程式碼從這裡開始寫入
import requests
import re
source = requests.get('https://www.douban.com/').text
demo = re.compile('<img id="captcha_image" src="(.*?)" .*?/>',re.S)
lists = demo.findall(source)
header = {
    'Host': 'www.douban.com',
    'Origin': 'https://www.douban.com',
    'Referer': 'https://www.douban.com/',
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/66.0.3359.117 Safari/537.36'
}
if len(lists)>0: # 這裡需要判斷獲取的驗證碼連結不為空
    pic = requests.get(lists[0]).content
    op = open('getimage.jpg','wb')
    op.write(pic)
    op.close()
    id = lists[0].split('id=')[-1].split('&')[0]
    print(id)
    #這裡往下是打碼軟體自帶的程式碼有寫地方需要進行改動
    # 使用者名稱 需要去註冊
    username = '雲打碼user'

    # 密碼
    password = '密碼'

    # 軟體ＩＤ，開發者分成必要引數。登入開發者後臺【我的軟體】獲得！
    appid = 1

    # 軟體金鑰，開發者分成必要引數。登入開發者後臺【我的軟體】獲得！
    appkey = '22cc5376925e9387a23cf797cb9ba745'

    # 圖片檔案
    filename = 'getimage.jpg'

    # 驗證碼型別，# 例：1004表示4位字母數字，不同型別收費不同。請準確填寫，否則影響識別率。在此查詢所有型別 http://www.yundama.com/price.html
    codetype = 3000

    # 超時時間，秒
    timeout = 60

    # 檢查
    if (username == 'username'):
        print('請設定好相關引數再測試')
    else:
        # 初始化
        yundama = YDMHttp(username, password, appid, appkey)

        # 登陸雲打碼
        uid = yundama.login();
        # print('uid: %s' % uid)

        # 查詢餘額
        balance = yundama.balance();
        # print('balance: %s' % balance)

        # 開始識別，圖片路徑，驗證碼型別ID，超時時間（秒），識別結果
        cid, result = yundama.decode(filename, codetype, timeout);
        print('cid: %s, result: %s' % (cid, result))
    # 為post請求構建引數
        data = {
            'source': 'index_nav',
            'form_email': '豆瓣賬號',
            'form_password': '密碼',
            'captcha-solution': result, # 這裡result是通過打碼工具返回來的解碼資料 
            'captcha-id': id # 這裡的id是通過url地址解析出來的圖片的id
        }

        s = requests.Session()
        responses = s.post('https://www.douban.com/accounts/login', data=data, headers=header).text
        # print(responses)
        print(s.get('https://www.douban.com/', headers=header).text)
else: # 這裡是如果沒有過驗證碼的話直接實現使用者登入
    data = {
    'source': 'index_nav',
    'form_email': '豆瓣賬號',
    'form_password': '密碼',
    }

    s = requests.Session()
    responses = s.post('https://www.douban.com/accounts/login',data = data,headers = header).text
    print(responses)
    print(s.get('https://www.douban.com/',headers = header).text)


# https://www.douban.com/misc/captcha?id=4aP1y9LgNSzOICxUa2cXMMYV:en&size=s
# source: index_nav
# form_email: 豆瓣賬號
# form_password: 密碼
# captcha-solution: water
# captcha-id: 4aP1y9LgNSzOICxUa2cXMMYV:en 驗證碼圖片id