import time import requests from requests.exceptions import ConnectionError from cookiespool.config import * class Yundama(): def __init__(self, username, password, app_id, app_key, api_url=YUNDAMA_API_URL): self.username = username self.password = password self.app_id = str(app_id) if not isinstance(app_id, str) else app_id self.app_key = app_key self.api_url = api_url def login(self): """ 登录云打码账户 :return: """ try: data = {'method': 'login', 'username': self.username, 'password': self.password, 'appid': self.app_id, 'appkey': self.app_key} response = requests.post(self.api_url, data=data) if response.status_code == 200: result = response.json() print(result) if 'ret' in result.keys() and result.get('ret') < 0: return self.error(result.get('ret')) else: return result return None except ConnectionError: return None def upload(self, files, timeout, code_type): """ 上传验证码得到识别结果 :param files: :param timeout: :param code_type: :return: """ try: data = {'method': 'upload', 'username': self.username, 'password': self.password, 'appid': self.app_id, 'appkey': self.app_key, 'codetype': str(code_type), 'timeout': str(timeout)} response = requests.post(self.api_url, data=data, files=files) if response.status_code == 200: return response.json() return None except ConnectionError: return None def retry(self, cid, try_count=1): """ 临时识别不出, 传入cid重试 :param cid: 验证码ID :param try_count: 重试次数 :return: 验证码结果 """ if try_count >= YUNDAMA_MAX_RETRY: return None print('Retrying: ', cid, 'Count: ', try_count) time.sleep(2) try: data = {'method': 'result', 'cid': cid} print(data) response = requests.post(self.api_url, data=data) if response.status_code == 200: result = response.json() print(result) if 'ret' in result.keys() and result.get('ret') < 0: print(self.error(result.get('ret'))) if result.get('ret') == 0 and 'text' in result.keys(): return result.get('text') else: return self.retry(cid, try_count + 1) return None except ConnectionError: return None def identify(self, file=None, stream=None, timeout=60, code_type=5000): """ 主函数 :param file: 文件名 :param stream: 文件流, 优先于文件名 :param timeout: 超时时间 :param code_type: 验证码类型 :return: 识别结果 """ if stream: files = {'file': stream} elif file: files = {'file': open(file, 'rb')} else: return None result = self.upload(files, timeout, code_type) if 'ret' in result.keys() and result.get('ret') < 0: print(self.error(result.get('ret'))) if result.get('text'): print('验证码识别成功', result.get('text')) return result.get('text') else: return self.retry(result.get('cid')) def error(self, code): """ 报错原因 :param code: 错误码 :return: 错误原因 """ map = { -1001: '密码错误', -1002: '软件ID/密钥有误', -1003: '用户被封', -1004: 'IP被封', -1005: '软件被封', -1006: '登录IP与绑定的区域不匹配', -1007: '账号余额为零', -2001: '验证码类型有误', -2002: '验证码图片太大', -2003: '验证码图片损坏', -2004: '上传验证码图片失败', -3001: '验证码ID不存在 ', -3002: '验证码还在识别', -3003: '验证码识别超时', -3004: '验证码看不清', -3005: '验证码报错失败', -4001: '充值卡号不正确或已使用', -5001: '注册用户失败' } return '云打码' + map.get(code) if __name__ == '__main__': ydm = Yundama(YUNDAMA_USERNAME, YUNDAMA_PASSWORD, YUNDAMA_APP_ID, YUNDAMA_APP_KEY) result = ydm.identify(file='getimage.jpg') print(result)