feat: added seseion for sqlite, alembic and ruff, model Timerr, schemas.py, mixin for id, cruds: create timer, get timer, settings.py for db

This commit is contained in:
Arduinum628
2025-05-08 10:59:28 +03:00
parent 546c8885ac
commit 310f7c2680
19 changed files with 1001 additions and 33 deletions

View File

@@ -0,0 +1,57 @@
from sqlalchemy import select, insert
from sqlalchemy.exc import SQLAlchemyError, OperationalError, NoResultFound
from pydantic import ValidationError
from kawai_focus.schemas import TimerModel
from kawai_focus.database.session import db
from kawai_focus.database.models import Timer
from kawai_focus.main import Logger
def get_timer(timer_id: int) -> TimerModel:
"""Функция для получения данных таймера"""
try:
with db.get_session() as session:
timer_model = Timer
query = select(
timer_model.id,
timer_model.title,
timer_model.pomodoro_time,
timer_model.break_time,
timer_model.break_long_time,
timer_model.count_pomodoro
).where(timer_id == timer_model.id)
result = session.execute(query)
timer = result.mappings().first()
return TimerModel.model_validate(obj=timer, from_attributes=True)
except (ConnectionError, SQLAlchemyError, TimeoutError, OperationalError, ValidationError, NoResultFound) as err:
Logger.error(f'{err.__class__.__name__}: {err}')
def new_timer(data: TimerModel) -> bool | None:
"""Функция для создания нового таймера"""
try:
with db.get_session() as session:
timer_model = Timer
query = insert(timer_model).values(**data.model_dump())
session.execute(query)
session.commit()
except (ConnectionError, SQLAlchemyError, TimeoutError, OperationalError, ValidationError) as err:
Logger.error(f'{err.__class__.__name__}: {err}')
else:
return True
if __name__ == '__main__':
# new_timer(data=TimerValidModel(
# title='test_timer_2',
# pomodoro_time=50,
# break_long_time=25,
# break_time=5,
# count_pomodoro=4
# ))
print(get_timer(timer_id=1))