a1-policy-manager-vth/app/helpers/error_helper.py

   1 from flask import current_app\r
   2 import datetime\r
   3 """\r
   4 Args:\r
   5 Returns:\r
   6 Examples:\r
   7 """\r
   8 \r
   9 def error_dic(error, status_code, response_message="Something went wrong, vth encountered an error"):\r
  10     """\r
  11     Args:\r
  12     Returns:\r
  13     Examples:\r
  14     """\r
  15     message = [str(x) for x in error.args]\r
  16     error_log={\r
  17             "error":{\r
  18                 "type": error.__class__.__name__,\r
  19                 "message": message\r
  20                 }\r
  21             }\r
  22     response_data = {\r
  23         "vthResponse": {\r
  24             "testDurationMS": 0,\r
  25             'dateTimeUTC': str(datetime.datetime.now()),\r
  26             "abstractMessage": "Failed",\r
  27             "error":response_message,\r
  28             "status_code": status_code,\r
  29             "resultData": {}\r
  30         }\r
  31     }\r
  32     current_app.logger.error(error_log)\r
  33     return response_data\r
  34 \r
  35 def error_dic2(error, status_code=500):\r
  36     """\r
  37     Args:\r
  38     Returns:\r
  39     Examples:\r
  40     """\r
  41     message = [str(x) for x in error.args]\r
  42     response = {\r
  43         "status_code" : status_code,\r
  44         "success": False,\r
  45         "error":{\r
  46             "type": error.__class__.__name__,\r
  47             "message": message\r
  48             }\r
  49         }\r
  50     return response\r
  51 \r