Migrate pagevote and bookmark to use content_type

judge/models/pagevote.py

@@ -1,6 +1,8 @@
 from django.db import models
 from django.db.models import CASCADE
 from django.utils.translation import gettext_lazy as _
+from django.contrib.contenttypes.fields import GenericForeignKey
+from django.contrib.contenttypes.models import ContentType
 
 from judge.models.profile import Profile
 
@@ -12,12 +14,19 @@ class PageVote(models.Model):
         max_length=30,
         verbose_name=_("associated page"),
         db_index=True,
-    )
+    )  # deprecated
     score = models.IntegerField(verbose_name=_("votes"), default=0)
+    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
+    object_id = models.PositiveIntegerField()
+    linked_object = GenericForeignKey("content_type", "object_id")
 
     class Meta:
         verbose_name = _("pagevote")
         verbose_name_plural = _("pagevotes")
+        indexes = [
+            models.Index(fields=["content_type", "object_id"]),
+        ]
+        unique_together = ("content_type", "object_id")
 
     def vote_score(self, user):
         page_vote = PageVoteVoter.objects.filter(pagevote=self, voter=user)
@@ -27,7 +36,7 @@ class PageVote(models.Model):
             return 0
 
     def __str__(self):
-        return f"pagevote for {self.page}"
+        return f"pagevote for {self.linked_object}"
 
 
 class PageVoteVoter(models.Model):