如何用PyMySQL获取MySQL错误类型？

Question

我正在用Python做一个应用程序，使用的是MySQL和PyMySQL。我想在遇到MySQL错误时，能够知道错误的编号，这样我就可以根据不同的错误做出不同的处理。

有没有办法通过try-except语句或者其他方法来实现这个呢？

Answer 1

for name, ddl in TABLES.iteritems():
    try:
        print("Creating table {}: ".format(name))
        db.execute(ddl)
    except pymysql.InternalError as error:
        code, message = error.args
        print ">>>>>>>>>>>>>", code, message

这只是个开始，但还有很多其他错误，比如说操作错误（OperationalError）。

Answer 2

pymysql会根据下面的表格，把MySQL的错误转换成Python的错误：

_map_error(ProgrammingError, ER.DB_CREATE_EXISTS, ER.SYNTAX_ERROR,
       ER.PARSE_ERROR, ER.NO_SUCH_TABLE, ER.WRONG_DB_NAME,
       ER.WRONG_TABLE_NAME, ER.FIELD_SPECIFIED_TWICE,
       ER.INVALID_GROUP_FUNC_USE, ER.UNSUPPORTED_EXTENSION,
       ER.TABLE_MUST_HAVE_COLUMNS, ER.CANT_DO_THIS_DURING_AN_TRANSACTION)
_map_error(DataError, ER.WARN_DATA_TRUNCATED, ER.WARN_NULL_TO_NOTNULL,
       ER.WARN_DATA_OUT_OF_RANGE, ER.NO_DEFAULT, ER.PRIMARY_CANT_HAVE_NULL,
       ER.DATA_TOO_LONG, ER.DATETIME_FUNCTION_OVERFLOW)
_map_error(IntegrityError, ER.DUP_ENTRY, ER.NO_REFERENCED_ROW,
       ER.NO_REFERENCED_ROW_2, ER.ROW_IS_REFERENCED, ER.ROW_IS_REFERENCED_2,
       ER.CANNOT_ADD_FOREIGN, ER.BAD_NULL_ERROR)
_map_error(NotSupportedError, ER.WARNING_NOT_COMPLETE_ROLLBACK,
       ER.NOT_SUPPORTED_YET, ER.FEATURE_DISABLED, ER.UNKNOWN_STORAGE_ENGINE)
_map_error(OperationalError, ER.DBACCESS_DENIED_ERROR, ER.ACCESS_DENIED_ERROR,
       ER.CON_COUNT_ERROR, ER.TABLEACCESS_DENIED_ERROR,
       ER.COLUMNACCESS_DENIED_ERROR)

如果你想要捕捉这些错误，就需要分别处理ProgrammingError、DataError、IntegrityError、NotSupportedError和OperationalError这几种错误。你可以通过把异常转换成字符串来查看具体捕捉到了哪个MySQL错误，方法是使用str。

try:
    #interact with pymysql
except ProgrammingError as e:
    print "Caught a Programming Error:",
    print e

Answer 3

在Python中，任何异常都有一个叫做 args 的成员，它可以告诉你这个异常是怎么产生的。例如：

>>> e = Exception(1, 2, 3, 4)
>>> e.args
(1, 2, 3, 4)

对于pymysql来说，它们总是用 (errno, errorvalue) 这种方式来构造的。所以：

try:
    do_stuff()
except MySQLError as e:
    print('Got error {!r}, errno is {}'.format(e, e.args[0]))

我不确定这个在文档中是否有保证，但你可以很容易地从 源代码 中看到它是怎么工作的。