Re-factor name retrieval

levitsky · Oct 10, 2023 · 51ffa01 · 51ffa01
1 parent 1bc5b35
commit 51ffa01
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pyteomics/proforma.py b/pyteomics/proforma.py
@@ -361,7 +361,7 @@ def __init__(self, **kwargs):
 
     def load_database(self):
         if _has_psims:
-            return load_unimod()
+            return obo_cache.load("http://www.unimod.org/obo/unimod.obo")
         return Unimod()
 
     def resolve(self, name=None, id=None, **kwargs):
@@ -397,9 +397,12 @@ def resolve(self, name=None, id=None, **kwargs):
                 'provider': self.name
             }
         else:
+            name = defn.ex_code_name
+            if not name:
+                name = defn.code_name
             return {
                 "composition": defn.composition,
-                "name": defn.ex_code_name,
+                "name": name,
                 "id": defn.id,
                 "mass": defn.monoisotopic_mass,
                 "provider": self.name