---

from datetime import date
from flask import Blueprint, render_template, request
from flask_login import login_required, current_user
from app.models import Intern, User
from app.services.statistics import intern_stats, center_stats
from app.utils import parse_period

main_bp = Blueprint("main", __name__)

@main_bp.route("/")
@login_required
def dashboard():
    date_from, date_to = parse_period(default_days=30)

    if current_user.is_center:
        mentors = User.query.filter_by(role="mentor").order_by(User.full_name).all()
        mentor_id = request.args.get("mentor_id", type=int)

        # Показываем только НЕ архивных стажёров
        q = Intern.query.filter_by(is_archived=False)
        if mentor_id:
            q = q.filter_by(mentor_id=mentor_id)
        interns = q.order_by(Intern.full_name).all()

        rows = [{"intern": i, "stats": intern_stats(i.id, date_from, date_to)} for i in interns]
        overview = center_stats(date_from, date_to, mentor_id)

        return render_template(
            "dashboard/center.html",
            rows=rows, 
            mentors=mentors, 
            selected_mentor=mentor_id,
            overview=overview, 
            date_from=date_from, 
            date_to=date_to,
            today=date.today()  # Передаем сегодняшнюю дату в шаблон
        )

    # Дашборд для наставника
    interns = Intern.query.filter_by(mentor_id=current_user.id, is_archived=False).order_by(Intern.full_name).all()
    rows = [{"intern": i, "stats": intern_stats(i.id, date_from, date_to)} for i in interns]
    overview = center_stats(date_from, date_to, current_user.id)
    
    return render_template(
        "dashboard/mentor.html", 
        rows=rows, 
        overview=overview, 
        date_from=date_from, 
        date_to=date_to
    )