error message update for FG get request

Issue Id- AIMLFW-181
Change-Id: I0f2c1c57d2df6e37794070504f4a257133e548ff
Signed-off-by: Swaraj Kumar <swaraj.kumar@samsung.com>

diff --git a/trainingmgr/controller/featuregroup_controller.py b/trainingmgr/controller/featuregroup_controller.py
index 385bfee..e0e2f5e 100644 (file)
--- a/trainingmgr/controller/featuregroup_controller.py
+++ b/trainingmgr/controller/featuregroup_controller.py
@@ -157,11 +157,9 @@ def get_feature_group():
     """
     LOGGER.debug("Request for getting all feature groups")
     api_response={}
-    response_code=status.HTTP_500_INTERNAL_SERVER_ERROR
     try:
-        api_response=featuregroups_schema.dump(get_all_featuregroups())
-        response_code=status.HTTP_200_OK    
+        api_response = featuregroups_schema.dump(get_all_featuregroups())
+        return jsonify({"FeatureGroups": api_response}), 200
     except Exception as err:
-        api_response =   {"Exception": "Failed to get featuregroups"}
-        LOGGER.error(str(err))
-    return jsonify({"FeatureGroups":api_response}), response_code
\ No newline at end of file
+        LOGGER.error(f"Failed to get featuregroups: {str(err)}")
+        return ProblemDetails(500, "Internal Server Error", "Failed to get featuregroups").to_json()