79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from sqlalchemy import select, insert, update, delete
|
|
from sqlalchemy.exc import NoResultFound
|
|
|
|
from kawai_focus.schemas import TimerModel, TimerListModel
|
|
from kawai_focus.database.session import db
|
|
from kawai_focus.database.models import Timer
|
|
from kawai_focus.utils.errors import ErrorMessage
|
|
from kawai_focus.database.decor_erors import crud_error_guard
|
|
|
|
|
|
@crud_error_guard
|
|
def get_timer(timer_id: int) -> TimerModel:
|
|
"""Функция для получения данных таймера"""
|
|
|
|
with db.get_session() as session:
|
|
query = select(
|
|
Timer.id,
|
|
Timer.title,
|
|
Timer.pomodoro_time,
|
|
Timer.break_time,
|
|
Timer.break_long_time,
|
|
Timer.count_pomodoro
|
|
).where(timer_id == Timer.id)
|
|
result = session.execute(query)
|
|
timer = result.mappings().first()
|
|
|
|
return TimerModel.model_validate(obj=timer, from_attributes=True)
|
|
|
|
|
|
@crud_error_guard
|
|
def list_timers() -> list[TimerListModel]:
|
|
"""Функция для получения списка таймеров"""
|
|
|
|
with db.get_session() as session:
|
|
query = select(Timer.id, Timer.title)
|
|
result = session.execute(query)
|
|
timers = result.mappings().fetchall()
|
|
|
|
return [TimerListModel.model_validate(obj=accept, from_attributes=True) for accept in timers]
|
|
|
|
|
|
@crud_error_guard
|
|
def new_timer(data: TimerModel) -> TimerModel:
|
|
"""Функция для создания нового таймера"""
|
|
|
|
with db.get_session() as session:
|
|
query = insert(Timer).values(**data.model_dump()).returning(Timer)
|
|
result = session.execute(query)
|
|
|
|
new_timer = result.scalar_one()
|
|
session.commit()
|
|
|
|
return TimerModel.model_validate(obj=new_timer, from_attributes=True)
|
|
|
|
|
|
@crud_error_guard
|
|
def update_timer(data: TimerModel) -> TimerModel:
|
|
"""Функция для обновления таймера"""
|
|
|
|
with db.get_session() as session:
|
|
query = update(Timer).values(**data.model_dump()).where(data.id == Timer.id).returning(Timer)
|
|
result = session.execute(query)
|
|
|
|
updated_timer = result.scalar_one()
|
|
session.commit()
|
|
|
|
return TimerModel.model_validate(obj=updated_timer, from_attributes=True)
|
|
|
|
|
|
@crud_error_guard
|
|
def del_timer(timer_id: int) -> None:
|
|
"""Функция для удаления таймера"""
|
|
|
|
with db.get_session() as session:
|
|
query = delete(Timer).where(timer_id == Timer.id)
|
|
session.execute(query)
|
|
|
|
session.commit()
|