From: subhash kumar singh <subh.singh@samsung.com>
Date: Thu, 14 Nov 2024 16:52:41 +0000 (+0000)
Subject: Implement get operation
X-Git-Tag: 3.0.0~36
X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=249b0ed741f810d94c3dd62ffbe1fffc96091850;p=aiml-fw%2Fawmf%2Ftm.git

Implement get operation

Implemented get operation for training job by id.

Change-Id: Iefa53aa484575ff72b9a0c83fc2269eb71a9660f
Signed-off-by: subhash kumar singh <subh.singh@samsung.com>
---

diff --git a/trainingmgr/controller/trainingjob_controller.py b/trainingmgr/controller/trainingjob_controller.py
index 1369d4f..d0edb8e 100644
--- a/trainingmgr/controller/trainingjob_controller.py
+++ b/trainingmgr/controller/trainingjob_controller.py
@@ -17,7 +17,7 @@
 # ==================================================================================
 from flask import Blueprint, jsonify, request
 from trainingmgr.common.trainingmgr_config import TrainingMgrConfig
-from trainingmgr.service.training_job_service import delete_training_job, create_training_job
+from trainingmgr.service.training_job_service import delete_training_job, create_training_job, get_training_job
 
 training_job_controller = Blueprint('training_job_controller', __name__)
 LOGGER = TrainingMgrConfig().logger
@@ -47,7 +47,13 @@ def create_trainingjob():
         create_training_job(data)
         LOGGER.debug(f'create training job Successfully: {data}')
         return '', 200
+
     except Exception as e:
         return jsonify({
             'message': str(e)
-        }), 500
\ No newline at end of file
+        }), 500
+
+@training_job_controller.route('/training-jobs/<int:training_job_id>', methods=['GET'])
+def get_trainingjob(training_job_id):
+    LOGGER.debug(f'get the trainingjob correspoinding to id: {training_job_id}')
+    return jsonify(get_training_job(training_job_id)), 200
diff --git a/trainingmgr/db/trainingjob_db.py b/trainingmgr/db/trainingjob_db.py
index b57c652..bff0d6c 100644
--- a/trainingmgr/db/trainingjob_db.py
+++ b/trainingmgr/db/trainingjob_db.py
@@ -316,3 +316,7 @@ def delete_trainingjob_by_id(id: int):
     except Exception as e:
         db.session.rollback()
         raise DBException(f'{DB_QUERY_EXEC_ERROR} : {str(e)}' )
+
+def get_trainingjob(id: int):
+    tj = db.session.query(TrainingJob).get(id)
+    return tj
diff --git a/trainingmgr/service/training_job_service.py b/trainingmgr/service/training_job_service.py
index 183d5b3..285722c 100644
--- a/trainingmgr/service/training_job_service.py
+++ b/trainingmgr/service/training_job_service.py
@@ -15,10 +15,14 @@
 #   limitations under the License.
 #
 # ==================================================================================
-from trainingmgr.db.trainingjob_db import delete_trainingjob_by_id, create_trainingjob
+from trainingmgr.db.trainingjob_db import delete_trainingjob_by_id, create_trainingjob, get_trainingjob
 from trainingmgr.common.exceptions_utls import DBException
 from trainingmgr.schemas import TrainingJobSchema
 
+def get_training_job(training_job_id: int):
+    tj =get_trainingjob(training_job_id)
+    return TrainingJobSchema().dump(tj)
+
 def create_training_job(data):
     create_trainingjob(data)