fix: corrige uso do decorator require_instance_permission - Modifica o decorator para aceitar o nome do parâmetro da instância - Atualiza as rotas de pagamentos para usar o decorator corretamente - Adiciona verificação do ID da instância nos argumentos da função - Melhora mensagens de erro para casos de permissão negada

functions/decorators.py

@@ -10,7 +10,7 @@ def require_login(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')
+            flash('Por favor, faça login para acessar esta página.', 'error')
             return redirect(url_for('login'))
         
         db = get_db_connection()
@@ -39,23 +39,14 @@ def require_permission(permission_name):
         @wraps(f)
         def decorated_function(*args, **kwargs):
             if not current_user.is_authenticated:
-                flash('Você precisa estar logado para acessar esta página.', 'error')
+                flash('Por favor, faça login 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()
+            if not current_user.has_permission(permission_name):
+                flash('Você não tem permissão para acessar esta página.', 'error')
+                return redirect(url_for('home'))
+            
+            return f(*args, **kwargs)
         return decorated_function
     return decorator
 
@@ -116,37 +107,26 @@ def require_minimum_role(min_level):
         return decorated_function
     return decorator
 
-def require_instance_permission(permission_name):
+def require_instance_permission(permission_name, instance_param):
     """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')
+            if not current_user.is_authenticated:
+                flash('Por favor, faça login 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()
+            # Obtém o ID da instância dos argumentos da função
+            instance_id = kwargs.get(instance_param)
+            if instance_id is None:
+                flash('ID da instância não encontrado.', 'error')
+                return redirect(url_for('home'))
+            
+            if not current_user.has_instance_permission(permission_name, instance_id):
+                flash('Você não tem permissão para acessar esta instância.', 'error')
+                return redirect(url_for('home'))
+            
+            return f(*args, **kwargs)
         return decorated_function
     return decorator
 
@@ -155,43 +135,34 @@ def require_instance_access(instance_type, instance_id):
     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')
+            if not current_user.is_authenticated:
+                flash('Por favor, faça login 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()
+            # Verificar acesso baseado na instância do usuário
+            if instance_type == 'celula':
+                if not (current_user.celula_id == instance_id or
+                       current_user.has_permission(Permission.VIEW_SECTOR_REPORTS) or
+                       current_user.has_permission(Permission.VIEW_CR_REPORTS) or
+                       current_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 (current_user.setor_id == instance_id or
+                       current_user.has_permission(Permission.VIEW_CR_REPORTS) or
+                       current_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 (current_user.cr_id == instance_id or
+                       current_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
+            current_user.update_last_activity()
+            db_session.commit()
+            
+            return f(*args, **kwargs)
         return decorated_function
     return decorator