from keycloak.exceptions import KeycloakDeleteError, KeycloakPostError


class MyVbvKeycloakDeleteError(KeycloakDeleteError):
    def __init__(
        self, keycloak_error: KeycloakDeleteError, additional_data: list | dict
    ):
        super().__init__(
            keycloak_error.error_message,
            keycloak_error.response_code,
            keycloak_error.response_body,
        )
        self.additional_data = additional_data


class MyVbvKeycloakPostError(KeycloakPostError):
    def __init__(self, keycloak_error: KeycloakPostError, additional_data: list | dict):
        super().__init__(
            keycloak_error.error_message,
            keycloak_error.response_code,
            keycloak_error.response_body,
        )
        self.additional_data = additional_data