Fix #123 -- init broker configuration before loading other app models

django_dramatiq/apps.py

@@ -35,7 +35,22 @@ class DjangoDramatiqConfig(AppConfig):
     name = "django_dramatiq"
     verbose_name = "Django Dramatiq"
 
+    def import_models(self):
+        """
+        Initialize Dramatiq configuration right after Django has loaded its models.
+
+        Django calls `ready()` on all apps only after it has loaded all models.
+        Due to the fact that tasks often imported from models, we need to make sure
+        that Dramatiq is fully configured before that happens.
+        """
+        super().import_models()
+        self.ready()
+
     def ready(self):
+        if getattr(self, "_is_ready", False):
+            return
+        super().ready()
+
         global RATE_LIMITER_BACKEND
         dramatiq.set_encoder(self.select_encoder())
 
@@ -76,6 +91,8 @@ def ready(self):
         broker = broker_class(middleware=middleware, **broker_options)
         dramatiq.set_broker(broker)
 
+        self._is_ready = True
+
     @property
     def rate_limiter_backend(self):
         return type(self).get_rate_limiter_backend()

tests/testapp1/apps.py

@@ -3,3 +3,6 @@
 
 class Testapp1Config(AppConfig):
     name = "tests.testapp1"
+
+    def ready(self):
+        from . import tasks  # noqa

tests/testapp1/models.py

@@ -0,0 +1,2 @@
+# Quite common use-case when you want to run tasks within models.py code
+from tests.testapp1 import tasks  # noqa