import hashlib
import os
import time

import requests

from base.utils import yzm_client, config

# for i in range(5000):
# url = 'https://weixin.sogou.com/antispider/util/seccode.php?tc=1611128720'
# headers = {
#     "Accept": "image/webp,image/apng,image/*,*/*;q=0.8",
#     "Accept-Encoding": "gzip, deflate, br",
#     "Accept-Language": "zh-CN,zh;q=0.9",
#     "Connection": "keep-alive",
#     "Cookie": "IPLOC=CN4403; SUID=D04118742034990A00000000600688AE; SUV=1611040943430315; ABTEST=0|1611040948|v1; SNUID=7CECB4D9ACA814971EC3254FADDE563A; weixinIndexVisited=1; PHPSESSID=tn46iouqk0361hppqvpjr686s4; refresh=1",
#     "Host": "weixin.sogou.com",
#     "Referer": "https://weixin.sogou.com/antispider/?from=%2Flink%3Furl%3Ddn9a_-gY295K0Rci_xozVXfdMkSQTLW6cwJThYulHEtVjXrGTiVgS90l3pZgDpemO5p6ii_hdNUxM1mSwDO-_lqXa8Fplpd9qPsU7eq8zcBc3b5aBqIv75qomGt4HZFlcSecu431YjHTl_aj3j1EYjqM9qtaKjmWeaJBsbznXEAgmxNFY17v8ZUzvT0h_TdXiW1duCGgLy8JLQuADD8wbxVB6c6W--Ly7BXPR1xGLO4YHFV1UsN_EtUPLgGjl-tTeSdOd9iCX1yrCDCayYmfpA..%26type%3D2%26query%3D%E8%81%94%E6%98%93%E8%9E%8D%26token%3Dempty",
#     "Sec-Fetch-Dest": "image",
#     "Sec-Fetch-Mode": "no-cors",
#     "Sec-Fetch-Site": "same-origin",
#     "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.105 Safari/537.36",
# }
#
# resp = requests.get(url=url, headers=headers)
# con = resp.content
# print(type(con), con)
# # 保存文件
#
# m = hashlib.md5()
# m.update(con)
# md5code = m.hexdigest()

pathDir = os.listdir('./base_yzm')
for allDir in pathDir:
    png_path = os.path.join('./base_yzm', allDir)
    if os.path.isfile(png_path):
        with open(png_path, 'rb')as f:
            con = f.read()
        result = yzm_client.parse_yzm_path(config.YZM_WORD_NUMS_7, png_path)
        result = result['Result']
        with open('./yzm/' + result + '.png', 'wb') as f:
            f.write(con)
        os.remove(png_path)