aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--wqflask/wqflask/correlation/pre_computes.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/wqflask/wqflask/correlation/pre_computes.py b/wqflask/wqflask/correlation/pre_computes.py
index cb2f4470..95047fc7 100644
--- a/wqflask/wqflask/correlation/pre_computes.py
+++ b/wqflask/wqflask/correlation/pre_computes.py
@@ -167,3 +167,14 @@ def get_datasets_data(base_dataset, target_dataset_data):
samples_fetched, base_traits_data)
return (target_results, base_results)
+
+def fetch_text_file(dataset_name, text_dir, conn):
+ """fetch textfiles with strain vals if exists"""
+ with conn.cursor() as cursor:
+ query = 'SELECT Id, FullName FROM ProbeSetFreeze WHERE Name = "%s"' % dataset_name
+ cursor.execute(query)
+ results = cursor.fetchone()
+ if (results):
+ for file in os.listdir(text_dir):
+ if file.startswith(f"ProbeSetFreezeId_{results[0]}_"):
+ return os.path.join(text_dir, file)