import urllib3
import requests
from utils.log_utils import LogUtils


class MyRequest:
    """接口访问基类"""
    ip = 'https://10.90.1.226'
    login_url = ip + ':8089'
    base_url = ip + ':9402'

    def __init__(self) -> None:
        self.session = None

    def get_session(self):
        if self.session == None:
            urllib3.disable_warnings()
            self.session = requests.session()
            self.session.headers = {
                "Accept": "application/json, text/javascript, */*; q=0.01",
                "Accept-Encoding": "gzip, deflate, br",
                "Connection": "keep-alive",
                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36',
            }
        return self.session

    def set_cookie(self, cookie):
        # if isinstance(cookie, str):
        #     cookie = eval(cookie)
        # requests.utils.add_dict_to_cookiejar(self.get_session().cookies, cookie_dict=cookie)
        self.get_session().headers["Cookie"] = cookie

    def post(self, url, data):
        try:
            res = self.get_session().post(url, data=data, timeout=30, verify=False, stream=True)
            if res.status_code != 200:
                # 打印参数
                LogUtils.error('请求数据失败:'+url+',' + str(res.status_code) + str(data))
                return False
            else:
                return res
        except Exception as e:
            LogUtils.error('网络访问失败:', e)
            return False


my_Request = MyRequest()