Add initial project
This commit is contained in:
@@ -0,0 +1 @@
|
||||
"""Repositories module for data access layer."""
|
||||
@@ -0,0 +1,49 @@
|
||||
"""Base repository for common database operations."""
|
||||
from typing import Generic, TypeVar, Type, List, Optional
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
ModelType = TypeVar("ModelType")
|
||||
|
||||
|
||||
class BaseRepository(Generic[ModelType]):
|
||||
"""Base repository with common CRUD operations."""
|
||||
|
||||
def __init__(self, model: Type[ModelType], db: Session):
|
||||
"""Initialize repository with model and database session."""
|
||||
self.model = model
|
||||
self.db = db
|
||||
|
||||
def get_by_id(self, id: int) -> Optional[ModelType]:
|
||||
"""Get a single record by ID."""
|
||||
return self.db.query(self.model).filter(self.model.id == id).first()
|
||||
|
||||
def get_all(self) -> List[ModelType]:
|
||||
"""Get all records."""
|
||||
return self.db.query(self.model).all()
|
||||
|
||||
def create(self, **kwargs) -> ModelType:
|
||||
"""Create a new record."""
|
||||
db_obj = self.model(**kwargs)
|
||||
self.db.add(db_obj)
|
||||
self.db.commit()
|
||||
self.db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def update(self, id: int, **kwargs) -> Optional[ModelType]:
|
||||
"""Update a record by ID."""
|
||||
db_obj = self.get_by_id(id)
|
||||
if db_obj:
|
||||
for key, value in kwargs.items():
|
||||
setattr(db_obj, key, value)
|
||||
self.db.commit()
|
||||
self.db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def delete(self, id: int) -> bool:
|
||||
"""Delete a record by ID."""
|
||||
db_obj = self.get_by_id(id)
|
||||
if db_obj:
|
||||
self.db.delete(db_obj)
|
||||
self.db.commit()
|
||||
return True
|
||||
return False
|
||||
@@ -0,0 +1,33 @@
|
||||
"""Challenge repository for data access operations."""
|
||||
from typing import List, Optional
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import func
|
||||
from backend.models.challenge import Challenge
|
||||
from backend.repositories.base import BaseRepository
|
||||
|
||||
|
||||
class ChallengeRepository(BaseRepository[Challenge]):
|
||||
"""Repository for Challenge model operations."""
|
||||
|
||||
def __init__(self, db: Session):
|
||||
"""Initialize challenge repository."""
|
||||
super().__init__(Challenge, db)
|
||||
|
||||
def get_by_user_id(self, user_id: int) -> List[Challenge]:
|
||||
"""Get all challenges for a user ordered by display_order."""
|
||||
return self.db.query(Challenge).filter(
|
||||
Challenge.user_id == user_id
|
||||
).order_by(Challenge.display_order.asc()).all()
|
||||
|
||||
def get_by_user_and_id(self, user_id: int, challenge_id: int) -> Optional[Challenge]:
|
||||
"""Get a challenge by user ID and challenge ID."""
|
||||
return self.db.query(Challenge).filter(
|
||||
Challenge.id == challenge_id,
|
||||
Challenge.user_id == user_id
|
||||
).first()
|
||||
|
||||
def get_max_display_order(self, user_id: int) -> Optional[int]:
|
||||
"""Get the maximum display_order for a user."""
|
||||
return self.db.query(func.max(Challenge.display_order)).filter(
|
||||
Challenge.user_id == user_id
|
||||
).scalar()
|
||||
@@ -0,0 +1,17 @@
|
||||
"""User repository for data access operations."""
|
||||
from typing import Optional
|
||||
from sqlalchemy.orm import Session
|
||||
from backend.models.user import User
|
||||
from backend.repositories.base import BaseRepository
|
||||
|
||||
|
||||
class UserRepository(BaseRepository[User]):
|
||||
"""Repository for User model operations."""
|
||||
|
||||
def __init__(self, db: Session):
|
||||
"""Initialize user repository."""
|
||||
super().__init__(User, db)
|
||||
|
||||
def get_by_username(self, username: str) -> Optional[User]:
|
||||
"""Get a user by username."""
|
||||
return self.db.query(User).filter(User.username == username).first()
|
||||
Reference in New Issue
Block a user