51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
"""大周天运行记录数据访问。"""
|
|
|
|
from sqlalchemy import func, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.big_cycle import BigCycle
|
|
|
|
|
|
async def add_cycle(session: AsyncSession, cycle: BigCycle) -> BigCycle:
|
|
session.add(cycle)
|
|
await session.flush()
|
|
return cycle
|
|
|
|
|
|
async def get_cycle(session: AsyncSession, cycle_id: str) -> BigCycle | None:
|
|
return await session.get(BigCycle, cycle_id)
|
|
|
|
|
|
async def get_last_completed(session: AsyncSession, character_id: str) -> BigCycle | None:
|
|
result = await session.execute(
|
|
select(BigCycle)
|
|
.where(
|
|
BigCycle.character_id == character_id,
|
|
BigCycle.completed_at.is_not(None),
|
|
)
|
|
.order_by(BigCycle.completed_at.desc())
|
|
.limit(1)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def list_cycles(
|
|
session: AsyncSession,
|
|
character_id: str,
|
|
*,
|
|
page: int = 1,
|
|
page_size: int = 20,
|
|
) -> tuple[list[BigCycle], int]:
|
|
conditions = [BigCycle.character_id == character_id]
|
|
total = (
|
|
await session.execute(select(func.count()).select_from(BigCycle).where(*conditions))
|
|
).scalar_one()
|
|
items_result = await session.execute(
|
|
select(BigCycle)
|
|
.where(*conditions)
|
|
.order_by(BigCycle.created_at.desc())
|
|
.offset(max(page - 1, 0) * page_size)
|
|
.limit(page_size)
|
|
)
|
|
return list(items_result.scalars().all()), int(total or 0)
|