From: rajdeep11 <rajdeep.sin@samsung.com>
Date: Tue, 25 Apr 2023 04:58:41 +0000 (+0530)
Subject: adding query
X-Git-Tag: 1.1.0~21
X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=6bb373cd8e33353d66407cdab15261a65a108de4;p=aiml-fw%2Fawmf%2Ftm.git

adding query

Issue-Id: AIMLFW-40

Change-Id: Id5962643499f25aa5cdf12e125880ab96f7271f2
Signed-off-by: rajdeep11 <rajdeep.sin@samsung.com>
---

diff --git a/trainingmgr/db/common_db_fun.py b/trainingmgr/db/common_db_fun.py
index 484def8..46de3e4 100644
--- a/trainingmgr/db/common_db_fun.py
+++ b/trainingmgr/db/common_db_fun.py
@@ -583,6 +583,53 @@ def add_featuregroup(featureGroup_name, feature_list, datalake_source, enable_Dm
         if conn is not None:
             conn.close()
 
+def get_feature_groups_db(ps_db_obj):
+    """
+    This function returns feature_groups
+    """
+    conn=None
+    result=None
+    try:
+        conn=ps_db_obj.get_new_conn()
+        cursor=conn.cursor()
+        query = '''select * from {}'''.format(fg_table_name)
+        cursor.execute(query)
+        result=cursor.fetchall()
+        conn.commit()
+        cursor.close()
+    except Exception as err:
+        if conn is not None:
+            conn.rollback()
+        raise DBException(DB_QUERY_EXEC_ERROR + \
+            "get_feature_groups"  + str(err))
+    finally:
+        if conn is not None:
+            conn.close()
+    return result
+
+def get_feature_group_by_name_db(ps_db_obj, featuregroup_name):
+    """
+    This Function return a feature group with name "featuregroup_name"
+    """
+    conn=None
+    result=None
+    try:
+        conn=ps_db_obj.get_new_conn()
+        cursor=conn.cursor()
+        cursor.execute(''' select * from  {} where featuregroup_name = %s '''.format(fg_table_name),
+                       (featuregroup_name, ))
+        result=cursor.fetchall()
+        conn.commit()
+        cursor.close()
+    except Exception as err:
+        if conn is not None:
+            conn.rollback()
+        raise DBException(DB_QUERY_EXEC_ERROR + \
+            "get_feature_groups"  + str(err))
+    finally:
+        if conn is not None:
+            conn.close()
+    return result 
      
 def get_all_jobs_latest_status_version(ps_db_obj):
     """