# _*_ coding:utf-8 _*_
# @Time      : 2019/10/14 0014
# @          :requests.py
# @          :PyCharm
import json
import requests
from logging_class import log


class HttpRequest:
    """2、 封装一个不需要记住cookie信息的请求类：HttpRequest"""
    def get(self, url, params=None, **kwargs):
        """get方法"""
        try:
            res = requests.get(url, params=params, **kwargs)
            log.debug(f"GET请求完成")
        except Exception as e:
            log.error(f"GET请求方法报错：{e}")
        else:
            return res

    def post(self, url, data=None, json=None, **kwargs):
        """post方法"""
        try:
            res = requests.post(url, data=data, json=json, **kwargs)
            log.debug(f"POST请求完成")
        except Exception as e:
            log.error(f"POST请求方法报错：{e}")
        else:
            return res

    def visit(self, method, url, params=None, data=None,  **kwargs):
        """visit方法 (可以同时处理 get 和 post 请求)"""
        if method.upper() == "GET":
            return self.get(url, params=params, **kwargs)
        elif method.upper() == "POST":
            return self.post(url, data=data,  **kwargs)
        else:
            return requests.request(method.lower(), url, **kwargs)

    def visit_json(self, method=None, url=None, params=None, data=None, json=None, **kwargs):
        """get_json方法 (可以同时处理 get 和 post 请求,及其他所有请求，且返回 json 格式数据)"""
        res = self.visit(method, url, params=params, data=data, json=json, **kwargs)
        try:
            return res.json()
        except Exception as e:
            log.error(f"解析JSON数据错误{e}")


class HttpSession:
    """1、 封装一个能记住cookie信息的请求类：HttpSession"""
    def __init__(self):
        self.session = requests.session()

    def get(self, url, params=None, **kwargs):
        """get方法"""
        try:
            res = requests.get(url, params=params, **kwargs)
        except Exception as e:
            log.error(f"GET请求方法报错：{e}")
        else:
            return res

    def post(self, url, data=None, json=None, **kwargs):
        """post方法"""
        try:
            res = requests.post(url, data=data, json=json, **kwargs)
        except Exception as e:
            log.error(f"POST请求方法报错：{e}")
        else:
            return res

    def visit(self, method, url, **kwargs):
        """visit方法 (可以同时处理 get 和 post 请求,及其他所有请求)"""
        try:
            if method.upper() == "GET":
                return self.get(url, **kwargs)
            elif method.upper() == "POST":
                return self.post(url,**kwargs)
            else:
                return requests.request(method, url, **kwargs)
        except Exception as e:
            log.error(f"{method}请求方法报错：{e}")

    def json(self, method, url, params=None, data=None, json=None, **kwargs):
        """get_json方法 (可以同时处理 get 和 post 请求,及其他所有请求，且返回 json 格式数据)"""
        res = self.visit(method, url, params=params, data=data, json=json, **kwargs)
        try:
            return res.json()
        except Exception as e:
            log.error(f"解析JSON数据错误{e}")


if __name__ == '__main__':
    pass