From 204bcd649f9cf4a4e5f741302abbe37a2d7eb759 Mon Sep 17 00:00:00 2001
From: naglepuff <mike.nagler@kitware.com>
Date: Thu, 19 Dec 2024 14:37:17 -0500
Subject: [PATCH] Use gene query logic for all gene table values

---
 nmdc_server/query.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/nmdc_server/query.py b/nmdc_server/query.py
index 4e19b946..9cdfe17d 100644
--- a/nmdc_server/query.py
+++ b/nmdc_server/query.py
@@ -384,7 +384,13 @@ def query(self, db) -> Query:
 
             # Gene function queries are treated differently because they join
             # in three different places (metaT, metaG and metaP).
-            if table == Table.gene_function:
+            if table in [
+                Table.gene_function,
+                Table.kegg_function,
+                Table.go_function,
+                Table.pfam_function,
+                Table.cog_function,
+            ]:
                 metag_matches = filter.matches(db, self.table)
                 metap_conditions = [
                     SimpleConditionSchema(