处理 PyMySql 异常 - 最佳实践

Question

我的问题是关于异常最佳实践的。我将针对 PyMySQL 的一个特定案例提出我的问题，但它通常涉及错误处理。我正在使用 PyMySQL，在许多可能的异常中，有一个我想以特定方式处理。“重复”异常。

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)

我想专门捕获 ER.DUP_ENTRY 但我只知道如何捕获 IntegrityError 并且这会导致我的异常捕获中出现多余的情况。

cur.execute(query, values)
except IntegrityError as e:
     if e and e[0] == PYMYSQL_DUPLICATE_ERROR:
          handel_duplicate_pymysql_exception(e, func_a)
     else:
          handel_unknown_pymysql_exception(e, func_b)
except Exception as e:
     handel_unknown_pymysql_exception(e, func_b)

有没有办法简单地只捕获 ER.DUP_ENTRY 一些如何？寻找类似的东西：

except IntegrityError.DUP_ENTRY as e:
     handel_duplicate_pymysql_exception(e, func_a)

预先感谢您的指导，

Answer 1

显然，您不能基于异常实例属性指定期望子句，甚至不能在异常类属性 FWIW 上指定 - 它仅适用于异常类型。

您的问题的解决方案是有两个嵌套的 try/except 块，内部一个处理重复条目并重新引发其他IntegrityErrors，外部一个是通用情况：

try:
    try:
        cur.execute(query, values)
    except IntegrityError as e:
        if e.args[0] == PYMYSQL_DUPLICATE_ERROR:
            handle_duplicate_pymysql_exception(e, func_a)
        else:
            raise

except Exception as e:
    handle_unknown_pymysql_exception(e, func_b)

这是否比重复调用更好handle_unknown_pymysql_exception取决于您...

Answer 2

使用 pymysql 错误处理有非常通用的方法。我将它用于 sqlutil 模块。通过这种方式，您可以在不考虑其类型的情况下捕获 pymysql 引发的所有错误。

try:
    connection.close()
    print("connection closed successfully")
except pymysql.Error as e:
    print("could not close connection error pymysql %d: %s" %(e.args[0], e.args[1]))