python - Python-MySQL 中的错误处理

Question

我正在运行一个基于 python flask 的小网络服务，我想在其中执行一个小的 MySQL 查询。当我为我的 SQL 查询获得有效输入时，一切都按预期工作，并且我得到了正确的值。但是，如果该值未存储在数据库中，我会收到一个 TypeError

    Traceback (most recent call last):
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1836, in __call__
    return self.wsgi_app(environ, start_response)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1820, in wsgi_app
    response = self.make_response(self.handle_exception(e))
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1403, in handle_exception
    reraise(exc_type, exc_value, tb)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1817, in wsgi_app
    response = self.full_dispatch_request()
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1478, in full_dispatch_request
    response = self.make_response(rv)
  File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1566, in make_response
    raise ValueError('View function did not return a response')
ValueError: View function did not return a response

我试图利用自己的错误处理并将此代码用于我的项目，但它似乎无法正常工作。

#!/usr/bin/python

from flask import Flask, request
import MySQLdb

import json

app = Flask(__name__)


@app.route("/get_user", methods=["POST"])
def get_user():
    data = json.loads(request.data)
    email = data["email"]

    sql = "SELECT userid FROM oc_preferences WHERE configkey='email' AND configvalue LIKE '" + email + "%';";

    conn = MySQLdb.connect( host="localhost",
                            user="root",
                            passwd="ubuntu",
                            db="owncloud",
                            port=3306)
    curs = conn.cursor()

    try:
        curs.execute(sql)
        user = curs.fetchone()[0]
        return user
    except MySQLdb.Error, e:
        try:
            print "MySQL Error [%d]: %s" % (e.args[0], e.args[1])
            return None
        except IndexError:
            print "MySQL Error: %s" % str(e)
            return None
    except TypeError, e:
        print(e)
        return None
    except ValueError, e:
        print(e)
        return None
    finally:
        curs.close()
        conn.close()

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000, debug=True)

基本上，当一切正常时，我只想返回一个值，如果我的服务器上没有最好的错误消息，我什么也不想返回。如何以正确的方式使用错误处理？

编辑更新了当前代码+错误消息。

Answer 1

第一点:您的 try/except block 中的代码太多。当您有两个可能引发不同错误的语句(或两组语句)时，最好使用不同的 try/except block :

try:
    try:
        curs.execute(sql)
        # NB : you won't get an IntegrityError when reading
    except (MySQLdb.Error, MySQLdb.Warning) as e:
        print(e)
        return None

    try: 
        user = curs.fetchone()[0]
        return user
    except TypeError as e:
        print(e)
        return None

finally:
    conn.close()

现在你真的必须在这里捕获 TypeError 吗？如果您阅读回溯，您会注意到您的错误来自于在 None 上调用 __getitem__()(注意:__getitem__() 是下标运算符 [] 的实现)，这意味着如果没有匹配的行 cursor.fetchone() 返回 None，因此您可以只需测试 currsor.fetchone() 的返回:

try:
    try:
        curs.execute(sql)
        # NB : you won't get an IntegrityError when reading
    except (MySQLdb.Error, MySQLdb.Warning) as e:
        print(e)
        return None

    row = curs.fetchone()
    if row:
        return row[0]
    return None

finally:
    conn.close()

现在你真的需要在这里捕获 MySQL 错误吗？您的查询应该经过良好测试，并且它只是一个读取操作，因此它不会崩溃 - 因此，如果您在这里出现问题，那么您显然遇到了更大的问题，并且您不想将其隐藏在地毯下。 IOW:要么记录异常(使用标准的 logging 包和 logger.exception())并重新引发它们，要么更简单地让它们传播(最终有更高的级别组件负责记录未处理的异常):

try:
    curs.execute(sql)
    row = curs.fetchone()
    if row:
        return row[0]
    return None

finally:
    conn.close()

最后:构建 sql 查询的方式是 utterly unsafe .改为使用 sql 占位符:

q = "%s%%" % data["email"].strip() 
sql = "select userid from oc_preferences where configkey='email' and configvalue like %s"
cursor.execute(sql, [q,])

哦，是的:wrt/“View function did not return a response”ValueError，这是因为，好吧，你的 View 在很多地方返回 None。 Flask View 应该返回可用作 HTTP 响应的内容，而 None 在这里不是有效选项。