[fix] Added Raise after exception connection error nlp

weni-ai · Jul 12, 2019 · 3aea9d4 · 3aea9d4
1 parent d49fab9
commit 3aea9d4
Showing 1 changed file with 25 additions and 15 deletions.
diff --git a/bothub/common/models.py b/bothub/common/models.py
@@ -184,24 +184,34 @@ class Meta:
 
     @classmethod
     def request_nlp_train(cls, user_authorization):
-        r = requests.post(  # pragma: no cover
-            cls.nlp_train_url,
-            data={},
-            headers={'Authorization': 'Bearer {}'.format(
-                user_authorization.uuid)})
-        return r  # pragma: no cover
+        try:
+            r = requests.post(  # pragma: no cover
+                cls.nlp_train_url,
+                data={},
+                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)
 
     @classmethod
     def request_nlp_analyze(cls, user_authorization, data):
-        r = requests.post(  # pragma: no cover
-            cls.nlp_analyze_url,
-            data={
-                'text': data.get('text'),
-                'language': data.get('language'),
-            },
-            headers={'Authorization': 'Bearer {}'.format(
-                user_authorization.uuid)})
-        return r  # pragma: no cover
+        try:
+            r = requests.post(  # pragma: no cover
+                cls.nlp_analyze_url,
+                data={
+                    'text': data.get('text'),
+                    '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)
 
     @classmethod
     def request_nlp_evaluate(cls, user_authorization, data):