mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
148 lines
4.2 KiB
Python
148 lines
4.2 KiB
Python
import struct
|
|
|
|
|
|
try:
|
|
StandardError, Warning
|
|
except ImportError:
|
|
try:
|
|
from exceptions import StandardError, Warning
|
|
except ImportError:
|
|
import sys
|
|
e = sys.modules['exceptions']
|
|
StandardError = e.StandardError
|
|
Warning = e.Warning
|
|
|
|
from constants import ER
|
|
import sys
|
|
|
|
class MySQLError(StandardError):
|
|
|
|
"""Exception related to operation with MySQL."""
|
|
|
|
|
|
class Warning(Warning, MySQLError):
|
|
|
|
"""Exception raised for important warnings like data truncations
|
|
while inserting, etc."""
|
|
|
|
class Error(MySQLError):
|
|
|
|
"""Exception that is the base class of all other error exceptions
|
|
(not Warning)."""
|
|
|
|
|
|
class InterfaceError(Error):
|
|
|
|
"""Exception raised for errors that are related to the database
|
|
interface rather than the database itself."""
|
|
|
|
|
|
class DatabaseError(Error):
|
|
|
|
"""Exception raised for errors that are related to the
|
|
database."""
|
|
|
|
|
|
class DataError(DatabaseError):
|
|
|
|
"""Exception raised for errors that are due to problems with the
|
|
processed data like division by zero, numeric value out of range,
|
|
etc."""
|
|
|
|
|
|
class OperationalError(DatabaseError):
|
|
|
|
"""Exception raised for errors that are related to the database's
|
|
operation and not necessarily under the control of the programmer,
|
|
e.g. an unexpected disconnect occurs, the data source name is not
|
|
found, a transaction could not be processed, a memory allocation
|
|
error occurred during processing, etc."""
|
|
|
|
|
|
class IntegrityError(DatabaseError):
|
|
|
|
"""Exception raised when the relational integrity of the database
|
|
is affected, e.g. a foreign key check fails, duplicate key,
|
|
etc."""
|
|
|
|
|
|
class InternalError(DatabaseError):
|
|
|
|
"""Exception raised when the database encounters an internal
|
|
error, e.g. the cursor is not valid anymore, the transaction is
|
|
out of sync, etc."""
|
|
|
|
|
|
class ProgrammingError(DatabaseError):
|
|
|
|
"""Exception raised for programming errors, e.g. table not found
|
|
or already exists, syntax error in the SQL statement, wrong number
|
|
of parameters specified, etc."""
|
|
|
|
|
|
class NotSupportedError(DatabaseError):
|
|
|
|
"""Exception raised in case a method or database API was used
|
|
which is not supported by the database, e.g. requesting a
|
|
.rollback() on a connection that does not support transaction or
|
|
has transactions turned off."""
|
|
|
|
|
|
error_map = {}
|
|
|
|
def _map_error(exc, *errors):
|
|
for error in errors:
|
|
error_map[error] = exc
|
|
|
|
_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)
|
|
_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.TABLEACCESS_DENIED_ERROR, ER.COLUMNACCESS_DENIED_ERROR)
|
|
|
|
del _map_error, ER
|
|
|
|
|
|
def _get_error_info(data):
|
|
errno = struct.unpack('<h', data[1:3])[0]
|
|
if sys.version_info[0] == 3:
|
|
is_41 = data[3] == ord("#")
|
|
else:
|
|
is_41 = data[3] == "#"
|
|
if is_41:
|
|
# version 4.1
|
|
sqlstate = data[4:9].decode("utf8")
|
|
errorvalue = data[9:].decode("utf8")
|
|
return (errno, sqlstate, errorvalue)
|
|
else:
|
|
# version 4.0
|
|
return (errno, None, data[3:].decode("utf8"))
|
|
|
|
def _check_mysql_exception(errinfo):
|
|
errno, sqlstate, errorvalue = errinfo
|
|
errorclass = error_map.get(errno, None)
|
|
if errorclass:
|
|
raise errorclass, (errno,errorvalue)
|
|
|
|
# couldn't find the right error number
|
|
raise InternalError, (errno, errorvalue)
|
|
|
|
def raise_mysql_exception(data):
|
|
errinfo = _get_error_info(data)
|
|
_check_mysql_exception(errinfo)
|
|
|
|
|
|
|
|
|
|
|