init

new-planet-backend/app/api/v1/schedules.py

@@ -0,0 +1,141 @@
+from typing import List
+from fastapi import APIRouter, Depends, HTTPException, status, Query
+from sqlalchemy.ext.asyncio import AsyncSession
+from datetime import date
+from app.db.session import get_db
+from app.api.deps import get_current_active_user
+from app.models.user import User
+from app.schemas.schedule import Schedule, ScheduleCreate, ScheduleUpdate
+from app.crud import schedule as crud_schedule
+from app.services.schedule_generator import schedule_generator
+from app.schemas.ai import ScheduleGenerateRequest, ScheduleGenerateResponse
+
+router = APIRouter()
+
+
+@router.get("", response_model=List[Schedule])
+async def get_schedules(
+    skip: int = Query(0, ge=0),
+    limit: int = Query(100, ge=1, le=100),
+    schedule_date: date = Query(None),
+    current_user: User = Depends(get_current_active_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """Получить список расписаний пользователя"""
+    if schedule_date:
+        schedule = await crud_schedule.get_by_date(db, current_user.id, schedule_date)
+        return [schedule] if schedule else []
+    else:
+        schedules = await crud_schedule.get_by_user(db, current_user.id, skip, limit)
+        return schedules
+
+
+@router.get("/{schedule_id}", response_model=Schedule)
+async def get_schedule(
+    schedule_id: str,
+    current_user: User = Depends(get_current_active_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """Получить расписание по ID"""
+    schedule = await crud_schedule.get_with_tasks(db, schedule_id)
+    if not schedule:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="Schedule not found"
+        )
+    if schedule.user_id != current_user.id:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not enough permissions"
+        )
+    return schedule
+
+
+@router.post("", response_model=Schedule, status_code=status.HTTP_201_CREATED)
+async def create_schedule(
+    schedule_in: ScheduleCreate,
+    current_user: User = Depends(get_current_active_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """Создать новое расписание"""
+    schedule_data = schedule_in.model_dump()
+    schedule_data["user_id"] = current_user.id
+    schedule = await crud_schedule.create(db, schedule_data)
+    return schedule
+
+
+@router.put("/{schedule_id}", response_model=Schedule)
+async def update_schedule(
+    schedule_id: str,
+    schedule_in: ScheduleUpdate,
+    current_user: User = Depends(get_current_active_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """Обновить расписание"""
+    schedule = await crud_schedule.get(db, schedule_id)
+    if not schedule:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="Schedule not found"
+        )
+    if schedule.user_id != current_user.id:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not enough permissions"
+        )
+    
+    update_data = schedule_in.model_dump(exclude_unset=True)
+    schedule = await crud_schedule.update(db, schedule, update_data)
+    return schedule
+
+
+@router.delete("/{schedule_id}", status_code=status.HTTP_204_NO_CONTENT)
+async def delete_schedule(
+    schedule_id: str,
+    current_user: User = Depends(get_current_active_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """Удалить расписание"""
+    schedule = await crud_schedule.get(db, schedule_id)
+    if not schedule:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="Schedule not found"
+        )
+    if schedule.user_id != current_user.id:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Not enough permissions"
+        )
+    
+    await crud_schedule.delete(db, schedule_id)
+    return None
+
+
+@router.post("/generate", response_model=ScheduleGenerateResponse)
+async def generate_schedule(
+    request: ScheduleGenerateRequest,
+    current_user: User = Depends(get_current_active_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """Сгенерировать расписание через ИИ"""
+    try:
+        result = await schedule_generator.generate(
+            db=db,
+            user_id=current_user.id,
+            child_age=request.child_age,
+            preferences=request.preferences,
+            schedule_date=request.date,
+            description=request.description
+        )
+        return ScheduleGenerateResponse(
+            schedule_id=result["schedule_id"],
+            title=result["title"],
+            tasks=result["tasks"]
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"Failed to generate schedule: {str(e)}"
+        )
+