views.py

from flask import Flask, render_template, request, redirect, url_for, session
from sqlalchemy.orm import sessionmaker
from datetime import datetime
from services import criptografar_senha,aluno_required,admin_required,analisar_sentimento
from flask_bootstrap import Bootstrap
from sqlalchemy.orm import sessionmaker
from sqlalchemy import MetaData
from sqlalchemy.ext.automap import automap_base
from conexao import ConexaoSingleton

app = Flask(__name__)
Bootstrap(app)
app.secret_key = '#&aARfvJKg!sgv'

conexao_instance = ConexaoSingleton()
engine = conexao_instance.get_engine()

metadata = MetaData()
metadata.reflect(engine)

Base = automap_base(metadata=metadata)
Base.prepare()

Aluno = Base.classes.aluno
DiarioBordo = Base.classes.diariobordo

Session = sessionmaker(bind=engine)

@app.route("/verificar_ra", methods=["POST"])
def verificar_ra():
    ra = request.form['ra']
    session_db = Session()
    try:
        aluno = session_db.query(Aluno).filter(Aluno.ra == ra).one_or_none()
        if aluno:
            return {"valido": True}
        else:
            return {"valido": False}
    except Exception as e:
        session_db.rollback()
        return {"valido": False}, 500
    finally:
        session_db.close()

@app.route("/", methods=["GET", "POST"])
def index():
    if request.method == "POST":
        ra = request.form['ra']
        senha = request.form['senha']
        session_db = Session()
        
        senha_criptografada = criptografar_senha(senha)

        try:
            aluno = session_db.query(Aluno).filter(Aluno.ra == ra, Aluno.senha == senha_criptografada).one_or_none()
            if aluno:
                ra_str = aluno.ra
                session['user_id'] = str(ra_str)
                return redirect(url_for('detalhe_aluno', ra=aluno.ra))
            else:
                mensagem = "ID ou senha inválidos!"
                return render_template("home.html", mensagem=mensagem, active_page='index')
        except Exception as e:
            session_db.rollback()
            return render_template("home.html", mensagem="Erro ao tentar fazer login.", active_page='index')
        finally:
            session_db.close()
    
    return render_template("home.html", mensagem="", active_page='index')

@app.route("/cadastro")
def cadastro():
    return render_template("cadastro.html", active_page='cadastro')

@app.route("/quemsomos")
def quem_somos():
    return render_template("quemsomos.html", active_page='quemsomos')

@app.route('/novoaluno', methods=['POST'])
def inserir_aluno():
    session_db = Session()
    ra = request.form['ra']
    nome = request.form['nome']
    treino_semanal = request.form['treino_semanal']
    senha = request.form['senha'] 

    senha_criptografada = criptografar_senha(senha)

    aluno = Aluno(ra=ra, nome=nome, treino_semanal=treino_semanal, senha=senha_criptografada)

    try:
        session_db.add(aluno)
        session_db.commit()
    except:
        session_db.rollback()
    finally:
        session_db.close()

    return redirect(url_for('index'))

@app.route('/alunos', methods=['GET'])
@admin_required
def listar_alunos():
    session_db = Session()
    page = request.args.get('page', 1, type=int)
    per_page = 10
    search_query = request.args.get('search', '')

    try:
        alunos_query = session_db.query(Aluno).filter(
            (Aluno.nome.like(f'%{search_query}%')) | (Aluno.ra.like(f'%{search_query}%'))
        ).order_by(Aluno.id)

        alunos_paginated = alunos_query.offset((page - 1) * per_page).limit(per_page).all()
        total_alunos = alunos_query.count()
        total_pages = (total_alunos + per_page - 1) // per_page 
    except:
        session_db.rollback()
        msg = "Erro ao tentar recuperar a lista de Atletas"
        return render_template('index.html', msgbanco=msg, active_page='listar_alunos')
    finally:
        session_db.close()

    return render_template('listaralunos.html', alunos=alunos_paginated, page=page, total_pages=total_pages, search=search_query, active_page='listar_alunos')

