处理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)

感谢您的指导。

- oria general

Answer 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]))

- lalit gangwar

Answer 2

显然，您无法基于异常实例属性或异常类属性指定expect子句--它只适用于异常类型。

解决您的问题的方法是有两个嵌套的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更好，都取决于您...

- bruno desthuilliers

---

PYMYSQL_DUPLICATE_ERROR 错误代码为 1062。我不得不在我的代码中设置它，而 e.args[0] 只是一个整数值。 - Nasif Imtiaz Ohi