wip

epdb/models.py

@@ -867,18 +867,25 @@ class Compound(
 
         standardized_smiles = FormatConverter.standardize(smiles, remove_stereo=True)
 
+        subclasses = CompoundStructure.__subclasses__()
+
+        qs = CompoundStructure.objects.filter(smiles=smiles, compound__package=package)
+        if subclasses:
+            qs = qs.not_instance_of(*subclasses)
+
         # Check if we find a direct match for a given SMILES
-        if CompoundStructure.objects.filter(smiles=smiles, compound__package=package).exists():
-            return CompoundStructure.objects.get(smiles=smiles, compound__package=package).compound
+        if qs.exists():
+            return qs.first().compound
+
+
+        qs = CompoundStructure.objects.filter(smiles=standardized_smiles, compound__package=package)
+        if subclasses:
+            qs = qs.not_instance_of(*subclasses)
 
         # Check if we can find the standardized one
-        if CompoundStructure.objects.filter(
-            smiles=standardized_smiles, compound__package=package
-        ).exists():
+        if qs.exists():
             # TODO should we add a structure?
-            return CompoundStructure.objects.get(
-                smiles=standardized_smiles, compound__package=package
-            ).compound
+            return qs.first().compound
 
         # Generate Compound
         c = Compound()