init

api/repository/user.py

@@ -0,0 +1,48 @@
+from contextlib import AbstractContextManager
+from typing import Callable
+
+from model.user import User
+from sqlalchemy.orm import Session
+
+
+class UserRepository:
+    def __init__(self, session_factory: Callable[..., AbstractContextManager[Session]]) -> None:
+        self.session_factory = session_factory
+
+    def get_all(self):
+        with self.session_factory() as session:
+            return session.query(User).all()
+
+    def get_by_id(self, user_id: int) -> User:
+        with self.session_factory() as session:
+            user = session.query(User).filter(User.id == user_id).first()
+            if not user:
+                raise UserNotFoundError(user_id)
+            return user
+
+    def add(self, email: str, password: str, is_active: bool = True) -> User:
+        with self.session_factory() as session:
+            user = User(email=email, hashed_password=password, is_active=is_active)
+            session.add(user)
+            session.commit()
+            session.refresh(user)
+            return user
+
+    def delete_by_id(self, user_id: int) -> None:
+        with self.session_factory() as session:
+            entity: User = session.query(User).filter(User.id == user_id).first()
+            if not entity:
+                raise UserNotFoundError(user_id)
+            session.delete(entity)
+            session.commit()
+
+
+class NotFoundError(Exception):
+    entity_name: str
+
+    def __init__(self, entity_id):
+        super().__init__(f"{self.entity_name} not found, id: {entity_id}")
+
+
+class UserNotFoundError(NotFoundError):
+    entity_name: str = "User"