优化常规检查模块

This commit is contained in:
Jing Ling
2020-08-22 01:05:23 +08:00
parent 9847ca6ecc
commit b4ea31561f
8 changed files with 54 additions and 55 deletions
+9 -5
View File
@@ -73,7 +73,7 @@ class Module(object):
:param dict params: request parameters
:param bool check: check response
:param kwargs: other params
:return: requests's response object
:return: response object
"""
try:
resp = requests.head(url,
@@ -93,16 +93,20 @@ class Module(object):
return resp
return None
def get(self, url, params=None, check=True, **kwargs):
def get(self, url, params=None, check=True, ignore=False, **kwargs):
"""
Custom get request
:param str url: request url
:param dict params: request parameters
:param bool check: check response
:param bool ignore: ignore error
:param kwargs: other params
:return: requests's response object
:return: response object
"""
level = 'ERROR'
if ignore:
level = 'DEBUG'
try:
resp = requests.get(url,
params=params,
@@ -113,7 +117,7 @@ class Module(object):
verify=self.verify,
**kwargs)
except Exception as e:
logger.log('ERROR', e.args)
logger.log(level, e.args)
return None
if not check:
return resp
@@ -129,7 +133,7 @@ class Module(object):
:param dict data: request data
:param bool check: check response
:param kwargs: other params
:return: requests's response object
:return: response object
"""
try:
resp = requests.post(url,