update

src/dutylog/application/__main__.py

@@ -8,10 +8,10 @@ from aiogram_dialog import setup_dialogs
 from dishka import make_async_container
 from dishka.integrations.aiogram import setup_dishka
 
-from src.dutylog.application.bot.user_handlers import router as user_router
-from src.dutylog.application.bot.user_dialogs import main_menu_dialog
-from src.dutylog.infrastructure.ioc import ConfigProvider, DatabaseProvider, DAOProvider
-from src.dutylog.infrastructure.utils.config import load_config
+from dutylog.application.bot.user_handlers import router as user_router
+from dutylog.application.bot.user_dialogs import main_menu_dialog
+from dutylog.infrastructure.ioc import ConfigProvider, DatabaseProvider, DAOProvider
+from dutylog.infrastructure.utils.config import load_config
 
 
 async def main():

src/dutylog/application/bot/user_dialogs/__init__.py

@@ -1,3 +1,3 @@
-from src.dutylog.application.bot.user_dialogs.main_menu_dialog import main_menu_dialog
+from dutylog.application.bot.user_dialogs.main_menu_dialog import main_menu_dialog
 
 __all__ = ["main_menu_dialog"]

src/dutylog/application/bot/user_dialogs/main_menu_dialog.py

@@ -1,7 +1,7 @@
 from aiogram_dialog import Dialog, Window
 from aiogram_dialog.widgets.text import Const
 
