fix: initialize with db_alias (#24)

pycasbin · Aug 2, 2023 · e3f283f · e3f283f
1 parent 9f97327
commit e3f283f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/casbin_adapter/apps.py b/casbin_adapter/apps.py
@@ -1,4 +1,5 @@
 from django.apps import AppConfig
+from django.conf import settings
 
 
 class CasbinAdapterConfig(AppConfig):
@@ -7,4 +8,5 @@ class CasbinAdapterConfig(AppConfig):
     def ready(self):
         from .enforcer import initialize_enforcer
 
-        initialize_enforcer()
+        db_alias = getattr(settings, "CASBIN_DB_ALIAS", "default")
+        initialize_enforcer(db_alias)
diff --git a/casbin_adapter/enforcer.py b/casbin_adapter/enforcer.py
@@ -1,6 +1,6 @@
 import logging
 from django.conf import settings
-from django.db import connection
+from django.db import connection, connections
 from django.db.utils import OperationalError, ProgrammingError
 
 from casbin import Enforcer
@@ -64,7 +64,7 @@ def initialize_enforcer(db_alias=None):
     try:
         row = None
         if db_alias:
-            with connection[db_alias].cursor() as cursor:
+            with connections[db_alias].cursor() as cursor:
                 cursor.execute(
                     """
                     SELECT app, name applied FROM django_migrations