permission.py 947 B

1234567891011121314151617181920212223242526
  1. from flask import redirect, url_for, request
  2. from functools import wraps
  3. from .user_session import UserSession
  4. class Permission:
  5. @staticmethod
  6. def authorize(f):
  7. @wraps(f)
  8. def decorated_function(*args, **kwargs):
  9. current_user = UserSession.get_current_user()
  10. if not current_user.is_authenticated:
  11. return redirect(url_for('auth.login', next=request.url))
  12. return f(*args, **kwargs)
  13. return decorated_function
  14. # @staticmethod
  15. # def edit(f):
  16. # @wraps(f)
  17. # def decorated_function(*args, **kwargs):
  18. # current_user = UserSession.get_current_user()
  19. # if not current_user.is_authenticated:
  20. # return redirect(url_for('auth.login', next=request.url))
  21. # if not current_user.is_admin:
  22. # return redirect(url_for('auth.login', next=request.url))
  23. # return f(*args, **kwargs)