如何在Python应用程序中处理HTTP状态代码?

时间:2018-01-18 16:30:07

标签: python http error-handling python-requests http-error

我正在尝试标准化我对从各种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返回的每个状态代码是否合理。这也需要我将此代码复制到我开始处理的任何新项目中。

有更好的方法吗?

1 个答案:

答案 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')