初始化提交
This commit is contained in:
commit
887c1f449e
963 files changed
+166869
No files matched your search
@@ -0,0 +1,65 @@
|
||||
"""资料导入数据访问(api-spec §8)。"""
|
||||
|
||||
from sqlalchemy import func, select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.models.imports import CharacterImportJob, ImportItem
|
||||
|
||||
|
||||
async def get_job(session: AsyncSession, job_id: str) -> CharacterImportJob | None:
|
||||
return await session.get(CharacterImportJob, job_id)
|
||||
|
||||
|
||||
async def list_jobs(
|
||||
session: AsyncSession,
|
||||
character_id: str,
|
||||
*,
|
||||
page: int = 1,
|
||||
page_size: int = 20,
|
||||
) -> tuple[list[CharacterImportJob], int]:
|
||||
conditions = [CharacterImportJob.character_id == character_id]
|
||||
total = (
|
||||
await session.execute(
|
||||
select(func.count()).select_from(CharacterImportJob).where(*conditions)
|
||||
)
|
||||
).scalar_one()
|
||||
items_result = await session.execute(
|
||||
select(CharacterImportJob)
|
||||
.where(*conditions)
|
||||
.order_by(CharacterImportJob.created_at.desc())
|
||||
.offset(max(page - 1, 0) * page_size)
|
||||
.limit(page_size)
|
||||
)
|
||||
return list(items_result.scalars().all()), int(total or 0)
|
||||
|
||||
|
||||
async def add_job(session: AsyncSession, job: CharacterImportJob) -> CharacterImportJob:
|
||||
session.add(job)
|
||||
await session.flush()
|
||||
return job
|
||||
|
||||
|
||||
async def list_items(
|
||||
session: AsyncSession,
|
||||
job_id: str,
|
||||
*,
|
||||
page: int = 1,
|
||||
page_size: int = 20,
|
||||
) -> tuple[list[ImportItem], int]:
|
||||
conditions = [ImportItem.import_job_id == job_id]
|
||||
total = (
|
||||
await session.execute(select(func.count()).select_from(ImportItem).where(*conditions))
|
||||
).scalar_one()
|
||||
items_result = await session.execute(
|
||||
select(ImportItem)
|
||||
.where(*conditions)
|
||||
.order_by(ImportItem.sequence.asc())
|
||||
.offset(max(page - 1, 0) * page_size)
|
||||
.limit(page_size)
|
||||
)
|
||||
return list(items_result.scalars().all()), int(total or 0)
|
||||
|
||||
|
||||
async def add_items(session: AsyncSession, items: list[ImportItem]) -> None:
|
||||
session.add_all(items)
|
||||
await session.flush()
|
||||
Reference in new issue
Block a user