From 01e12ddb0c7f71c8c1700a4fac0b9447d83f3f9f Mon Sep 17 00:00:00 2001
From: Munyoki Kilyungi
Date: Wed, 15 Feb 2023 00:24:57 +0300
Subject: Fetch trait metadata from RDF

* gn3/db/rdf.py (get_dataset_metadata): New function.

Signed-off-by: Munyoki Kilyungi <me@bonfacemunyoki.com>
---
 gn3/db/rdf.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

(limited to 'gn3/db/rdf.py')

diff --git a/gn3/db/rdf.py b/gn3/db/rdf.py
index f46e33c..deecefa 100644
--- a/gn3/db/rdf.py
+++ b/gn3/db/rdf.py
@@ -90,3 +90,31 @@ WHERE {
     )[0].items():
         result[key] = value.bind(lambda x: Just(x["value"]))
     return result
+
+
+def get_trait_metadata(
+        sparql_conn: SPARQLWrapper,
+        trait_name: str,
+        dataset_name: str
+):
+    """Return metadata about a given trait"""
+    __metadata_query = """
+PREFIX gn: <http://genenetwork.org/>
+
+SELECT strafter((str(?key)), "http://genenetwork.org/sampledata:") as ?key
+    ?value WHERE {
+    gn:sampledata_$trait_name gn:sampledata:dataset "$dataset_name" .
+    gn:sampledata_$trait_name ?key ?value .
+}
+"""
+    result: MonadicDict = MonadicDict()
+    for _r in sparql_query(
+            sparql_conn,
+            Template(__metadata_query)
+            .substitute(trait_name=trait_name,
+                        dataset_name=dataset_name)
+    ):
+        _key = _r["key"].bind(lambda x: x["value"])  # type:ignore
+        if _key:
+            result[_key] = _r["value"].bind(lambda x: Just(x["value"]))  # type:ignore
+    return result
-- 
cgit v1.2.3