1 files changed, 14 insertions, 1 deletions
diff --git a/wqflask/base/data_set.py b/wqflask/base/data_set.py
index 5d21c901..16f9da5d 100755
--- a/wqflask/base/data_set.py
+++ b/wqflask/base/data_set.py
@@ -1078,7 +1078,20 @@ class MrnaAssayDataSet(DataSet):
 
     def retrieve_gene_symbols(self):
         query = """
-                    select ProbeSet.Name, ProbeSet.Symbol
+                    select ProbeSet.Name, ProbeSet.Symbol, ProbeSet.GeneId
+                    from ProbeSet,ProbeSetXRef
+                    where ProbeSetXRef.ProbeSetFreezeId = %s and
+                    ProbeSetXRef.ProbeSetId=ProbeSet.Id;
+                """ % (self.id)
+        results = g.db.execute(query).fetchall()
+        symbol_dict = {}
+        for item in results:
+            symbol_dict[item[0]] = item[1]
+        return symbol_dict
+    
+    def retrieve_gene_ids(self):
+        query = """
+                    select ProbeSet.Name, ProbeSet.GeneId
                     from ProbeSet,ProbeSetXRef
                     where ProbeSetXRef.ProbeSetFreezeId = %s and
                     ProbeSetXRef.ProbeSetId=ProbeSet.Id;