from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from datetime import datetime
from ..db import get_connection

router = APIRouter()
from datetime import datetime

# 在文件末尾添加新接口
@router.get("/active_announcements")
def get_active_announcements():
    conn = get_connection()
    try:
        cursor = conn.cursor(dictionary=True)
        cursor.execute("""
            SELECT text, color 
            FROM announcements 
            WHERE start_time <= %s AND end_time >= %s
            ORDER BY created_at DESC
        """, (datetime.now(), datetime.now()))
        return cursor.fetchall()
    finally:
        cursor.close()
        conn.close()