pre-commit changes
This commit is contained in:
@@ -1,19 +1,22 @@
|
||||
from flask import redirect, url_for, session, request, current_app, render_template
|
||||
from functools import wraps
|
||||
|
||||
from flask import (current_app, redirect, render_template, request, session,
|
||||
url_for)
|
||||
|
||||
|
||||
def check_auth(func):
|
||||
@wraps(func)
|
||||
def wrapper(*args, **kwargs):
|
||||
if 'login' not in session:
|
||||
return redirect(url_for('auth_bp.auth'))
|
||||
user_role = session.get('role')
|
||||
if "login" not in session:
|
||||
return redirect(url_for("auth_bp.auth"))
|
||||
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']
|
||||
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
|
||||
return render_template("error.html", error_message="Недостаточно прав")
|
||||
|
||||
return wrapper
|
||||
|
||||
Reference in New Issue
Block a user