Добавление нового типа пользователя + исправления из второй лабы

This commit is contained in:
Anton Kamalov
2024-11-04 21:01:39 +03:00
parent af6860446b
commit b0924ae0bc
10 changed files with 41 additions and 37 deletions

View File

@@ -1,4 +1,4 @@
from flask import redirect, url_for, session, request, current_app
from flask import redirect, url_for, session, request, current_app, render_template
from functools import wraps
@@ -10,18 +10,17 @@ def check_auth(func):
return func(*args, **kwargs)
return wrapper
def group_required(func):
@wraps(func)
def wrapper(*args, **kwargs):
if 'role' in session:
user_role = session.get('role')
user_request = request.endpoint
print('request_endpoint=', user_request)
user_bp = user_request.split('.')[0]
access = current_app.config['db_access']
if user_role in access and user_bp in access[user_role]:
return func(*args, **kwargs)
else:
return redirect(url_for('index'))
user_role = session.get('role')
user_request = request.endpoint
print('request_endpoint=', user_request)
user_bp = user_request.split('.')[0]
access = current_app.config['db_access']
if user_role in access and user_bp in access[user_role]:
return func(*args, **kwargs)
else:
return render_template('error.html', error_message='Недостаточно прав')
return wrapper