fix

One-Click-Auth · Apr 3, 2024 · 8433e5c · 8433e5c
1 parent 26a463e
commit 8433e5c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setup(
     name='trustauthx',
-    version='0.6.5',
+    version='0.6.4',
     description='Official connector SDK for TrustAuthx',
     long_description=long_description,
     long_description_content_type='text/markdown',  # This is important!

diff --git a/trustauthx/authlite.py b/trustauthx/authlite.py
@@ -155,8 +155,9 @@ def wrapper(*args, **kwargs):
             response = func(*args, **kwargs)
             # Check for "X-EDGE"
             x_edge = response.headers.get("X-EDGE")
-            if int(x_edge) != _EdgeDBRoleQuery.total_roles:
-                _EdgeDBRoleQuery.reinitialize_all() # Add data
+            if x_edge != None or bool:
+                if int(x_edge) != _EdgeDBRoleQuery.total_roles:
+                    _EdgeDBRoleQuery.reinitialize_all() # Add data
             return response
         return wrapper