139 lines
3.0 KiB
Python
139 lines
3.0 KiB
Python
"""
|
|
基础仓库类
|
|
|
|
提供通用的 CRUD 操作封装。
|
|
"""
|
|
|
|
from typing import Any, Generic, TypeVar
|
|
from uuid import uuid4
|
|
|
|
from sqlalchemy import select, func
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database import Base
|
|
|
|
ModelT = TypeVar("ModelT", bound=Base)
|
|
|
|
|
|
class BaseRepository(Generic[ModelT]):
|
|
"""
|
|
基础仓库类
|
|
|
|
提供通用的数据库操作方法。
|
|
"""
|
|
|
|
model: type[ModelT]
|
|
|
|
def __init__(self, session: AsyncSession):
|
|
"""
|
|
初始化仓库
|
|
|
|
Args:
|
|
session: 异步数据库会话
|
|
"""
|
|
self.session = session
|
|
|
|
async def get_by_id(self, id: str) -> ModelT | None:
|
|
"""
|
|
通过 ID 获取实体
|
|
|
|
Args:
|
|
id: 实体 ID
|
|
|
|
Returns:
|
|
实体对象或 None
|
|
"""
|
|
return await self.session.get(self.model, id)
|
|
|
|
async def get_all(
|
|
self,
|
|
*,
|
|
offset: int = 0,
|
|
limit: int = 100,
|
|
) -> list[ModelT]:
|
|
"""
|
|
获取所有实体
|
|
|
|
Args:
|
|
offset: 偏移量
|
|
limit: 限制数量
|
|
|
|
Returns:
|
|
实体列表
|
|
"""
|
|
stmt = select(self.model).offset(offset).limit(limit)
|
|
result = await self.session.execute(stmt)
|
|
return list(result.scalars().all())
|
|
|
|
async def count(self) -> int:
|
|
"""
|
|
获取实体总数
|
|
|
|
Returns:
|
|
实体数量
|
|
"""
|
|
stmt = select(func.count()).select_from(self.model)
|
|
result = await self.session.execute(stmt)
|
|
return result.scalar() or 0
|
|
|
|
async def create(self, **kwargs: Any) -> ModelT:
|
|
"""
|
|
创建新实体
|
|
|
|
Args:
|
|
**kwargs: 实体属性
|
|
|
|
Returns:
|
|
新创建的实体
|
|
"""
|
|
if "id" not in kwargs:
|
|
kwargs["id"] = str(uuid4())
|
|
|
|
entity = self.model(**kwargs)
|
|
self.session.add(entity)
|
|
await self.session.flush()
|
|
await self.session.refresh(entity)
|
|
return entity
|
|
|
|
async def update(
|
|
self,
|
|
entity: ModelT,
|
|
**kwargs: Any,
|
|
) -> ModelT:
|
|
"""
|
|
更新实体
|
|
|
|
Args:
|
|
entity: 要更新的实体
|
|
**kwargs: 要更新的属性
|
|
|
|
Returns:
|
|
更新后的实体
|
|
"""
|
|
for key, value in kwargs.items():
|
|
if hasattr(entity, key):
|
|
setattr(entity, key, value)
|
|
|
|
await self.session.flush()
|
|
await self.session.refresh(entity)
|
|
return entity
|
|
|
|
async def delete(self, entity: ModelT) -> None:
|
|
"""
|
|
删除实体
|
|
|
|
Args:
|
|
entity: 要删除的实体
|
|
"""
|
|
await self.session.delete(entity)
|
|
await self.session.flush()
|
|
|
|
async def commit(self) -> None:
|
|
"""提交事务"""
|
|
await self.session.commit()
|
|
|
|
async def rollback(self) -> None:
|
|
"""回滚事务"""
|
|
await self.session.rollback()
|
|
|