提交 789affa5 编写于 作者: H hjdhnx

优化fetch,post,request函数

上级 1e56b979
...@@ -140,6 +140,8 @@ def dealObj(obj=None): ...@@ -140,6 +140,8 @@ def dealObj(obj=None):
obj = {} obj = {}
encoding = obj.get('encoding') or 'utf-8' encoding = obj.get('encoding') or 'utf-8'
encoding = str(encoding).replace("'", "") encoding = str(encoding).replace("'", "")
method = obj.get('method') or 'get'
method = str(method).replace("'", "")
# print(type(url),url) # print(type(url),url)
# headers = dict(obj.get('headers')) if obj.get('headers') else {} # headers = dict(obj.get('headers')) if obj.get('headers') else {}
# headers = obj.get('headers').to_dict() if obj.get('headers') else {} # headers = obj.get('headers').to_dict() if obj.get('headers') else {}
...@@ -161,13 +163,16 @@ def dealObj(obj=None): ...@@ -161,13 +163,16 @@ def dealObj(obj=None):
'headers':new_headers, 'headers':new_headers,
'timeout':timeout, 'timeout':timeout,
'body': new_body, 'body': new_body,
'method':method
} }
def base_request(url,obj,method=None): def base_request(url,obj):
# verify=False 关闭证书验证 # verify=False 关闭证书验证
url = str(url).replace("'", "") url = str(url).replace("'", "")
method = obj.get('method') or ''
if not method: if not method:
method = 'get' method = 'get'
obj['method'] = 'method'
# print(obj) # print(obj)
print(f'{method}:{url}') print(f'{method}:{url}')
try: try:
...@@ -186,28 +191,24 @@ def base_request(url,obj,method=None): ...@@ -186,28 +191,24 @@ def base_request(url,obj,method=None):
print(f'{method}请求发生错误:{e}') print(f'{method}请求发生错误:{e}')
return '' return ''
def fetch(url,obj,method=None): def fetch(url,obj):
if not method:
method = 'get'
obj = dealObj(obj) obj = dealObj(obj)
# print(f'{method}:{url}')
if not obj.get('headers') or not obj['headers'].get('User-Agent'): if not obj.get('headers') or not obj['headers'].get('User-Agent'):
obj['headers']['User-Agent'] = PC_UA obj['headers']['User-Agent'] = PC_UA
return base_request(url,obj,method) return base_request(url,obj)
def post(url,obj): def post(url,obj):
obj = dealObj(obj) obj = dealObj(obj)
return base_request(url,obj,'post') obj['method'] = 'post'
return base_request(url,obj)
def request(url,obj,method=None): def request(url,obj):
if not method:
method = 'get'
obj = dealObj(obj) obj = dealObj(obj)
# print(f'{method}:{url}') # print(f'{method}:{url}')
if not obj.get('headers') or not obj['headers'].get('User-Agent'): if not obj.get('headers') or not obj['headers'].get('User-Agent'):
obj['headers']['User-Agent'] = UC_UA obj['headers']['User-Agent'] = UC_UA
return base_request(url, obj, method) return base_request(url, obj)
def buildUrl(url,obj=None): def buildUrl(url,obj=None):
url = str(url).replace("'", "") url = str(url).replace("'", "")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册