cleanup

src/gene/query.py

@@ -21,6 +21,7 @@
 from gene.database import AbstractDatabase, DatabaseReadException
 from gene.schemas import (
     NAMESPACE_TO_SYSTEM_URI,
+    SYSTEM_URI_TO_NAMESPACE,
     BaseGene,
     BaseNormalizationService,
     Gene,
@@ -347,9 +348,9 @@ def _add_merged_meta(self, response: NormalizeService) -> NormalizeService:
 
         sources = []
         for m in gene.mappings or []:
-            for ns, system in NAMESPACE_TO_SYSTEM_URI.items():
-                if system == m.coding.system and ns.value in PREFIX_LOOKUP:
-                    sources.append(PREFIX_LOOKUP[ns.value])
+            ns = SYSTEM_URI_TO_NAMESPACE.get(m.coding.system)
+            if ns in PREFIX_LOOKUP:
+                sources.append(PREFIX_LOOKUP[ns])
 
         for src in sources:
             if src not in sources_meta:

src/gene/schemas.py

@@ -195,6 +195,11 @@ class NamespacePrefix(Enum):
     NamespacePrefix.IUPHAR: "https://www.guidetopharmacology.org/GRAC/ObjectDisplayForward?objectId=",
 }
 
+# URI to source
+SYSTEM_URI_TO_NAMESPACE = {
+    system_uri: ns.value for ns, system_uri in NAMESPACE_TO_SYSTEM_URI.items()
+}
+
 
 class DataLicenseAttributes(BaseModel):
     """Define constraints for data license attributes."""