add current site manager

recoco/apps/hitcount/managers.py

@@ -1,4 +1,5 @@
 from django.contrib.contenttypes.models import ContentType
+from django.contrib.sites.managers import CurrentSiteManager
 from django.db import models
 
 
@@ -17,3 +18,17 @@ def for_context_object(self, context_object) -> models.QuerySet:
 
     def for_user(self, user) -> models.QuerySet:
         return self.filter(hits__user=user)
+
+
+class HitCountOnSiteManagerBase(CurrentSiteManager):
+    pass
+
+
+HitCountOnSiteManager = HitCountOnSiteManagerBase.from_queryset(HitCountQuerySet)
+
+
+class HitCountManagerBase(models.Manager):
+    pass
+
+
+HitCountManager = HitCountManagerBase.from_queryset(HitCountQuerySet)

recoco/apps/hitcount/models.py

@@ -6,7 +6,7 @@
 from django.utils.translation import gettext_lazy as _
 from model_utils.models import TimeStampedModel
 
-from .managers import HitCountQuerySet
+from .managers import HitCountManager, HitCountOnSiteManager
 
 
 class HitCount(TimeStampedModel):
@@ -30,7 +30,8 @@ class HitCount(TimeStampedModel):
     context_object_id = models.PositiveIntegerField(null=True, blank=True)
     context_object = GenericForeignKey("context_object_ct", "context_object_id")
 
-    objects = HitCountQuerySet.as_manager()
+    objects = HitCountManager()
+    on_site = HitCountOnSiteManager()
 
     class Meta:
         verbose_name = _("hit count")

recoco/apps/hitcount/tests/test_managers.py

@@ -9,7 +9,7 @@
 
 
 @pytest.mark.django_db
-def test_manager():
+def test_manager_queryset():
     contact = baker.make(Contact)
     project = baker.make(Project)
     resource = baker.make(Task)