197 lines
8.2 KiB
Python
197 lines
8.2 KiB
Python
from functools import wraps
|
|
from flask import session, redirect, url_for, flash
|
|
from flask_login import current_user, login_required
|
|
from sqlalchemy.orm import joinedload
|
|
from .database import get_db_connection, Usuario
|
|
from .rbac import Permission
|
|
|
|
def require_login(f):
|
|
"""Decorador para verificar se o usuário está logado"""
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
if not current_user.is_authenticated:
|
|
flash('Você precisa estar logado para acessar esta página.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
db = get_db_connection()
|
|
try:
|
|
# Carregar o usuário com suas roles
|
|
user = db.query(Usuario).options(
|
|
joinedload(Usuario.roles)
|
|
).get(current_user.id)
|
|
|
|
if not user:
|
|
flash('Usuário não encontrado.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
# Atualiza timestamp da última atividade
|
|
user.update_last_activity()
|
|
db.commit()
|
|
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
db.close()
|
|
return decorated_function
|
|
|
|
def require_permission(permission_name):
|
|
"""Decorador para verificar se o usuário tem uma permissão específica"""
|
|
def decorator(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
if not current_user.is_authenticated:
|
|
flash('Você precisa estar logado para acessar esta página.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
db = get_db_connection()
|
|
try:
|
|
user = db.query(Usuario).get(current_user.id)
|
|
if not user or not user.has_permission(permission_name):
|
|
flash('Você não tem permissão para acessar esta página.', 'error')
|
|
return redirect(url_for('index'))
|
|
|
|
# Atualiza timestamp da última atividade
|
|
user.update_last_activity()
|
|
db.commit()
|
|
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
db.close()
|
|
return decorated_function
|
|
return decorator
|
|
|
|
def require_role(role_name):
|
|
"""Decorador para verificar se o usuário tem um papel específico"""
|
|
def decorator(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
if not current_user.is_authenticated:
|
|
flash('Você precisa estar logado para acessar esta página.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
db = get_db_connection()
|
|
try:
|
|
user = db.query(Usuario).get(current_user.id)
|
|
if not user or not user.has_role(role_name):
|
|
flash('Você não tem permissão para acessar esta página.', 'error')
|
|
return redirect(url_for('index'))
|
|
|
|
# Atualiza timestamp da última atividade
|
|
user.update_last_activity()
|
|
db.commit()
|
|
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
db.close()
|
|
return decorated_function
|
|
return decorator
|
|
|
|
def require_minimum_role(min_level):
|
|
"""Decorador para verificar se o usuário tem um papel com nível mínimo"""
|
|
def decorator(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
if not current_user.is_authenticated:
|
|
flash('Você precisa estar logado para acessar esta página.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
db = get_db_connection()
|
|
try:
|
|
user = db.query(Usuario).get(current_user.id)
|
|
if not user:
|
|
flash('Usuário não encontrado.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
highest_role = user.get_highest_role()
|
|
if not highest_role or highest_role.nivel < min_level:
|
|
flash('Você não tem permissão para acessar esta página.', 'error')
|
|
return redirect(url_for('index'))
|
|
|
|
# Atualiza timestamp da última atividade
|
|
user.update_last_activity()
|
|
db.commit()
|
|
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
db.close()
|
|
return decorated_function
|
|
return decorator
|
|
|
|
def require_instance_permission(permission_name):
|
|
"""Decorator para verificar se o usuário tem permissão em uma instância específica"""
|
|
def decorator(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
if 'user_id' not in session:
|
|
flash('Você precisa estar logado para acessar esta página.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
db = get_db_connection()
|
|
try:
|
|
user = db.query(Usuario).get(session['user_id'])
|
|
if not user:
|
|
flash('Usuário não encontrado.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
# Verificar se o usuário tem a permissão em alguma instância
|
|
if not (user.has_permission(permission_name) or
|
|
user.has_permission(f"{permission_name}_sector") or
|
|
user.has_permission(f"{permission_name}_cr") or
|
|
user.has_permission(f"{permission_name}_cc")):
|
|
flash('Você não tem permissão para acessar esta página.', 'error')
|
|
return redirect(url_for('index'))
|
|
|
|
# Atualiza timestamp da última atividade
|
|
user.update_last_activity()
|
|
db.commit()
|
|
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
db.close()
|
|
return decorated_function
|
|
return decorator
|
|
|
|
def require_instance_access(instance_type, instance_id):
|
|
"""Decorator para verificar se o usuário tem acesso a uma instância específica"""
|
|
def decorator(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
if 'user_id' not in session:
|
|
flash('Você precisa estar logado para acessar esta página.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
db = get_db_connection()
|
|
try:
|
|
user = db.query(Usuario).get(session['user_id'])
|
|
if not user:
|
|
flash('Usuário não encontrado.', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
# Verificar acesso baseado na instância do usuário
|
|
if instance_type == 'celula':
|
|
if not (user.celula_id == instance_id or
|
|
user.has_permission(Permission.VIEW_SECTOR_REPORTS) or
|
|
user.has_permission(Permission.VIEW_CR_REPORTS) or
|
|
user.has_permission(Permission.VIEW_CC_REPORTS)):
|
|
flash('Você não tem acesso a esta célula.', 'error')
|
|
return redirect(url_for('index'))
|
|
elif instance_type == 'setor':
|
|
if not (user.setor_id == instance_id or
|
|
user.has_permission(Permission.VIEW_CR_REPORTS) or
|
|
user.has_permission(Permission.VIEW_CC_REPORTS)):
|
|
flash('Você não tem acesso a este setor.', 'error')
|
|
return redirect(url_for('index'))
|
|
elif instance_type == 'cr':
|
|
if not (user.cr_id == instance_id or
|
|
user.has_permission(Permission.VIEW_CC_REPORTS)):
|
|
flash('Você não tem acesso a este CR.', 'error')
|
|
return redirect(url_for('index'))
|
|
|
|
# Atualiza timestamp da última atividade
|
|
user.update_last_activity()
|
|
db.commit()
|
|
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
db.close()
|
|
return decorated_function
|
|
return decorator |