Cheng Zhou
300 lines
10 KiB
Python
300 lines
10 KiB
Python
"""概览数据聚合CRUD操作"""
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Dict, List, Tuple
|
|
|
|
from sqlalchemy import func, select, and_
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.site import Site
|
|
from app.models.startup_feasibility import StartupFeasibility
|
|
from app.models.startup_ethics import StartupEthics
|
|
from app.models.finance_contract import FinanceContract
|
|
from app.models.kickoff_meeting import KickoffMeeting
|
|
from app.models.subject import Subject
|
|
|
|
|
|
async def get_project_overview(db: AsyncSession, study_id: uuid.UUID) -> Dict:
|
|
"""
|
|
获取项目概览数据，聚合各中心的进度信息
|
|
|
|
返回格式:
|
|
{
|
|
"study_id": "...",
|
|
"updated_at": "2026-01-16T16:10:00",
|
|
"centers": [...],
|
|
"enrollment_by_month": [...]
|
|
}
|
|
"""
|
|
# 获取所有中心
|
|
stmt = select(Site).where(Site.study_id == study_id).order_by(Site.created_at)
|
|
result = await db.execute(stmt)
|
|
sites = result.scalars().all()
|
|
|
|
if not sites:
|
|
return {
|
|
"study_id": str(study_id),
|
|
"updated_at": datetime.now().isoformat(),
|
|
"centers": [],
|
|
"enrollment_by_month": []
|
|
}
|
|
|
|
site_ids = [site.id for site in sites]
|
|
|
|
# 并行获取各类数据
|
|
feasibility_data = await _get_feasibility_data(db, study_id, site_ids)
|
|
ethics_data = await _get_ethics_data(db, study_id, site_ids)
|
|
contract_data = await _get_contract_data(db, study_id, site_ids)
|
|
kickoff_data = await _get_kickoff_data(db, study_id, site_ids)
|
|
enrollment_data = await _get_enrollment_data(db, study_id, site_ids)
|
|
monthly_enrollment = await _get_monthly_enrollment(db, study_id)
|
|
|
|
# 组装中心数据
|
|
centers = []
|
|
for site in sites:
|
|
site_id_str = str(site.id)
|
|
|
|
# 机构立项状态
|
|
feas = feasibility_data.get(site_id_str, {})
|
|
institution_status = "COMPLETED" if feas.get("approved_date") else "IN_PROGRESS" if feas.get("submit_date") else "NOT_STARTED"
|
|
institution_completed = feas.get("approved_date")
|
|
|
|
# 伦理审批状态 - 依赖机构立项完成
|
|
eth = ethics_data.get(site_id_str, {})
|
|
if institution_status == "COMPLETED":
|
|
ethics_status = "COMPLETED" if eth.get("approved_date") else "IN_PROGRESS" if eth.get("submit_date") else "NOT_STARTED"
|
|
else:
|
|
# 机构立项未完成,伦理强制为未开始
|
|
ethics_status = "NOT_STARTED"
|
|
ethics_completed = eth.get("approved_date") if institution_status == "COMPLETED" else None
|
|
|
|
# 合同签署状态 - 依赖伦理审批完成
|
|
contract = contract_data.get(site_id_str, {})
|
|
if ethics_status == "COMPLETED":
|
|
contract_status = "COMPLETED" if contract.get("signed_date") else "NOT_STARTED"
|
|
else:
|
|
# 伦理未完成,合同强制为未开始
|
|
contract_status = "NOT_STARTED"
|
|
contract_completed = contract.get("signed_date") if ethics_status == "COMPLETED" else None
|
|
|
|
# 启动会状态 - 依赖合同签署完成
|
|
kickoff = kickoff_data.get(site_id_str, {})
|
|
if contract_status == "COMPLETED":
|
|
startup_status = "COMPLETED" if kickoff.get("kickoff_date") else "NOT_STARTED"
|
|
else:
|
|
# 合同未完成,启动会强制为未开始
|
|
startup_status = "NOT_STARTED"
|
|
startup_completed = kickoff.get("kickoff_date") if contract_status == "COMPLETED" else None
|
|
|
|
# 入组状态 - 依赖启动会完成
|
|
enrollment = enrollment_data.get(site_id_str, {"actual": 0})
|
|
enrollment_actual = enrollment["actual"]
|
|
enrollment_target = site.enrollment_target or 0
|
|
|
|
if startup_status == "COMPLETED":
|
|
if enrollment_actual >= enrollment_target and enrollment_target > 0:
|
|
enrollment_status = "COMPLETED"
|
|
elif enrollment_actual > 0:
|
|
enrollment_status = "IN_PROGRESS"
|
|
else:
|
|
enrollment_status = "NOT_STARTED"
|
|
else:
|
|
# 启动会未完成,入组强制为未开始
|
|
enrollment_status = "NOT_STARTED"
|
|
|
|
# 检查状态 - 依赖入组完成
|
|
# 暂未实现数据源,固定为NOT_STARTED
|
|
if enrollment_status == "COMPLETED":
|
|
inspection_status = "NOT_STARTED" # 未来从inspection表获取
|
|
else:
|
|
inspection_status = "NOT_STARTED"
|
|
|
|
# 关中心状态 - 依赖检查完成
|
|
# 暂未实现数据源,固定为NOT_STARTED
|
|
if inspection_status == "COMPLETED":
|
|
closeout_status = "NOT_STARTED" # 未来从closeout表获取
|
|
else:
|
|
closeout_status = "NOT_STARTED"
|
|
|
|
|
|
centers.append({
|
|
"center_id": str(site.id),
|
|
"center_name": site.name,
|
|
"institution_initiation_status": institution_status,
|
|
"ethics_status": ethics_status,
|
|
"contract_sign_status": contract_status,
|
|
"startup_status": startup_status,
|
|
"enrollment_status": enrollment_status,
|
|
"inspection_status": inspection_status,
|
|
"closeout_status": closeout_status,
|
|
"institution_initiation_completed_at": institution_completed,
|
|
"ethics_completed_at": ethics_completed,
|
|
"contract_sign_completed_at": contract_completed,
|
|
"startup_completed_at": startup_completed,
|
|
"enrollment_completed_at": None, # 暂未实现
|
|
"inspection_completed_at": None,
|
|
"closeout_completed_at": None,
|
|
"enrollment_target": enrollment_target,
|
|
"enrollment_actual": enrollment_actual,
|
|
})
|
|
|
|
return {
|
|
"study_id": str(study_id),
|
|
"updated_at": datetime.now().isoformat(),
|
|
"centers": centers,
|
|
"enrollment_by_month": monthly_enrollment
|
|
}
|
|
|
|
|
|
async def _get_feasibility_data(db: AsyncSession, study_id: uuid.UUID, site_ids: List[uuid.UUID]) -> Dict:
|
|
"""获取各中心的立项数据"""
|
|
stmt = select(
|
|
StartupFeasibility.site_id,
|
|
StartupFeasibility.submit_date,
|
|
StartupFeasibility.approved_date
|
|
).where(
|
|
and_(
|
|
StartupFeasibility.study_id == study_id,
|
|
StartupFeasibility.site_id.in_(site_ids)
|
|
)
|
|
).order_by(StartupFeasibility.created_at.desc())
|
|
|
|
result = await db.execute(stmt)
|
|
rows = result.all()
|
|
|
|
data = {}
|
|
for row in rows:
|
|
site_id_str = str(row.site_id)
|
|
if site_id_str not in data:
|
|
data[site_id_str] = {
|
|
"submit_date": row.submit_date.isoformat() if row.submit_date else None,
|
|
"approved_date": row.approved_date.isoformat() if row.approved_date else None,
|
|
}
|
|
|
|
return data
|
|
|
|
|
|
async def _get_ethics_data(db: AsyncSession, study_id: uuid.UUID, site_ids: List[uuid.UUID]) -> Dict:
|
|
"""获取各中心的伦理数据"""
|
|
stmt = select(
|
|
StartupEthics.site_id,
|
|
StartupEthics.submit_date,
|
|
StartupEthics.approved_date
|
|
).where(
|
|
and_(
|
|
StartupEthics.study_id == study_id,
|
|
StartupEthics.site_id.in_(site_ids)
|
|
)
|
|
).order_by(StartupEthics.created_at.desc())
|
|
|
|
result = await db.execute(stmt)
|
|
rows = result.all()
|
|
|
|
data = {}
|
|
for row in rows:
|
|
site_id_str = str(row.site_id)
|
|
if site_id_str not in data:
|
|
data[site_id_str] = {
|
|
"submit_date": row.submit_date.isoformat() if row.submit_date else None,
|
|
"approved_date": row.approved_date.isoformat() if row.approved_date else None,
|
|
}
|
|
|
|
return data
|
|
|
|
|
|
async def _get_contract_data(db: AsyncSession, study_id: uuid.UUID, site_ids: List[uuid.UUID]) -> Dict:
|
|
"""获取各中心的合同签署数据"""
|
|
# FinanceContract 通过 site_name 关联，需要先获取site名称映射
|
|
stmt_sites = select(Site.id, Site.name).where(Site.id.in_(site_ids))
|
|
result = await db.execute(stmt_sites)
|
|
site_name_map = {row.name: str(row.id) for row in result.all()}
|
|
|
|
stmt = select(
|
|
FinanceContract.site_name,
|
|
func.min(FinanceContract.signed_date).label("signed_date")
|
|
).where(
|
|
FinanceContract.study_id == study_id
|
|
).group_by(FinanceContract.site_name)
|
|
|
|
result = await db.execute(stmt)
|
|
rows = result.all()
|
|
|
|
data = {}
|
|
for row in rows:
|
|
site_id_str = site_name_map.get(row.site_name)
|
|
if site_id_str:
|
|
data[site_id_str] = {
|
|
"signed_date": row.signed_date.isoformat() if row.signed_date else None,
|
|
}
|
|
|
|
return data
|
|
|
|
|
|
async def _get_kickoff_data(db: AsyncSession, study_id: uuid.UUID, site_ids: List[uuid.UUID]) -> Dict:
|
|
"""获取各中心的启动会数据"""
|
|
stmt = select(
|
|
KickoffMeeting.site_id,
|
|
KickoffMeeting.kickoff_date
|
|
).where(
|
|
and_(
|
|
KickoffMeeting.study_id == study_id,
|
|
KickoffMeeting.site_id.in_(site_ids)
|
|
)
|
|
).order_by(KickoffMeeting.created_at.desc())
|
|
|
|
result = await db.execute(stmt)
|
|
rows = result.all()
|
|
|
|
data = {}
|
|
for row in rows:
|
|
site_id_str = str(row.site_id)
|
|
if site_id_str not in data:
|
|
data[site_id_str] = {
|
|
"kickoff_date": row.kickoff_date.isoformat() if row.kickoff_date else None,
|
|
}
|
|
|
|
return data
|
|
|
|
|
|
async def _get_enrollment_data(db: AsyncSession, study_id: uuid.UUID, site_ids: List[uuid.UUID]) -> Dict:
|
|
"""获取各中心的入组数据"""
|
|
stmt = select(
|
|
Subject.site_id,
|
|
func.count(Subject.id).label("count")
|
|
).where(
|
|
and_(
|
|
Subject.study_id == study_id,
|
|
Subject.site_id.in_(site_ids),
|
|
Subject.enrollment_date.isnot(None)
|
|
)
|
|
).group_by(Subject.site_id)
|
|
|
|
result = await db.execute(stmt)
|
|
rows = result.all()
|
|
|
|
data = {}
|
|
for row in rows:
|
|
data[str(row.site_id)] = {"actual": row.count}
|
|
|
|
return data
|
|
|
|
|
|
async def _get_monthly_enrollment(db: AsyncSession, study_id: uuid.UUID) -> List[Dict]:
|
|
"""获取月度入组统计"""
|
|
month_col = func.to_char(Subject.enrollment_date, 'YYYY-MM').label("month")
|
|
|
|
stmt = select(
|
|
month_col,
|
|
func.count(Subject.id).label("count")
|
|
).where(
|
|
and_(
|
|
Subject.study_id == study_id,
|
|
Subject.enrollment_date.isnot(None)
|
|
)
|
|
).group_by(month_col).order_by(month_col)
|
|
|
|
result = await db.execute(stmt)
|
|
rows = result.all()
|
|
|
|
return [{"month": row.month, "count": row.count} for row in rows]
|