Don't use singleton CredentialStore in requires_credentials decorator

Fix for jamesremuscat#10
epenet · Jun 2, 2020 · 427d898 · 427d898
1 parent e6a4f0f
commit 427d898
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/pyze/api/credentials.py b/src/pyze/api/credentials.py
@@ -20,8 +20,13 @@ class MissingCredentialException(Exception):
 def requires_credentials(*names):
     def _requires_credentials(func):
         def inner(*args, **kwargs):
+            credentials = None
+            if args[0] and hasattr(args[0], '_credentials'):
+                credentials = args[0]._credentials
+            elif args[0] and hasattr(args[0], '_kamereon'):
+                credentials = args[0]._kamereon._credentials
             for name in names:
-                if name not in CredentialStore():
+                if name not in credentials:
                     raise MissingCredentialException(name)
             return func(*args, **kwargs)