From d49fab93b810d3109a419fb4422546f27d00fbe0 Mon Sep 17 00:00:00 2001 From: Daniel Yohan Date: Thu, 11 Jul 2019 16:57:01 -0300 Subject: [PATCH] [fix] Added Raise after exception connection error nlp --- bothub/common/models.py | 23 +++++++++++++++-------- 1 file changed, 15 insertions(+), 8 deletions(-) diff --git a/bothub/common/models.py b/bothub/common/models.py index b19d3dfb2..98fb9fd91 100644 --- a/bothub/common/models.py +++ b/bothub/common/models.py @@ -12,6 +12,8 @@ from django.template.loader import render_to_string from django.dispatch import receiver from django.core.exceptions import ValidationError +from rest_framework import status +from rest_framework.exceptions import APIException from bothub.authentication.models import User @@ -203,14 +205,19 @@ def request_nlp_analyze(cls, user_authorization, data): @classmethod def request_nlp_evaluate(cls, user_authorization, data): - r = requests.post( # pragma: no cover - cls.nlp_evaluate_url, - data={ - 'language': data.get('language'), - }, - headers={'Authorization': 'Bearer {}'.format( - user_authorization.uuid)}) - return r # pragma: no cover + try: # pragma: no cover + r = requests.post( # pragma: no cover + cls.nlp_evaluate_url, + data={ + 'language': data.get('language'), + }, + headers={'Authorization': 'Bearer {}'.format( + user_authorization.uuid)}) + return r # pragma: no cover + except requests.exceptions.ConnectionError: # pragma: no cover + raise APIException( # pragma: no cover + {'status_code': status.HTTP_503_SERVICE_UNAVAILABLE}, + code=status.HTTP_503_SERVICE_UNAVAILABLE) @property def available_languages(self):