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 (file)
--- 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):
     """