92 lines
3.2 KiB
Python
92 lines
3.2 KiB
Python
from flask import Blueprint, render_template, jsonify
|
|
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 datetime import datetime, timedelta
|
|
from sqlalchemy import func
|
|
|
|
main_bp = Blueprint('main', __name__)
|
|
|
|
@main_bp.route('/')
|
|
@login_required
|
|
def dashboard():
|
|
"""Main dashboard page"""
|
|
# Get statistics
|
|
total_classrooms = Classroom.query.filter_by(is_active=True).count()
|
|
total_subjects = Subject.query.filter_by(is_active=True).count()
|
|
|
|
# Get today's reservations
|
|
today = datetime.utcnow().date()
|
|
today_reservations = Reservation.query.filter(
|
|
func.date(Reservation.start_time) == today
|
|
).count()
|
|
|
|
# Get this week's reservations
|
|
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 upcoming reservations (next 7 days)
|
|
upcoming_reservations = Reservation.query.filter(
|
|
Reservation.start_time >= datetime.utcnow(),
|
|
Reservation.start_time <= datetime.utcnow() + timedelta(days=7),
|
|
Reservation.status.in_(['CONFIRMED', 'PENDING'])
|
|
).join(Classroom).join(Subject).limit(10).all()
|
|
|
|
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))
|
|
}) |