# 封装http请求
import requests


class HttpRequest:
    # 需要的参数
    # method：请求方式
    # url：请求的url
    # param：请求参数
    # headers：请求头
    # cookie：请求的cookie值
    def __init__(self, method, url, param=None, headers=None, cookie=None):
        self.method = method
        self.url = url
        self.param = param
        self.headers = headers
        self.cookie = cookie

    def http_request(self):
        if self.method.lower() == "post":
            # if self.headers is None:
                # self.headers = {"Content-Type": "application/json;charset=UTF-8s"}
            return requests.post(self.url, self.param,headers=self.headers, cookies=self.cookie,verify=False)
        elif self.method.lower() == "get":
            if self.headers is None:
                self.headers = {"Content-Type": "application/json;charset=UTF-8s"}
            return requests.get(self.url, self.param, headers=self.headers, cookies=self.cookie,verify=False)
        else:
            print("请求方式错误：{0}".format(self.method))


# 主程序入口
if __name__ == '__main__':
    login_url = "http://www.qabujiaban.com/user/login"
    data = {"username": "uuuu222都44", "password": "WJHasb124*1"}
    # 登陆
    res = HttpRequest("Post", login_url, data).http_request()
    print("登陆响应文本：", res.json())  #
    cookie = res.cookies  # 获取cookie
    query_url = "http://www.qabujiaban.com/user/query"
    query_headers = {"Content-Type": "application/json;charset=UTF-8s"}
    # 查询
    rqs = HttpRequest("Get", query_url, headers=query_headers, cookie=cookie).http_request()
    print("查询响应文本：", rqs.json())
    # print("Uu".lower())