From 9eb0cb91dbcb95deec693c8b278834ad0b8ee3e6 Mon Sep 17 00:00:00 2001 From: George Katsikas <giorgakis.katsikas@gmail.com> Date: Wed, 17 Jul 2024 16:09:08 +0300 Subject: [PATCH] add typehints to invitation and scipost --- .../submissions/models/referee_invitation.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/scipost_django/submissions/models/referee_invitation.py b/scipost_django/submissions/models/referee_invitation.py index 15a8c7cce..3068c7db9 100644 --- a/scipost_django/submissions/models/referee_invitation.py +++ b/scipost_django/submissions/models/referee_invitation.py @@ -3,6 +3,7 @@ __license__ = "AGPL v3" import datetime +from typing import TYPE_CHECKING from django.db import models from django.urls import reverse @@ -14,6 +15,11 @@ from ..behaviors import SubmissionRelatedObjectMixin from ..managers import RefereeInvitationQuerySet from ..models import EditorialAssignment +if TYPE_CHECKING: + from profiles.models import Profile + from scipost.models import Contributor + from submissions.models import Submission + class RefereeInvitation(SubmissionRelatedObjectMixin, models.Model): """Invitation to an active professional scientist to referee a Submission. @@ -25,15 +31,15 @@ class RefereeInvitation(SubmissionRelatedObjectMixin, models.Model): """ - profile = models.ForeignKey( + profile = models.ForeignKey["Profile"]( "profiles.Profile", on_delete=models.SET_NULL, blank=True, null=True ) - submission = models.ForeignKey( + submission = models.ForeignKey["Submission"]( "submissions.Submission", on_delete=models.CASCADE, related_name="referee_invitations", ) - referee = models.ForeignKey( + referee = models.ForeignKey["Contributor"]( "scipost.Contributor", related_name="referee_invitations", blank=True, -- GitLab