diff --git a/src/documents/consumer.py b/src/documents/consumer.py index 4e0fae5cc..1dee610de 100644 --- a/src/documents/consumer.py +++ b/src/documents/consumer.py @@ -102,6 +102,11 @@ class ConsumerStatusShortMessage(str, Enum): class ConsumerPluginMixin: + if TYPE_CHECKING: + from logging import LoggerAdapter + + log: "LoggerAdapter" + def __init__( self, input_doc: ConsumableDocument, diff --git a/src/documents/models.py b/src/documents/models.py index 804e1f155..249de4b4a 100644 --- a/src/documents/models.py +++ b/src/documents/models.py @@ -1,5 +1,6 @@ import datetime from pathlib import Path +from typing import TYPE_CHECKING from typing import Final import pathvalidate @@ -156,6 +157,11 @@ class StoragePath(MatchingModel): class Document(SoftDeleteModel, ModelWithOwner): + if TYPE_CHECKING: + from django.db.models.query import QuerySet + + versions: "QuerySet[Document]" + correspondent = models.ForeignKey( Correspondent, blank=True,