26 lines
849 B
Python
26 lines
849 B
Python
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from app.core.base_crud import CRUDBase
|
||
|
|
from app.core.base_schema import AuthSchema
|
||
|
|
from app.core.exceptions import CustomException
|
||
|
|
|
||
|
|
from .model import VersionModel
|
||
|
|
from .schema import VersionCreateSchema, VersionUpdateSchema
|
||
|
|
|
||
|
|
|
||
|
|
class VersionCRUD(CRUDBase[VersionModel, VersionCreateSchema, VersionUpdateSchema]):
|
||
|
|
"""版本数据层"""
|
||
|
|
|
||
|
|
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
|
||
|
|
super().__init__(model=VersionModel, auth=auth, db=db)
|
||
|
|
|
||
|
|
async def set_status(self, id: int, status: int) -> VersionModel:
|
||
|
|
"""更新版本状态"""
|
||
|
|
obj = await self.get(id=id)
|
||
|
|
if not obj:
|
||
|
|
raise CustomException(msg="版本不存在")
|
||
|
|
obj.status = status
|
||
|
|
await self.db.flush()
|
||
|
|
await self.db.refresh(obj)
|
||
|
|
return obj
|