在进行request方法时,需要对request返回的数据进行判断是否是正确的数值,这时我们可以单独创建一个方法,进行判断,这样可以实现方法的复用性。具体代码如下:
def handle_response(response: requests.Response, is_exit=True, excption: Exception = None): if hasattr(response, "status_code"): if response.status_code >= 200 and response.status_code < 300: return True, response.status_code res_err_detail = f"request url:{response.request.url} code:{response.status_code} text:{response.text}" if is_exit: sys.exit(1) if excption: raise excption(res_err_detail) return False, response.status_code
上图就是创建的方法,在我们调用API时只需要去调用该方法就能知道,访问的 方法是否正确。调用的代码如下:
response = requests.get(folder_url, headers=self.headers, verify=False) print(folder_url) result, _ = handle_response(response, is_exit=False, excption=SPCallException)
if result:
'''说明有返回值True,是正常的数据‘’‘
else:
'''URL出现错误False,没有进行返回数据'''