我正在尝试标准化我对从各种API返回的HTTP状态代码的处理,以减少我在文件中复制的代码量。
对于我当前的应用程序,我有一组文件,每个文件包含一个类,它们都从一个主类派生继承。我正在使用Python请求模块来使用API。
到目前为止,我已经在每个函数中编写了自定义状态代码处理以继续200,记录我用400发送的请求,记录404的URL,重试5xx,但是继续复制是一件麻烦事这个代码跨函数和类。
我正在考虑以下内容(请注意,我已经将我的代码简化为仅使用GET,但实际上,我主要是POST并接收json响应):
apiMaster.py
class ApiMaster(object):
def _handle_response(self, resp):
if resp.status_code == 200: # or requests.code.ok
return resp.json()
if resp.status_code == 400:
err_msg = "400 Error - Bad Request\n" + resp.request.url + "\n" + resp.request.data
raise HTTPError(err_msg)
...
apiA.py
class Api_A(ApiMaster):
def query_json_a(self):
resp = requests.get(self.url + '/a.json')
try:
resp_json = self._handle_response(resp)
except HTTPError as e:
logger.error(str(e))
apiB.py
class Api_B(ApiMaster):
def query_json_b(self):
# same as API A but with different endpoints and purpose
然而,这似乎仍然不确定。我不知道尝试考虑API返回的每个状态代码是否合理。这也需要我将此代码复制到我开始处理的任何新项目中。
有更好的方法吗?
答案 0 :(得分:0)
您的解决方案似乎很好,您可以通过将更多内容移动到共享方法中来进一步实现,例如
class ApiMaster(object):
def make_request(self, url, method, query_params=None, body=None, logger=None):
if not logger:
get_default_logger_somehow()
response = requests.request(method, url, params=query_params, json=body)
if response.status_code == 400:
logger.error(response.request.url)
elif response.status_code == 200:
try:
return reponse.json()
except SomeError:
return None
class Api1(ApiMaster):
json = self.make_request('post', 'http://example/com')