move all broker stuff to package, schedule is still todo

Signed-off-by: Marc 'risson' Schmitt <marc.schmitt@risson.space>

packages/django-dramatiq-postgres/django_dramatiq_postgres/conf.py

@@ -1,33 +1,57 @@
+from typing import Any
+
 from django.conf import settings
+from django.core.exceptions import ImproperlyConfigured
 
 
 class Conf:
-    try:
-        conf = settings.DRAMATIQ.copy()
-    except AttributeError:
-        conf = {}
+    def __init__(self):
+        try:
+            self.conf = settings.DRAMATIQ
+        except AttributeError as exc:
+            raise ImproperlyConfigured("Setting DRAMATIQ not set.") from exc
+        if "task_class" not in self.conf:
+            raise ImproperlyConfigured("DRAMATIQ.task_class not defined")
 
-    encoder_class = conf.get("encoder_class", "dramatiq.encoder.PickleEncoder")
+    @property
+    def encoder_class(self) -> str:
+        return self.conf.get("encoder_class", "dramatiq.encoder.PickleEncoder")
 
-    broker_class = conf.get("broker_class", "django_dramatiq_postgres.broker.PostgresBroker")
-    broker_args = conf.get("broker_args", ())
-    broker_kwargs = conf.get("broker_kwargs", {})
+    @property
+    def broker_class(self) -> str:
+        return self.conf.get("broker_class", "django_dramatiq_postgres.broker.PostgresBroker")
 
-    middlewares = conf.get(
-        "middlewares",
-        (
-            ("django_dramatiq_postgres.middleware.DbConnectionMiddleware", {}),
-            ("dramatiq.middleware.age_limit.AgeLimit", {}),
-            ("dramatiq.middleware.time_limit.TimeLimit", {}),
-            ("dramatiq.middleware.shutdown.ShutdownNotifications", {}),
-            ("dramatiq.middleware.callbacks.Callbacks", {}),
-            ("dramatiq.middleware.pipelines.Pipelines", {}),
-            ("dramatiq.middleware.retries.Retries", {}),
-        ),
-    )
+    @property
+    def broker_args(self) -> tuple[Any]:
+        return self.conf.get("broker_args", ())
 
-    channel_prefix = conf.get("channel_prefix", "dramatiq.tasks")
+    @property
+    def broker_kwargs(self) -> dict[str, Any]:
+        return self.conf.get("broker_kwargs", {})
 
-    task_class = conf.get("task_class", None)
+    @property
+    def middlewares(self) -> tuple[tuple[str, dict[str, Any]]]:
+        return self.conf.get(
+            "middlewares",
+            (
+                ("django_dramatiq_postgres.middleware.DbConnectionMiddleware", {}),
+                ("dramatiq.middleware.age_limit.AgeLimit", {}),
+                ("dramatiq.middleware.time_limit.TimeLimit", {}),
+                ("dramatiq.middleware.shutdown.ShutdownNotifications", {}),
+                ("dramatiq.middleware.callbacks.Callbacks", {}),
+                ("dramatiq.middleware.pipelines.Pipelines", {}),
+                ("dramatiq.middleware.retries.Retries", {}),
+            ),
+        )
 
-    test = conf.get("test", False)
+    @property
+    def channel_prefix(self) -> str:
+        return self.conf.get("channel_prefix", "dramatiq.tasks")
+
+    @property
+    def task_class(self) -> str:
+        return self.conf["task_class"]
+
+    @property
+    def test(self) -> bool:
+        return self.conf.get("test", False)