import requests
from utils.logger import get_logger
logger = get_logger(__file__)
class BaseRequest:
def __init__(self):
self.url = None
self.method = None
def request(self, **kwargs):
"""
调用request发送http请求
:param json: http请求入参为json
:param params: http请求入参为form
:param headers: http请求头
:return: requests响应
"""
req_json = kwargs.get('json', None)
req_params = kwargs.get('params', None)
headers = kwargs.get('headers', None)
cookies = kwargs.get('cookies', None)
logger.info(f'【Request URL】 {self.url}')
if req_json:
logger.info(f'<------ Request ------> {req_json}')
res = requests.request(url=self.url, method=self.method, json=req_json, headers=headers, cookies=cookies)
elif req_params:
logger.info(f'<------ Request ------> {req_params}')
res = requests.request(url=self.url, method=self.method, params=req_params, headers=headers,
cookies=cookies)
else:
res = requests.request(url=self.url, method=self.method, headers=headers, cookies=cookies)
logger.info(f'【Response Code】 {res.status_code}')
logger.info(f'======> Response <====== {res.text}')
# logger.info(f'======> Headers <====== {res.headers}')
return res
from config.settings import host
from interface.base_request import BaseRequest
class Login(BaseRequest):
"""
登录接口
"""
def __init__(self):
super().__init__()
self.url = f'{host}/Login'
self.method = 'POST'
# 请求头实例
self.data = {
"userAccount": "admin",
"userPwd": "XXXX"
}
if __name__ == '__main__':
l = Login()
cookies = l.request(json=l.data).cookies
