start work on moving to a package

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

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

@@ -0,0 +1,42 @@
+import dramatiq
+from django.apps import AppConfig
+from django.core.exceptions import ImproperlyConfigured
+from django.utils.module_loading import import_string
+
+from django_dramatiq_postgres.conf import Conf
+
+
+class DjangoDramatiqPostgres(AppConfig):
+    name = "django_dramatiq_postgres"
+    verbose_name = "Django DramatiQ postgres"
+
+    def ready(self):
+        old_broker = dramatiq.get_broker()
+
+        if len(old_broker.actors) != 0:
+            raise ImproperlyConfigured(
+                "Actors were previously registered. "
+                "Make sure your actors are not imported too early."
+            )
+
+        encoder: dramatiq.encoder.Encoder = import_string(Conf.encoder_class)()
+        dramatiq.set_encoder(encoder)
+
+        broker_args = Conf.broker_args
+        broker_kwargs = {
+            **Conf.broker_kwargs,
+            "middleware": [],
+        }
+        broker: dramatiq.broker.Broker = import_string(Conf.broker_class)(
+            *broker_args, **broker_kwargs
+        )
+
+        for middleware_class, middleware_kwargs in Conf.middlewares.items():
+            middleware: dramatiq.middleware.middleware.Middleware = import_string(middleware_class)(
+                **middleware_kwargs
+            )
+            broker.add_middleware(middleware)
+
+        dramatiq.set_broker(broker)
+
+        return super().ready()

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

@@ -0,0 +1,26 @@
+from django.conf import settings
+
+
+class Conf:
+    try:
+        conf = settings.DRAMATIQ.copy()
+    except AttributeError:
+        conf = {}
+
+    encoder_class = 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", {})
+
+    middlewares = conf.get(
+        "middlewares",
+        (
+            ("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", {}),
+        ),
+    )