_utils/custom_exception_handler.py

# myapp/exception_handler.py

from datetime import datetime
import pytz
from typing import Any, Dict, Union
from rest_framework.views import exception_handler
import logging
from _utils.bubble_integrations.enviar_resposta_final import enviar_resposta_final
from gerar_documento.serializer import GerarDocumentoSerializerData
from setup.logging import Axiom

logger = logging.getLogger(__name__)


def custom_exception_handler(exc, context):
    if context:
        serializer: Dict = context["view"].serializer
        axiom_instance: Axiom = context["view"].axiom_instance
    else:
        serializer = {}
        axiom_instance: Axiom = Axiom()
    axiom_instance.send_axiom_error(
        "---------------- CHEGOU NA FUNÇÃO PERSONALIZADA DE ERRO ----------------"
    )
    axiom_instance.send_axiom_error("INICIANDO RESPOSTA DE ERRO PARA O BUBBLE")
    resposta_bubble = enviar_resposta_final(
        serializer.get("doc_id", ""),
        serializer.get("form_response_id", ""),
        serializer.get("version", ""),
        serializer.get("texto_completo", ""),
        True,
    )
    axiom_instance.send_axiom_error(
        f"resposta_bubble.status_code: {resposta_bubble.status_code}"
    )
    axiom_instance.send_axiom_error(f"resposta_bubble.text: {resposta_bubble.text}")
    # Call REST framework's default exception handler first
    response = exception_handler(exc, context)

    if response and str(response.status_code)[0] != "2":
        axiom_instance.send_axiom_error(f"Validation error: {response.data}")
        logger.error(f"Validation error: {response.data}")

    return response


def custom_exception_handler_wihout_api_handler(
    error, serializer: Union[GerarDocumentoSerializerData, Any], axiom_instace: Axiom
):
    bahia_tz = pytz.timezone("America/Bahia")
    axiom_instace.send_axiom_error("INICIANDO RESPOSTA DE ERRO PARA O BUBBLE")
    resposta_bubble = enviar_resposta_final(
        serializer.doc_id,
        serializer.form_response_id,
        serializer.version,
        f"------------ ERRO NO BACKEND ÀS {datetime.now(bahia_tz).strftime("%d/%m/%Y - %H:%M:%S")} ------------:\nMensagem de erro: {error} ",  # serializer.get("texto_completo", ""),
        True,
    )
    axiom_instace.send_axiom_error(
        f"resposta_bubble.status_code: {resposta_bubble.status_code}"
    )
    axiom_instace.send_axiom_error(f"resposta_bubble.text: {resposta_bubble.text}")
    axiom_instace.send_axiom_error(f"------------ MOTIVO DO ERRO -----------: {error}")