61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from pydantic import BaseModel, field_validator, Field
|
|
from kawai_focus_v2.core.messages.errors import ErrorMessage
|
|
|
|
|
|
class TimerModel(BaseModel):
|
|
"""Модель схемы данных таймера"""
|
|
|
|
id: int
|
|
title: str
|
|
pomodoro_time: int
|
|
break_time: int
|
|
break_long_time: int
|
|
count_pomodoro: int
|
|
|
|
|
|
class NewTimerModel(BaseModel):
|
|
"""Модель схемы нового таймера"""
|
|
|
|
title: str
|
|
pomodoro_time: int
|
|
break_time: int
|
|
break_long_time: int
|
|
count_pomodoro: int
|
|
|
|
|
|
class UpdateTimerModel(BaseModel):
|
|
"""Модель схемы обновления данных таймера"""
|
|
|
|
title: str | None = None
|
|
pomodoro_time: int | None = None
|
|
break_time: int | None = None
|
|
break_long_time: int | None = None
|
|
count_pomodoro: int | None = None
|
|
|
|
|
|
class TimerListModel(BaseModel):
|
|
"""Модель схемы данных таймера для списка"""
|
|
|
|
id: int
|
|
title: str
|
|
pomodoro_time: int
|
|
count_pomodoro: int
|
|
|
|
|
|
class TimerTimeModel(BaseModel):
|
|
"""Модель схемы данных времени таймера"""
|
|
|
|
hh: int = Field(0, ge=0, le=23)
|
|
mm: int = Field(0, ge=0, le=59)
|
|
|
|
@field_validator('hh', 'mm')
|
|
@classmethod
|
|
def check_all_time_fields(cls, value: int) -> int:
|
|
"""Метод валидирует все поля времени"""
|
|
|
|
# гарантирует, что время не равно 00:00:00
|
|
if value == 0:
|
|
raise ValueError(ErrorMessage.NO_TIME.value)
|
|
|
|
return value
|