Merge pull request #950 from procrastinate-org/django-fix

procrastinate-org · Feb 28, 2024 · 2ba511c · 2ba511c
2 parents 201dd4b + 36da94d
commit 2ba511c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/procrastinate/contrib/django/django_connector.py b/procrastinate/contrib/django/django_connector.py
@@ -9,8 +9,7 @@
 from django.db.backends.base.base import BaseDatabaseWrapper
 from typing_extensions import LiteralString
 
-from procrastinate import connector, psycopg_connector
-from procrastinate.contrib.aiopg import aiopg_connector
+from procrastinate import connector
 from procrastinate.contrib.django import utils
 
 if TYPE_CHECKING:
@@ -129,10 +128,14 @@ def get_worker_connector(self) -> connector.BaseAsyncConnector:
         alias = utils.get_setting("DATABASE_ALIAS", default="default")
 
         if utils.package_is_installed("psycopg3"):
+            from procrastinate import psycopg_connector
+
             return psycopg_connector.PsycopgConnector(
                 kwargs=utils.connector_params(alias)
             )
         if utils.package_is_installed("aiopg"):
+            from procrastinate.contrib.aiopg import aiopg_connector
+
             return aiopg_connector.AiopgConnector(**utils.connector_params(alias))
 
         raise django_exceptions.ImproperlyConfigured(