-from src.dutylog.application.bot.user_dialogs.states import MainMenuSG
+from dutylog.application.bot.user_dialogs.states import MainMenuSG
 
 main_menu_dialog = Dialog(
     Window(

src/dutylog/application/bot/user_handlers.py

@@ -3,7 +3,7 @@ from aiogram.filters import CommandStart
 from aiogram.types import Message
 from aiogram_dialog import DialogManager, StartMode
 
-from src.dutylog.application.bot.user_dialogs.states import MainMenuSG
+from dutylog.application.bot.user_dialogs.states import MainMenuSG
 
 router = Router()
 

src/dutylog/infrastructure/database/dao/hours_transactions_dao.py

@@ -0,0 +1,41 @@
+from sqlalchemy import select, delete
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from dutylog.infrastructure.database.models.hours_transaction import HoursTransaction
+
+
+class HoursTransactionsDAO:
+    def __init__(self, session: AsyncSession):
+        self.session = session
+
+    async def get_by_id(self, transaction_id: int) -> HoursTransaction | None:
+        result = await self.session.execute(
+            select(HoursTransaction).where(HoursTransaction.id == transaction_id)
+        )
+        return result.scalar_one_or_none()
+
+    async def get_all(self) -> list[HoursTransaction]:
+        result = await self.session.execute(
+            select(HoursTransaction).order_by(HoursTransaction.created_at.desc())
+        )
+        return list(result.scalars().all())
+
+    async def get_by_user_id(self, user_id: int) -> list[HoursTransaction]:
+        result = await self.session.execute(
+            select(HoursTransaction)
+            .where(HoursTransaction.user_id == user_id)
+            .order_by(HoursTransaction.created_at.desc())
+        )
+        return list(result.scalars().all())
+
+    async def create(self, transaction: HoursTransaction) -> HoursTransaction:
+        self.session.add(transaction)
+        await self.session.commit()
+        await self.session.refresh(transaction)
+        return transaction
+
+    async def delete(self, transaction_id: int) -> None:
+        await self.session.execute(
+            delete(HoursTransaction).where(HoursTransaction.id == transaction_id)
+        )
+        await self.session.commit()

src/dutylog/infrastructure/database/dao/users_dao.py

@@ -1,21 +1,38 @@
-from sqlalchemy import select
+from sqlalchemy import select, update, delete
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from src.dutylog.infrastructure.database.models.user import User
+from dutylog.infrastructure.database.models.user import User
 
 
 class UsersDAO:
     def __init__(self, session: AsyncSession):
         self.session = session
 
-    async def get_user(self, user_id: int) -> User | None:
+    async def get_by_id(self, user_id: int) -> User | None:
         result = await self.session.execute(
             select(User).where(User.id == user_id)
         )
         return result.scalar_one_or_none()
 
-    async def create_user(self, user_id: int, username: str | None) -> User:
-        user = User(id=user_id, username=username)
+    async def get_all(self) -> list[User]:
+        result = await self.session.execute(select(User))
+        return list(result.scalars().all())
+
+    async def create(self, user: User) -> User:
         self.session.add(user)
         await self.session.commit()
+        await self.session.refresh(user)
         return user
+
+    async def update(self, user_id: int, **kwargs) -> User | None:
+        await self.session.execute(
+            update(User).where(User.id == user_id).values(**kwargs)
+        )
+        await self.session.commit()
+        return await self.get_by_id(user_id)
+
+    async def delete(self, user_id: int) -> None:
+        await self.session.execute(
+            delete(User).where(User.id == user_id)
+        )
+        await self.session.commit()

src/dutylog/infrastructure/database/models/__init__.py

@@ -0,0 +1,5 @@
+from dutylog.infrastructure.database.models.base import Base
+from dutylog.infrastructure.database.models.user import User
+from dutylog.infrastructure.database.models.hours_transaction import HoursTransaction
+
+__all__ = ["Base", "User", "HoursTransaction"]

src/dutylog/infrastructure/database/models/hours_transaction.py

@@ -0,0 +1,24 @@
+from datetime import datetime, timezone, timedelta
+from enum import Enum
+
+from sqlalchemy import BigInteger, Integer, String, DateTime, ForeignKey
+from sqlalchemy.orm import Mapped, mapped_column
+
+from dutylog.infrastructure.database.models.base import Base
+from dutylog.infrastructure.utils.datetime import msk_now
+
+
+class TransactionType(str, Enum):
+    INCREASE = "increase"
+    DECREASE = "decrease"
+
+
+class HoursTransaction(Base):
+    __tablename__ = "hours_transactions"
+
+    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
+    user_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
+    transaction_type: Mapped[str] = mapped_column(String(50), nullable=False)
+    amount: Mapped[int] = mapped_column(Integer, nullable=False)
+    admin_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=msk_now)

src/dutylog/infrastructure/database/models/user.py

@@ -1,7 +1,10 @@
-from sqlalchemy import BigInteger, String
+from datetime import datetime
+
+from sqlalchemy import BigInteger, Boolean, Integer, String, DateTime
 from sqlalchemy.orm import Mapped, mapped_column
 
-from src.dutylog.infrastructure.database.models.base import Base
+from dutylog.infrastructure.database.models.base import Base
+from dutylog.infrastructure.utils.datetime import msk_now
 
 
 class User(Base):
@@ -9,3 +12,10 @@ class User(Base):
 
     id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
     username: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    first_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    last_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    is_admin: Mapped[bool] = mapped_column(Boolean, default=False, server_default="false")
+    active_hours: Mapped[int] = mapped_column(Integer, default=0, server_default="0")
+    inactive_hours: Mapped[int] = mapped_column(Integer, default=0, server_default="0")
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=msk_now)
+    updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=msk_now, onupdate=msk_now)

src/dutylog/infrastructure/ioc.py

@@ -5,6 +5,7 @@ from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker
 
 from dutylog.infrastructure.database.config import create_engine, create_session_maker
 from dutylog.infrastructure.database.dao.users_dao import UsersDAO
+from dutylog.infrastructure.database.dao.hours_transactions_dao import HoursTransactionsDAO
 from dutylog.infrastructure.utils.config import Config, load_config
 
 
@@ -36,3 +37,8 @@ class DAOProvider(Provider):
     def get_users_dao(self, session: AsyncSession) -> UsersDAO:
         return UsersDAO(session)
 
+    @provide(scope=Scope.REQUEST)
+    def get_hours_transactions_dao(self, session: AsyncSession) -> HoursTransactionsDAO:
+        return HoursTransactionsDAO(session)
+
+

src/dutylog/infrastructure/utils/datetime.py

@@ -0,0 +1,8 @@
+from datetime import datetime, timezone, timedelta
+
+
+MSK_TZ = timezone(timedelta(hours=3))
+
+
+def msk_now() -> datetime:
+    return datetime.now(MSK_TZ)