diff --git a/projects/models.py b/projects/models.py index 88ed65d..d034525 100644 --- a/projects/models.py +++ b/projects/models.py @@ -6,6 +6,7 @@ from common.models.choiceModels import InvitationStatus from django.contrib.postgres.fields import ArrayField from django.conf import settings +from notifications.models import Notification from users.models import User @@ -35,3 +36,4 @@ class ProjectInvitation(BaseModel): to_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='received_invitations') project = models.ForeignKey(Project, on_delete=models.CASCADE) status = models.CharField(max_length=10, choices=InvitationStatus.choices) + notification = models.OneToOneField(Notification, on_delete=models.CASCADE, related_name='project_invitation', null=True, blank=True) \ No newline at end of file