@app.route('/excluir_aluno/<int:ra>', methods=['POST'])
@admin_required
def excluir_aluno(ra):
    session_db = Session()
    aluno = session_db.query(Aluno).filter_by(ra=ra).first()
    if aluno:
        try:
            session_db.delete(aluno)
            session_db.commit()
        except:
            session_db.rollback()
        finally:
            session_db.close()
    return redirect(url_for('listar_alunos'))

@app.route('/atualizar_aluno/<int:ra>', methods=['GET', 'POST'])
@admin_required
def atualizar_aluno(ra):
    session_db = Session() 
    aluno = session_db.query(Aluno).filter_by(ra=ra).first()
    if request.method == 'POST':
        aluno.nome = request.form['nome']

        try:
            session_db.commit()
        except:
            session_db.rollback()
        finally:
            session_db.close()
        return redirect(url_for('listar_alunos'))

    return render_template('atualizaraluno.html', aluno=aluno, active_page='listar_alunos')

@app.route("/admin", methods=["GET", "POST"])
def admin_login():
    if request.method == "POST":
        ADMIN_USERNAME = 'admin'
        ADMIN_PASSWORD = '123'

        username = request.form.get('username')
        password = request.form.get('password')

        if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
            session['user_id'] = username 
            return redirect(url_for('listar_alunos'))
        else:
            mensagem = "Credenciais inválidas!"
            return render_template("admin.html", mensagem=mensagem, active_page='admin')

    return render_template("admin.html", mensagem="", active_page='admin')

@app.route('/adicionar_diario/<string:ra>', methods=['POST'])
@aluno_required
def adicionar_diario(ra):
    session_db = Session()
    try:
        texto_diario = request.form['texto']
        duracao = float(request.form['duracao'])  # Convertendo para float
        distancia = float(request.form['distancia'])  # Convertendo para float
        sentimento = analisar_sentimento(texto_diario)
        
        # Calcular a velocidade
        if duracao > 0:
            velocidade = distancia / duracao  # Velocidade em km/min, por exemplo
        else:
            velocidade = 0  # Evitar divisão por zero

        novo_diario = DiarioBordo(
            texto=texto_diario,
            duracao_treino=duracao,
            distancia_treino=distancia,
            velocidade_treino=velocidade,  # Se o campo existir
            datahora=datetime.now(),
            fk_Aluno_id=session_db.query(Aluno).filter(Aluno.ra == ra).one().id,
            sentimento=sentimento
        )
        session_db.add(novo_diario)
        session_db.commit()
        return redirect(url_for('detalhe_aluno', ra=ra))
    except Exception as e:
        session_db.rollback()
        print(f"Erro ao adicionar diário: {e}")
        return "Erro ao adicionar diário", 500
    finally:
        session_db.close()


@app.route('/aluno/<string:ra>', methods=['GET'])
@aluno_required
def detalhe_aluno(ra):
    session_db = Session() 
    try:
        aluno = session_db.query(Aluno).filter(Aluno.ra == ra).one_or_none()
        if aluno is None:
            return "Atleta não encontrado", 404
        
        diariobordo = session_db.query(DiarioBordo).filter(DiarioBordo.fk_Aluno_id == aluno.id).order_by(DiarioBordo.id.desc()).all()
        
        diariobordo_list = [{
            'datahora': diario.datahora,
            'texto': diario.texto,
            'sentimento': diario.sentimento,
            'duracao_treino': diario.duracao_treino,
            'distancia_treino': diario.distancia_treino,
            'velocidade_treino': diario.velocidade_treino
        } for diario in diariobordo]
        
    except Exception as e:
        session_db.rollback()
        print(f"Erro ao buscar o atleta: {e}")
        return "Erro ao buscar o atleta", 500
    finally:
        session_db.close()

    return render_template('detalhealuno.html', aluno=aluno, diariobordo=diariobordo_list, active_page='listar_alunos')

@app.route('/logout')
def logout():
    session.pop('user_id', None)
    return redirect(url_for('index'))

if __name__ == "__main__":
    app.run(debug=True)