在 Python 中检查函数的返回值

Question

我在Django中有一个用于数据库连接的小功能如下：

def db_connection(query_name):
    try:
        cursor = connection.cursor()
        cursor.execute(query_name)
        descr = cursor.description
        rows = cursor.fetchall()
        result = [dict(zip([column[0] for column in descr], row)) for row in rows]
        return result
    except Exception as e:
        return e 
    finally:
       cursor.close()

这个函数在我的函数视图中用于Django rest框架执行SQL查询。一般语法如下：

@api_view(['GET'])
def foo_bar(request):
    ....
    ....
    ....
    query1 = "Select name from table"
    result = db_connection(query1)

    return Response(result, status=status.HTTP_200_OK)

但是，我需要根据我的 db_connection 函数的返回值更改响应元组中的状态值，即如果没有异常发生则返回 200 OK，否则返回 500。如何检查返回值是否函数是异常？

Answer 1

您明确地绕过了异常已经为您提供的所有良好控制流。不要那样做。

如果您只想返回 500，则根本不要捕获异常；让它冒泡，Django 的异常处理中间件会捕获它并返回 500 响应。

在任何情况下，您都不应该捕获基类 Exception ；只抓住你实际可以处理的事情。当你确实捕获了一个异常时，你应该实际上处理它；归还它而不是提高它是没有任何用处的。

Answer 2

我会将foo_bar 函数中的try catch 块移到result = db_connection(query1) 代码上，并从那里返回500 状态，以及某种错误字典，其中包含解释失败原因的消息。我这样说是因为我假设您正在制作一个 REST API，它必须始终返回答案并且不会以正常的 django 方式崩溃。