61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from pydantic import BaseModel, Field
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
from models import TaskStatus, RecurrenceType
|
|
|
|
# Reminder Schemas
|
|
class ReminderBase(BaseModel):
|
|
trigger_time: datetime
|
|
recurrence_type: RecurrenceType = RecurrenceType.none
|
|
recurrence_interval: Optional[int] = None
|
|
remind_count_limit: int = 0
|
|
notify_push: bool = False
|
|
notify_telegram: bool = False
|
|
|
|
class ReminderCreate(ReminderBase):
|
|
task_id: int
|
|
|
|
class ReminderUpdate(BaseModel):
|
|
trigger_time: Optional[datetime] = None
|
|
recurrence_type: Optional[RecurrenceType] = None
|
|
recurrence_interval: Optional[int] = None
|
|
remind_count_limit: Optional[int] = None
|
|
remind_count_current: Optional[int] = None
|
|
notify_push: Optional[bool] = None
|
|
notify_telegram: Optional[bool] = None
|
|
|
|
class Reminder(ReminderBase):
|
|
id: int
|
|
task_id: int
|
|
remind_count_current: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Task Schemas
|
|
class TaskBase(BaseModel):
|
|
title: str
|
|
description: Optional[str] = None
|
|
telegram_user_id: str
|
|
|
|
class TaskCreate(TaskBase):
|
|
pass
|
|
|
|
class TaskUpdate(BaseModel):
|
|
title: Optional[str] = None
|
|
description: Optional[str] = None
|
|
status: Optional[TaskStatus] = None
|
|
|
|
class Task(TaskBase):
|
|
id: int
|
|
status: TaskStatus
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
reminders: List[Reminder] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|