update get_or_create_cid_user_exam

2023-07-03 12:51:37 +01:00
parent f458dc1306
commit 15b6233802
1 changed files with 9 additions and 2 deletions
@@ -452,10 +452,17 @@ class ExamCollectionGenericBase(models.Model):
        return False

    def get_or_create_cid_user_exam(
-        self, cid_user=None, user_user=None, start_time=None
+        self,
+        cid: int | None = None,
+        cid_user: "CidUser | None" = None,
+        user_user: User | None = None,
+        start_time=None,
    ):
        content_type = ContentType.objects.get_for_model(self)

+        if cid is not None:
+            cid_user = CidUser.objects.filter(cid=cid).first()
+
        if cid_user is not None:
            c = CidUserExam.objects.filter(
                content_type=content_type, object_id=self.pk, cid_user=cid_user