From b7c77e2e846a6ddcf308fef5dfb9941f44d6e72a Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 26 Oct 2021 08:59:30 +0300
Subject: Implement `fetch_gene_symbol_tissue_value_dict_for_trait`

Issue:
https://github.com/genenetwork/gn-gemtext-threads/blob/main/topics/gn1-migration-to-gn2/partial-correlations.gmi

* Implement `fetch_gene_symbol_tissue_value_dict_for_trait` function which is
  a migration of the
  `web.webqtl.correlation.correlationFunction.getGeneSymbolTissueValueDictForTrait`
  function in GeneNetwork1.
---
 gn3/db/correlations.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'gn3/db')

diff --git a/gn3/db/correlations.py b/gn3/db/correlations.py
index 87ab082..cae8080 100644
--- a/gn3/db/correlations.py
+++ b/gn3/db/correlations.py
@@ -266,7 +266,22 @@ def fetch_tissue_probeset_xref_info(
         (tuple(), {}, {}, {}, {}, {}, {}))
 
 def correlations_of_all_tissue_traits() -> Tuple[dict, dict]:
+def fetch_gene_symbol_tissue_value_dict_for_trait(
+        gene_name_list: Tuple[str, ...], probeset_freeze_id: int,
+        conn: Any) -> dict:
+    """
+    Fetches a map of the gene symbols to the tissue values.
+
+    This is a migration of the
+    `web.webqtl.correlation.correlationFunction.getGeneSymbolTissueValueDictForTrait`
+    function in GeneNetwork1.
     """
+    xref_info = fetch_tissue_probeset_xref_info(
+        gene_name_list, probeset_freeze_id, conn)
+    if xref_info[0]:
+        return fetch_gene_symbol_tissue_value_dict(xref_info[0], xref_info[2], conn)
+    return {}
+
     This is a migration of the
     `web.webqtl.correlation.CorrelationPage.calculateCorrOfAllTissueTrait`
     function in GeneNetwork1.
-- 
cgit v1.2.3