"""Error classes."""

from enum import IntEnum


class AskarErrorCode(IntEnum):
    SUCCESS = 0
    BACKEND = 1
    BUSY = 2
    DUPLICATE = 3
    ENCRYPTION = 4
    INPUT = 5
    NOT_FOUND = 6
    UNEXPECTED = 7
    UNSUPPORTED = 8
    WRAPPER = 99
    CUSTOM = 100


class AskarError(Exception):
    def __init__(self, code: AskarErrorCode, message: str, extra: str = None):
        super().__init__(message)
        self.code = code
        self.extra = extra