40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from datetime import datetime, time
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.orm import Session
|
|
from typing import List
|
|
from app.core.deps import get_db, get_current_user
|
|
from app.models.homework import HomeworkType
|
|
from app.models.user import User
|
|
from app.schemas.homework import HomeworkTypeCreate, HomeworkTypeResponse
|
|
from app.crud.homework import create_homework_type, get_homework_types_by_user
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("", response_model=HomeworkTypeResponse)
|
|
def register_homework_type(
|
|
homework_data: HomeworkTypeCreate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user)
|
|
):
|
|
homework_type = HomeworkType(
|
|
user_id=current_user.id,
|
|
title=homework_data.title,
|
|
description=homework_data.description,
|
|
reset_type=homework_data.reset_type,
|
|
reset_time=homework_data.reset_time or time(6, 0),
|
|
clear_count=homework_data.clear_count or 0,
|
|
created_at=datetime.utcnow(),
|
|
)
|
|
db.add(homework_type)
|
|
db.commit()
|
|
db.refresh(homework_type)
|
|
return homework_type
|
|
|
|
@router.get("", response_model=List[HomeworkTypeResponse])
|
|
def list_homework_types(
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user)
|
|
):
|
|
return get_homework_types_by_user(current_user.id, db)
|