处理PyMySql异常 - 最佳实践

时间:2016-12-27 11:17:20

标签: python mysql error-handling exception-handling

我的问题是关于例外最佳做法。 我将在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,这导致我的异常catch中的多余情况。

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)

提前感谢您的指导,

2 个答案:

答案 0 :(得分:4)

您不能明显地基于异常实例属性指定expect子句,甚至不能在异常类属性FWIW上指定 - 它仅适用于异常类型。

你的问题的解决方案是有两个嵌套的try / except块,内部的一个处理重复的条目并重新引发其他IntegrityError s,外部的是一般案例:

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更好取决于你......

答案 1 :(得分:0)

有使用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]))