From c9ae69a30a972f47232f8457e9e1b8cd514f9832 Mon Sep 17 00:00:00 2001
From: Alexander Kabui
Date: Sat, 9 Oct 2021 22:04:17 +0300
Subject: add function to process trait sample data

---
 wqflask/wqflask/wgcna/gn3_wgcna.py | 44 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

(limited to 'wqflask')

diff --git a/wqflask/wqflask/wgcna/gn3_wgcna.py b/wqflask/wqflask/wgcna/gn3_wgcna.py
index f7ed4cef..9ab6b3e0 100644
--- a/wqflask/wqflask/wgcna/gn3_wgcna.py
+++ b/wqflask/wqflask/wgcna/gn3_wgcna.py
@@ -1,6 +1,7 @@
 """module contains code to consume gn3-wgcna api
 and process data to be rendered by datatables
 """
+from utility.helper_functions import get_trait_db_obs
 
 
 def fetch_trait_data(requestform):
@@ -13,6 +14,49 @@ def fetch_trait_data(requestform):
     return process_dataset(db_obj.trait_list)
 
 
+def process_dataset(trait_list):
+    """process datasets and strains"""
+
+    input_data = {}
+    traits = []
+    strains = []
+
+    # xtodo unique traits and strains
+
+    for trait in trait_list:
+        traits.append(trait[0].name)
+
+        input_data[trait[0].name] = {}
+        for strain in trait[0].data:
+            strains.append(strain)
+            input_data[trait[0].name][strain] = trait[0].data[strain].value
+
+    return {
+        "wgcna_input": input_data
+    }
+
+    def process_dataset(trait_list):
+    """process datasets and strains"""
+
+    input_data = {}
+    traits = []
+    strains = []
+
+    # xtodo unique traits and strains
+
+    for trait in trait_list:
+        traits.append(trait[0].name)
+
+        input_data[trait[0].name] = {}
+        for strain in trait[0].data:
+            strains.append(strain)
+            input_data[trait[0].name][strain] = trait[0].data[strain].value
+
+    return {
+        "wgcna_input": input_data
+    }
+
+
 def process_wgcna_data(response):
     """function for processing modeigene genes
     for create row data for datataba"""
-- 
cgit v1.2.3