Fixed exception message

Author: sujata-m

src/python_ms_core/core/queue/models/queue_message.py

@@ -1,8 +1,8 @@
 import json
 from datetime import datetime
 from typing import Union, List
-from dataclasses import dataclass, asdict
-from typing import Optional
+from dataclasses import dataclass
+from ...resource_errors import ExceptionHandler
 
 
 class Validations:
@@ -42,6 +42,7 @@ def send(self):
     def get_items(self):
         return self.queue
 
+    @ExceptionHandler.decorated
     def data_from(self):
         data = self
         if isinstance(data, str):
@@ -56,7 +57,8 @@ def data_from(self):
                 return QueueMessage(**kwargs)
             except Exception as e:
                 error = str(e).replace('QueueMessage', 'Invalid parameter,')
-                raise TypeError(error)
+                error = error.replace('__init__()', 'QueueMessage')
+                raise TypeError(f'{error} \nStatus code: 400')
 
     def to_dict(self):
         if isinstance(self, QueueMessage):

src/python_ms_core/core/resource_errors/__init__.py

@@ -1,4 +1,5 @@
 import azure
+import operator
 from functools import wraps
 from .errors import BadRequestError, UnauthorizedError, ForbiddenError, NotFoundError, ConflictError, \
     UnProcessableError, TooManyRequestError, ServiceError, TimeOutError
@@ -8,15 +9,19 @@ def parse_error(base_message, error):
     message = error
     status_code = 0
     description = ''
-    complete_message = error.split('\n')
-    if len(complete_message) > 0:
+    if error.__class__ == TypeError:
+        error = str(error)
+
+    complete_message = error.split('\n') if operator.contains(str(error), '\n') else [error]
+
+    if complete_message and len(complete_message) > 0:
         message = complete_message[0]
-    if len(complete_message) > 1:
+    if complete_message and len(complete_message) > 1:
         status_code_string = complete_message[1]
         status_code_string = status_code_string.split('Status code: ')
         if len(status_code_string) == 2:
             status_code = int(status_code_string[1])
-    if len(complete_message) > 2:
+    if complete_message and len(complete_message) > 2:
         description_string = complete_message[2]
         description_string = description_string.split('Description: ')
         if len(description_string) == 2: