处理 PyMySql 异常 - 最佳实践

Posted 2023-02-23

【中文标题】处理 PyMySql 异常 - 最佳实践

【英文标题】：Handling PyMySql exceptions - Best Practices

【发布时间】：2017-05-11 16:21:20

【问题描述】：

我的问题是关于异常最佳实践的。 我将就 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)

提前感谢您的指导，

【参考方案1】：

有一种非常通用的方式来使用 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]))

【参考方案2】：

显然，您不能基于异常实例属性指定 expect 子句，甚至不能基于异常类属性 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 更好取决于您...

【讨论】：

PYMYSQL_DUPLICATE_ERROR 代码是 1062。我必须在我的代码中设置它，而 e.args[0] 只是一个整数值。