mirror of
https://gitee.com/Lyon1998/pikapython.git
synced 2025-01-15 17:02:53 +08:00
84 lines
2.0 KiB
Python
84 lines
2.0 KiB
Python
import _requests
|
|
|
|
|
|
class Response(_requests.Response):
|
|
def __init__():
|
|
_requests.__init__()
|
|
|
|
def __del__():
|
|
_requests.__del__()
|
|
|
|
|
|
def _append_params_to_url(rqst: Response, url: str, params: dict) -> int:
|
|
if params is None:
|
|
ret = rqst.urlencode_write(url)
|
|
return 1
|
|
if '?' in url:
|
|
first_connect = '&'
|
|
|
|
else:
|
|
first_connect = '?'
|
|
# 初始化连接url
|
|
ret = rqst.urlencode_write(url)
|
|
if ret != 1:
|
|
return ret
|
|
count = 0
|
|
connect = '='
|
|
start = '&'
|
|
for k, v in params.items():
|
|
if count == 0:
|
|
ret = rqst.urlencode_write(str(k), str(v), first_connect, connect)
|
|
if ret != 1:
|
|
return ret
|
|
else:
|
|
ret = rqst.urlencode_write(str(k), str(v), start, connect)
|
|
if ret != 1:
|
|
return ret
|
|
return 1
|
|
|
|
def _append_headers(rqst: Response, headers: dict) -> int:
|
|
if headers is None:
|
|
return
|
|
for k, v in headers.items():
|
|
ret = rqst.header_write(str(k), str(v))
|
|
if ret != 1:
|
|
return ret
|
|
|
|
return 1
|
|
|
|
def request(method: str, url: str, params=None, headers=None, **kwargs) -> Response:
|
|
"""
|
|
初始化请求对象，分配内存和固定请求头
|
|
"""
|
|
rqst = Response()
|
|
# 初始化，分配内存, 写入方法POST/GET
|
|
ret = rqst.request_init(method)
|
|
if ret != 1:
|
|
del rqst
|
|
return None
|
|
# 写入URL
|
|
ret = _append_params_to_url(rqst, url, params)
|
|
if ret != 1:
|
|
# 出现错误，需要释放对象
|
|
del rqst
|
|
return None
|
|
# 写入默认HTTP版本号
|
|
ret = rqst.proto_write()
|
|
if ret != 1:
|
|
del rqst
|
|
return None
|
|
# 写入响应头数据
|
|
ret = _append_headers(rqst, headers)
|
|
if ret != 1:
|
|
del rqst
|
|
return None
|
|
# 进行实际request过程
|
|
ret = rqst.request(kwargs)
|
|
if ret != 1:
|
|
del rqst
|
|
return None
|
|
return rqst
|
|
|
|
def get(url: str, params=None) -> Response:
|
|
return request('GET', url, params)
|