Add Auth Decorator (mitre#1001)

ZhangTongLe · Jan 7, 2020 · 8b5c9a0 · 8b5c9a0
1 parent 1576ca0
commit 8b5c9a0
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/app/service/auth_svc.py b/app/service/auth_svc.py
@@ -13,6 +13,17 @@
 from app.utility.base_service import BaseService
 
 
+def check_authorization(func):
+    async def process(func, *args, **params):
+        return await func(*args, **params)
+
+    async def helper(*args, **params):
+        await args[0].auth_svc.check_permissions(args[1])
+        result = await process(func, *args, **params)
+        return result
+    return helper
+
+
 class AuthService(BaseService):
 
     User = namedtuple('User', ['username', 'password', 'permissions'])