Update utils.py

sondregronas · Aug 6, 2023 · 06d8765 · 06d8765
1 parent 8d56fc0
commit 06d8765
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/BookingSystem/utils.py b/BookingSystem/utils.py
@@ -44,9 +44,11 @@ def wrapper(*args, **kwargs) -> callable:
                 flask.abort(403)
             return func(*args, **kwargs)
 
+        # Add attributes to the wrapper function so we can check if the decorator is used on a function. (tests)
         wrapper.login_required = True
         wrapper.admin_required = admin_only
         wrapper.api_allowed = api
+
         return wrapper
 
     return decorator