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