refactor queryset

Signed-off-by: tdruez <tdruez@aboutcode.org>

Author: tdruez

aboutcode/notifications/__init__.py

@@ -8,7 +8,12 @@
 
 from aboutcode.notifications.models import AbstractWebhookDelivery
 from aboutcode.notifications.models import AbstractWebhookSubscription
+from aboutcode.notifications.models import WebhookSubscriptionQuerySetMixin
 
 __version__ = "0.1.0"
 
-__all__ = ["AbstractWebhookSubscription", "AbstractWebhookDelivery"]
+__all__ = [
+    "AbstractWebhookSubscription",
+    "AbstractWebhookDelivery",
+    "WebhookSubscriptionQuerySetMixin",
+]

aboutcode/notifications/models.py

@@ -19,6 +19,13 @@
 logger = logging.getLogger(__name__)
 
 
+class WebhookSubscriptionQuerySetMixin:
+    """Mixin for WebhookSubscription querysets. Combine with the project's base QuerySet."""
+
+    def active(self):
+        return self.filter(is_active=True)
+
+
 class AbstractWebhookSubscription(models.Model):
     """
     Abstract base for Webhook subscription models.

notification/models.py

@@ -15,15 +15,15 @@
 
 from aboutcode.notifications import AbstractWebhookDelivery
 from aboutcode.notifications import AbstractWebhookSubscription
+from aboutcode.notifications import WebhookSubscriptionQuerySetMixin
 from dje.models import DataspacedModel
 from dje.models import DataspacedQuerySet
 
 logger = logging.getLogger("dje")
 
 
-class WebhookSubscriptionQuerySet(DataspacedQuerySet):
-    def active(self):
-        return self.filter(is_active=True)
+class WebhookSubscriptionQuerySet(WebhookSubscriptionQuerySetMixin, DataspacedQuerySet):
+    pass
 
 
 class WebhookSubscription(DataspacedModel, AbstractWebhookSubscription):