from threading import Thread
import requests
from apps.utils.errors import XX_REQUEST_ERROR
from apps.utils.tools import print_roundtrip
from django.conf import settings
import time
from rest_framework.exceptions import APIException, ParseError
requests.packages.urllib3.disable_warnings()


class XxClient:
    """
    寻息
    """

    def __init__(self, licence=settings.XX_LICENCE, username=settings.XX_USERNAME) -> None:
        if not settings.XX_ENABLED:
            return None
        self.licence = licence
        self.username = username
        self.isGetingToken = False
        self.isRuning = True
        self.token = ''
        self.t = None
        self.setup()

    def _get_token_loop(self):
        while self.isRuning:
            json = {
                'licence': self.licence
            }
            r = requests.post(json=json, url=settings.XX_BASE_URL + '/getAccessTokenV2', verify=False, timeout=20)
            ret = r.json()
            if ret.get('errorCode', 1) == 0:
                self.token = ret['data']['token']
            time.sleep(3600)

    def get_token(self):
        self.isGetingToken = True
        json = {
            'licence': self.licence
        }
        r = requests.post(json=json, url=settings.XX_BASE_URL + '/getAccessTokenV2', verify=False, timeout=20)
        ret = r.json()
        if ret.get('errorCode', 1) == 0:
            self.isGetingToken = False
            self.token = ret['data']['token']

    def setup(self):
        self.t = Thread(target=self._get_token_loop, args=(), daemon=True)
        self.t.start()

    def __del__(self):
        """
        自定义销毁
        """
        self.isRuning = False
        self.t.join()

    def request(self, url: str, method: str = 'post', params=dict(), json=dict(), timeout=4, raise_exception=True):
        params['accessToken'] = self.token
        json['username'] = self.username
        json['buildId'] = settings.XX_BUILDID
        if self.isGetingToken:
            req_num = 0
            while True:
                time.sleep(0.5)
                if not self.isGetingToken:
                    self.request(url, method, params, json, timeout, raise_exception)
                req_num = req_num + 1
                if req_num > 4:
                    break
        else:
            r = getattr(requests, method)('{}{}'.format(settings.XX_BASE_URL, url),
                                          params=params, json=json, timeout=timeout, verify=False)
            # if settings.DEBUG:
            #     print_roundtrip(r)
            ret = r.json()
            if ret.get('errorCode') == '1060000':
                self.get_token()  # 重新获取token
                self.request(url, method, params, json, timeout, raise_exception)  # 重新请求
            else:
                if ret['errorCode'] != 0:
                    err_detail = dict(detail='寻息错误:' + '|'.join(ret['errorMsg']),
                                      code='xx_' + str(ret['errorCode']))
                    if raise_exception:
                        raise ParseError(**err_detail)
                    return 'fail', err_detail
                return 'success', ret['data']
        if raise_exception:
            raise APIException(**XX_REQUEST_ERROR)
        return 'error', XX_REQUEST_ERROR