Skip to content

Commit 22be8b4

Browse files
committed
Modify the __init__ method, add init_app method.
1 parent d8dc7d0 commit 22be8b4

1 file changed

Lines changed: 12 additions & 5 deletions

File tree

flask_authz/casbin_enforcer.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,18 +16,25 @@ class CasbinEnforcer:
1616

1717
e = None
1818

19-
def __init__(self, app, adapter, watcher=None):
19+
def __init__(self, app=None, adapter=None, watcher=None):
2020
"""
2121
Args:
2222
app (object): Flask App object to get Casbin Model
2323
adapter (object): Casbin Adapter
2424
"""
25-
self.app = app
25+
self.app = None
2626
self.adapter = adapter
27-
self.e = casbin.Enforcer(app.config.get("CASBIN_MODEL"), self.adapter)
28-
if watcher:
29-
self.e.set_watcher(watcher)
27+
self.watcher = watcher
3028
self._owner_loader = None
29+
self.user_name_headers = None
30+
if app:
31+
self.init_app(app)
32+
33+
def init_app(self, app):
34+
self.app = app
35+
self.e = casbin.Enforcer(app.config.get("CASBIN_MODEL"), self.adapter)
36+
if self.watcher:
37+
self.e.set_watcher(self.watcher)
3138
self.user_name_headers = app.config.get("CASBIN_USER_NAME_HEADERS", None)
3239

3340
def set_watcher(self, watcher):

0 commit comments

Comments
 (0)