upd

Author: ProFastCode

src/application/services/user_service.py

@@ -0,0 +1,12 @@
+from domain.user.entities.user import User
+from infrastructure.database.repositories.user_repo_impl import UserRepoImpl
+
+class UserService:
+    def __init__(self, user_repo: UserRepoImpl):
+        self.user_repo = user_repo
+
+    async def get_user_by_id(self, user_id: int) -> User:
+        return await self.user_repo.get_by_id(user_id)
+
+    async def create_user(self, user: User) -> None:
+        await self.user_repo.save(user)

src/domain/auth/exceptions.py src/domain/auth/exceptions/token.pysrc/domain/auth/exceptions.py renamed to src/domain/auth/exceptions/token.py

@@ -5,7 +5,7 @@
 import jwt
 
 from domain.base import BaseValueObject
-from domain.auth.exceptions import TokenInvalidError, TokenExpiredError
+from domain.auth.exceptions.token import TokenInvalidError, TokenExpiredError
 
 JWT_SECRET = "your_secret_key"  # Замените на секретный ключ из конфигурации
 

src/domain/auth/value_objects/token.py

@@ -3,33 +3,6 @@
 from domain.base import DomainError
 
 
-@dataclass(eq=False)
-class UsernameTooShortError(DomainError):
-    username: str
-
-    @property
-    def message(self) -> str:
-        return f"Username too short {self.username}"
-
-
-@dataclass(eq=False)
-class UsernameTooLongError(DomainError):
-    username: str
-
-    @property
-    def message(self) -> str:
-        return f"Username too long {self.username}"
-
-
-@dataclass(eq=False)
-class WrongUsernameFormatError(DomainError):
-    username: str
-
-    @property
-    def message(self) -> str:
-        return f'Wrong username format "{self.username}"'
-
-
 @dataclass(eq=False)
 class PasswordTooShortError(DomainError):
     password: str

src/domain/user/exceptions.py src/domain/user/exceptions/password.pysrc/domain/user/exceptions.py renamed to src/domain/user/exceptions/password.py

@@ -0,0 +1,30 @@
+from dataclasses import dataclass
+
+from domain.base import DomainError
+
+
+@dataclass(eq=False)
+class UsernameTooShortError(DomainError):
+    username: str
+
+    @property
+    def message(self) -> str:
+        return f"Username too short {self.username}"
+
+
+@dataclass(eq=False)
+class UsernameTooLongError(DomainError):
+    username: str
+
+    @property
+    def message(self) -> str:
+        return f"Username too long {self.username}"
+
+
+@dataclass(eq=False)
+class WrongUsernameFormatError(DomainError):
+    username: str
+
+    @property
+    def message(self) -> str:
+        return f'Wrong username format "{self.username}"'

src/domain/user/exceptions/username.py

@@ -0,0 +1,12 @@
+from abc import ABC, abstractmethod
+
+from domain.user.entities.user import User
+
+class UserRepo(ABC):
+    @abstractmethod
+    async def get_by_id(self, user_id: int) -> User:
+        pass
+
+    @abstractmethod
+    async def save(self, user: User) -> None:
+        pass

src/domain/user/repositories/user_repo.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass
 
 from domain.base import BaseValueObject
-from domain.user.exceptions import (
+from domain.user.exceptions.password import (
     PasswordTooShortError,
     PasswordTooLongError,
     PasswordRequiresDigitError,

src/domain/user/value_objects/password.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass
 
 from domain.base import BaseValueObject
-from domain.user.exceptions import (
+from domain.user.exceptions.username import (
     UsernameTooShortError,
     UsernameTooLongError,
     WrongUsernameFormatError,

src/domain/user/value_objects/username.py

@@ -0,0 +1,27 @@
+from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, create_async_engine
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+from infrastructure.database.repositories.user_repo_impl import UserRepoImpl
+
+class Database:
+    def __init__(self):
+        self.engine: AsyncEngine | None = None
+        self.session_maker: async_sessionmaker | None = None
+
+    async def __aenter__(self):
+        self.engine = create_async_engine("postgres://")
+        self.session_maker = async_sessionmaker(
+            autocommit=False,
+            autoflush=False,
+            bind=self.engine,
+            class_=AsyncSession,
+            expire_on_commit=False,
+        )
+        return self
+
+    async def __aexit__(self, exc_type, exc_value, traceback):
+        if self.engine is not None:
+            await self.engine.dispose()
+
+    def get_user_repo(self) -> UserRepoImpl:
+        return UserRepoImpl(self.session_maker)

src/infrastructure/database/database.py

@@ -0,0 +1,18 @@
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlmodel import select
+
+from domain.user.entities.user import User
+from domain.user.repositories.user_repo import UserRepo
+
+class UserRepoImpl(UserRepo):
+    def __init__(self, session: AsyncSession):
+        self.session = session
+
+    async def get_by_id(self, user_id: int) -> User:
+        query = select(User).where(User.id == user_id)
+        result = await self.session.execute(query)
+        return result.scalar_one_or_none()
+
+    async def save(self, user: User) -> None:
+        self.session.add(user)
+        await self.session.commit()