diff options
| -rw-r--r-- | ishtar_common/models.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/ishtar_common/models.py b/ishtar_common/models.py index c0ac5004f..2a50533db 100644 --- a/ishtar_common/models.py +++ b/ishtar_common/models.py @@ -2039,14 +2039,16 @@ class TemplateItem: @classmethod def _label_templates_q(cls): model_name = "{}.{}".format( - cls.__module__, cls.__name__).replace( + cls.__module__, cls.__name__) + q = Q(associated_model__klass=model_name, + for_labels=True, available=True) + alt_model_name = model_name.replace( "models_finds", "models").replace( "models_treatments", "models") - return DocumentTemplate.objects.filter( - associated_model__klass=model_name, - for_labels=True, - available=True - ) + if alt_model_name != model_name: + q |= Q(associated_model__klass=model_name, + for_labels=True, available=True) + return DocumentTemplate.objects.filter(q) @classmethod def has_label_templates(cls): |