Adding policy manager and a1 sdnc vth

[it/otf.git] / a1-policy-manager-vth / app / helpers / error_helper.py

diff --git a/a1-policy-manager-vth/app/helpers/error_helper.py b/a1-policy-manager-vth/app/helpers/error_helper.py
new file mode 100644 (file)
index 0000000..b34cedf
--- /dev/null
+++ b/a1-policy-manager-vth/app/helpers/error_helper.py
@@ -0,0 +1,51 @@
+from flask import current_app\r
+import datetime\r
+"""\r
+Args:\r
+Returns:\r
+Examples:\r
+"""\r
+\r
+def error_dic(error, status_code, response_message="Something went wrong, vth encountered an error"):\r
+    """\r
+    Args:\r
+    Returns:\r
+    Examples:\r
+    """\r
+    message = [str(x) for x in error.args]\r
+    error_log={\r
+            "error":{\r
+                "type": error.__class__.__name__,\r
+                "message": message\r
+                }\r
+            }\r
+    response_data = {\r
+        "vthResponse": {\r
+            "testDurationMS": 0,\r
+            'dateTimeUTC': str(datetime.datetime.now()),\r
+            "abstractMessage": "Failed",\r
+            "error":response_message,\r
+            "status_code": status_code,\r
+            "resultData": {}\r
+        }\r
+    }\r
+    current_app.logger.error(error_log)\r
+    return response_data\r
+\r
+def error_dic2(error, status_code=500):\r
+    """\r
+    Args:\r
+    Returns:\r
+    Examples:\r
+    """\r
+    message = [str(x) for x in error.args]\r
+    response = {\r
+        "status_code" : status_code,\r
+        "success": False,\r
+        "error":{\r
+            "type": error.__class__.__name__,\r
+            "message": message\r
+            }\r
+        }\r
+    return response\r
+\r