Source code for sqlpuzzle.exceptions

[docs]class SqlPuzzleException(Exception):
    """
    Base exception. If you want to handle any exception of SQL puzzle in same
    way, catch this one.
    """

    def __init__(self, message):
        super().__init__()
        self.message = message

    def __str__(self):
        return "SqlPuzzleException: %s" % self.message


[docs]class ConfirmException(SqlPuzzleException):
    """
    Base confirm exception. If you want to handle any confirm exception
    (of update or delete) in same way, catch this one.
    """

    # pylint: disable=super-init-not-called
    def __init__(self):
        pass


[docs]class ConfirmUpdateAllException(ConfirmException):
    """
    Raises when you want render update SQL without any where condition. It is
    security before changing all data by mistake.
    """

    def __str__(self):
        return "Are you sure, that you want update all records?"


[docs]class ConfirmDeleteAllException(ConfirmException):
    """
    Raises when you want render delete SQL without any where condition. It is
    security before droping all data by mistake.
    """

    def __str__(self):
        return "Are you sure, that you want delete all records?"


[docs]class InvalidArgumentException(SqlPuzzleException):
    """
    Raises when you pass invalid argument into SQL puzzle. For example instead
    of column reference some number and so.

    .. code-block:: python

        >>> sqlpuzzle.select(True)
        Traceback (most recent call last):
          ...
        InvalidArgumentException: Invalid argument: column_name cannot be of type <type 'bool'>.
    """

    def __init__(self, message=''):
        super().__init__(message)

    def __str__(self):
        if self.message:
            return "Invalid argument: %s" % self.message
        return "Invalid argument"


[docs]class InvalidQueryException(InvalidArgumentException):
    """
    Specific type of :py:class:`~.InvalidArgumentException`. Raises when you
    passed good arguments but final query does not make sense.

    .. code-block:: python

        >>> sqlpuzzle.select_from('t').on('t2')
        Traceback (most recent call last):
          ...
        InvalidQueryException: Invalid query: You can not set join condition to nothing. Specify join table first.
    """

    def __str__(self):
        if self.message:
            return "Invalid query: %s" % self.message
        return "Invalid query"