from flask import Blueprint, render_template, jsonify, redirect, url_for, session, request, flash
from flask_login import login_required, current_user
from app.models.reservation import Reservation
from app.models.classroom import Classroom
from app.models.subject import Subject
from app.models.subject import Commission
from datetime import datetime, timedelta
from sqlalchemy import func
from flask_babel import gettext as _

main_bp = Blueprint('main', __name__)

@main_bp.route('/')
@login_required
def dashboard():
    """Main dashboard page"""
# Get statistics
    try:
        total_classrooms = Classroom.query.filter_by(is_active=True).count()
    except:
        total_classrooms = 0
        
    try:
        total_subjects = Subject.query.filter_by(is_active=True).count()
    except:
        total_subjects = 0
    
    # Get today's reservations
    today = datetime.utcnow().date()
    try:
        today_reservations = Reservation.query.filter(
            func.date(Reservation.start_time) == today
        ).count()
    except:
        today_reservations = 0
    
    # Get this week's reservations
    week_start = today - timedelta(days=today.weekday())
    week_end = week_start + timedelta(days=6)
    try:
        week_reservations = Reservation.query.filter(
            func.date(Reservation.start_time) >= week_start,
            func.date(Reservation.start_time) <= week_end
        ).count()
    except:
        week_reservations = 0
    
    # Get upcoming reservations (next 7 days)
    try:
        upcoming_reservations = Reservation.query.filter(
            Reservation.start_time >= datetime.utcnow(),
            Reservation.start_time <= datetime.utcnow() + timedelta(days=7)
        ).limit(10).all()
    except:
        upcoming_reservations = []
    
    stats = {
        'total_classrooms': total_classrooms,
        'total_subjects': total_subjects,
        'today_reservations': today_reservations,
        'week_reservations': week_reservations
    }
    
    return render_template('dashboard.html', 
                         stats=stats, 
                         upcoming_reservations=upcoming_reservations)

@main_bp.route('/api/dashboard-stats')
@login_required
def dashboard_stats_api():
    """API endpoint for dashboard statistics"""
    # Get today's data
    today = datetime.utcnow().date()
    today_reservations = Reservation.query.filter(
        func.date(Reservation.start_time) == today
    ).count()
    
    # Get this week's data
    week_start = today - timedelta(days=today.weekday())
    week_end = week_start + timedelta(days=6)
    week_reservations = Reservation.query.filter(
        func.date(Reservation.start_time) >= week_start,
        func.date(Reservation.start_time) <= week_end
    ).count()
    
    # Get monthly data for chart
    current_date = datetime.utcnow()
    monthly_data = []
    
    for i in range(12):
        month_date = current_date - timedelta(days=30 * i)
        month_start = month_date.replace(day=1)
        month_end = (month_start + timedelta(days=32)).replace(day=1) - timedelta(days=1)
        
        monthly_reservations = Reservation.query.filter(
            func.date(Reservation.start_time) >= month_start.date(),
            func.date(Reservation.start_time) <= month_end.date()
        ).count()
        
        monthly_data.append({
            'month': month_start.strftime('%Y-%m'),
            'reservations': monthly_reservations
        })
    
    return jsonify({
        'today_reservations': today_reservations,
        'week_reservations': week_reservations,
        'monthly_data': list(reversed(monthly_data))
    })

@main_bp.route('/genetic-optimizer')
@login_required
def genetic_optimizer():
    """Serve the genetic algorithm optimization interface"""
    if current_user.role not in ['ADMIN']:
        flash('You do not have permission to access the optimization tool.', 'danger')
        return redirect(url_for('main.dashboard'))
    
    return render_template('genetic_optimizer.html')

@main_bp.route('/set_language/<language>')
def set_language(language=None):
    if language not in ['en', 'es']:
        language = 'en'
    
    session['language'] = language
    return redirect(request.referrer or url_for('main.dashboard'))