Cheng Zhou
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from typing import Sequence
|
|
|
|
from sqlalchemy import select, update as sa_update
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.document_version import DocumentVersion, DocumentVersionStatus
|
|
|
|
|
|
async def create(db: AsyncSession, version: DocumentVersion, *, commit: bool = True) -> DocumentVersion:
|
|
db.add(version)
|
|
if commit:
|
|
await db.commit()
|
|
await db.refresh(version)
|
|
return version
|
|
|
|
|
|
async def get(db: AsyncSession, version_id: uuid.UUID) -> DocumentVersion | None:
|
|
result = await db.execute(select(DocumentVersion).where(DocumentVersion.id == version_id))
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def get_by_document_version_no(
|
|
db: AsyncSession,
|
|
document_id: uuid.UUID,
|
|
version_no: str,
|
|
) -> DocumentVersion | None:
|
|
result = await db.execute(
|
|
select(DocumentVersion).where(
|
|
DocumentVersion.document_id == document_id,
|
|
DocumentVersion.version_no == version_no,
|
|
)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def list_by_document(
|
|
db: AsyncSession,
|
|
document_id: uuid.UUID,
|
|
skip: int = 0,
|
|
limit: int = 200,
|
|
) -> Sequence[DocumentVersion]:
|
|
result = await db.execute(
|
|
select(DocumentVersion)
|
|
.where(DocumentVersion.document_id == document_id)
|
|
.order_by(DocumentVersion.created_at.desc())
|
|
.offset(skip)
|
|
.limit(limit)
|
|
)
|
|
return result.scalars().all()
|
|
|
|
|
|
async def list_pending_versions(db: AsyncSession, document_id: uuid.UUID) -> Sequence[DocumentVersion]:
|
|
result = await db.execute(
|
|
select(DocumentVersion).where(
|
|
DocumentVersion.document_id == document_id,
|
|
DocumentVersion.status.in_(
|
|
[
|
|
DocumentVersionStatus.SUBMITTED,
|
|
DocumentVersionStatus.APPROVED,
|
|
]
|
|
),
|
|
)
|
|
)
|
|
return result.scalars().all()
|
|
|
|
|
|
async def update(db: AsyncSession, version_id: uuid.UUID, values: dict, *, commit: bool = True) -> None:
|
|
if values:
|
|
await db.execute(
|
|
sa_update(DocumentVersion)
|
|
.where(DocumentVersion.id == version_id)
|
|
.values(**values)
|
|
)
|
|
if commit:
|
|
await db.commit()
|