Cheng Zhou
73 lines
2.3 KiB
Python
73 lines
2.3 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 import Document
|
|
|
|
|
|
async def create(db: AsyncSession, doc: Document, *, commit: bool = True) -> Document:
|
|
db.add(doc)
|
|
if commit:
|
|
await db.commit()
|
|
await db.refresh(doc)
|
|
return doc
|
|
|
|
|
|
async def get(db: AsyncSession, document_id: uuid.UUID) -> Document | None:
|
|
result = await db.execute(select(Document).where(Document.id == document_id))
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def get_by_trial_doc_no(db: AsyncSession, trial_id: uuid.UUID, doc_no: str) -> Document | None:
|
|
result = await db.execute(select(Document).where(Document.trial_id == trial_id, Document.doc_no == doc_no))
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def list_documents(
|
|
db: AsyncSession,
|
|
*,
|
|
trial_id: uuid.UUID,
|
|
site_id: uuid.UUID | None = None,
|
|
doc_type: str | None = None,
|
|
status: str | None = None,
|
|
scope_type: str | None = None,
|
|
etmf_node_id: uuid.UUID | None = None,
|
|
cra_site_ids: set[uuid.UUID] | None = None,
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
) -> Sequence[Document]:
|
|
stmt = select(Document).where(Document.trial_id == trial_id)
|
|
if cra_site_ids is not None:
|
|
if not cra_site_ids:
|
|
stmt = stmt.where(Document.scope_type == "GLOBAL")
|
|
else:
|
|
stmt = stmt.where((Document.scope_type == "GLOBAL") | (Document.site_id.in_(cra_site_ids)))
|
|
if scope_type:
|
|
stmt = stmt.where(Document.scope_type == scope_type)
|
|
if doc_type:
|
|
stmt = stmt.where(Document.doc_type == doc_type)
|
|
if status:
|
|
stmt = stmt.where(Document.status == status)
|
|
if site_id:
|
|
stmt = stmt.where(Document.site_id == site_id)
|
|
if etmf_node_id:
|
|
stmt = stmt.where(Document.etmf_node_id == etmf_node_id)
|
|
stmt = stmt.order_by(Document.updated_at.desc()).offset(skip).limit(limit)
|
|
result = await db.execute(stmt)
|
|
return result.scalars().all()
|
|
|
|
|
|
async def update(db: AsyncSession, document_id: uuid.UUID, values: dict, *, commit: bool = True) -> None:
|
|
if values:
|
|
await db.execute(
|
|
sa_update(Document)
|
|
.where(Document.id == document_id)
|
|
.values(**values)
|
|
)
|
|
if commit:
|
|
await db.commit()
|