revert source priority (do in new issue)

cancervariants · Sep 27, 2023 · 894d319 · 894d319
1 parent 11bbc56
commit 894d319
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/disease/query.py b/disease/query.py
@@ -15,7 +15,7 @@
     RefType,
     SearchService,
     ServiceMeta,
-    SourcePriority,
+    SourceName,
 )
 
 from .version import __version__
@@ -371,11 +371,19 @@ def _record_order(self, record: Dict) -> Tuple[int, str]:
         :return: tuple with rank value and concept ID
         """
         src = record["src_name"]
-        try:
-            source_rank = SourcePriority[src].value
-        except KeyError:
+        if src == SourceName.NCIT.value:
+            source_rank = 1
+        elif src == SourceName.MONDO.value:
+            source_rank = 2
+        elif src == SourceName.ONCOTREE.value:
+            source_rank = 3
+        elif src == SourceName.OMIM.value:
+            source_rank = 4
+        elif src == SourceName.DO.value:
+            source_rank = 5
+        else:
             logger.warning(f"query.record_order: Invalid source name for " f"{record}")
-            source_rank = len(SourcePriority) + 1
+            source_rank = 4
         return source_rank, record["concept_id"]
 
     def _handle_failed_merge_ref(self, record, response, query) -> NormalizationService: