feat: 清理示例数据并统一空态体验
This commit is contained in:
Cheng Zhou
@@ -1,4 +1,4 @@
|
||||
"""seed overview demo data
|
||||
"""disable overview demo data seed
|
||||
|
||||
Revision ID: 20260116_03
|
||||
Revises: 20260116_02
|
||||
@@ -6,14 +6,7 @@ Create Date: 2026-01-16 16:30:00.000000
|
||||
|
||||
"""
|
||||
from typing import Sequence, Union
|
||||
from datetime import date, timedelta
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = '20260116_03'
|
||||
down_revision: Union[str, None] = '20260116_02'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
@@ -21,179 +14,11 @@ depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
# 获取connection
|
||||
conn = op.get_bind()
|
||||
|
||||
# 首先获取第一个项目ID
|
||||
result = conn.execute(sa.text("SELECT id FROM studies LIMIT 1"))
|
||||
study_row = result.first()
|
||||
|
||||
if not study_row:
|
||||
print("没有项目,跳过示例数据注入")
|
||||
return
|
||||
|
||||
study_id = study_row[0]
|
||||
|
||||
# 获取该项目的中心
|
||||
result = conn.execute(sa.text("SELECT id, name FROM sites WHERE study_id = :study_id ORDER BY created_at LIMIT 5"), {"study_id": study_id})
|
||||
sites = result.fetchall()
|
||||
|
||||
if not sites:
|
||||
print("没有中心,跳过示例数据注入")
|
||||
return
|
||||
|
||||
print(f"为 {len(sites)} 个中心注入示例数据...")
|
||||
|
||||
# 清理可能存在的演示数据(使用ON CONFLICT处理或先删除)
|
||||
# 收集site_ids
|
||||
site_ids = [site[0] for site in sites]
|
||||
|
||||
# 删除这些中心的existing数据 (如果是演示数据)
|
||||
# 由于我们无法判断哪些是演示数据,使用ON CONFLICT DO NOTHING策略
|
||||
|
||||
# 更新入组目标
|
||||
targets = [80, 60, 70, 50, 40]
|
||||
for idx, site in enumerate(sites):
|
||||
site_id = site[0]
|
||||
target = targets[idx] if idx < len(targets) else 50
|
||||
conn.execute(
|
||||
sa.text("UPDATE sites SET enrollment_target = :target WHERE id = :site_id"),
|
||||
{"target": target, "site_id": site_id}
|
||||
)
|
||||
|
||||
# 基准日期
|
||||
base_date = date(2024, 12, 1)
|
||||
|
||||
# 为每个中心添加启动流程数据
|
||||
for idx, site in enumerate(sites):
|
||||
site_id = site[0]
|
||||
|
||||
# 机构立项 - 使用 ON CONFLICT DO NOTHING (假设有唯一约束)
|
||||
submit_date = base_date + timedelta(days=idx * 10)
|
||||
accept_date = base_date + timedelta(days=idx * 10 + 5)
|
||||
approved_date = base_date + timedelta(days=idx * 10 + 15) if idx < 4 else None
|
||||
|
||||
# 检查是否已存在
|
||||
existing = conn.execute(sa.text(
|
||||
"SELECT 1 FROM startup_feasibility WHERE study_id = :study_id AND site_id = :site_id LIMIT 1"
|
||||
), {"study_id": study_id, "site_id": site_id}).first()
|
||||
|
||||
if not existing:
|
||||
conn.execute(sa.text("""
|
||||
INSERT INTO startup_feasibility (id, study_id, site_id, submit_date, accept_date, approved_date, project_no, created_at)
|
||||
VALUES (gen_random_uuid(), :study_id, :site_id, :submit_date, :accept_date, :approved_date, :project_no, NOW())
|
||||
"""), {
|
||||
"study_id": study_id,
|
||||
"site_id": site_id,
|
||||
"submit_date": submit_date,
|
||||
"accept_date": accept_date,
|
||||
"approved_date": approved_date,
|
||||
"project_no": f"PROJ-{idx+1:03d}"
|
||||
})
|
||||
|
||||
# 伦理审批
|
||||
ethics_submit = base_date + timedelta(days=idx * 10 + 20)
|
||||
ethics_accept = base_date + timedelta(days=idx * 10 + 25)
|
||||
meeting_date = base_date + timedelta(days=idx * 10 + 35) if idx < 4 else None
|
||||
ethics_approved = base_date + timedelta(days=idx * 10 + 40) if idx < 4 else None
|
||||
|
||||
existing = conn.execute(sa.text(
|
||||
"SELECT 1 FROM startup_ethics WHERE study_id = :study_id AND site_id = :site_id LIMIT 1"
|
||||
), {"study_id": study_id, "site_id": site_id}).first()
|
||||
|
||||
if not existing:
|
||||
conn.execute(sa.text("""
|
||||
INSERT INTO startup_ethics (id, study_id, site_id, submit_date, accept_date, meeting_date, approved_date, approval_no, created_at)
|
||||
VALUES (gen_random_uuid(), :study_id, :site_id, :submit_date, :accept_date, :meeting_date, :approved_date, :approval_no, NOW())
|
||||
"""), {
|
||||
"study_id": study_id,
|
||||
"site_id": site_id,
|
||||
"submit_date": ethics_submit,
|
||||
"accept_date": ethics_accept,
|
||||
"meeting_date": meeting_date,
|
||||
"approved_date": ethics_approved,
|
||||
"approval_no": f"EC-{idx+1:03d}" if idx < 4 else None
|
||||
})
|
||||
|
||||
# 合同签署 (前3个中心)
|
||||
if idx < 3:
|
||||
site_name = site[1]
|
||||
signed_date = base_date + timedelta(days=idx * 10 + 50)
|
||||
|
||||
# 检查合同是否已存在
|
||||
existing = conn.execute(sa.text(
|
||||
"SELECT 1 FROM finance_contracts WHERE study_id = :study_id AND site_name = :site_name LIMIT 1"
|
||||
), {"study_id": study_id, "site_name": site_name}).first()
|
||||
|
||||
if not existing:
|
||||
conn.execute(sa.text("""
|
||||
INSERT INTO finance_contracts (id, study_id, site_name, contract_no, signed_date, amount, currency, created_at)
|
||||
VALUES (gen_random_uuid(), :study_id, :site_name, :contract_no, :signed_date, 500000.0, 'CNY', NOW())
|
||||
"""), {
|
||||
"study_id": study_id,
|
||||
"site_name": site_name,
|
||||
"contract_no": f"CT-{idx+1:03d}",
|
||||
"signed_date": signed_date
|
||||
})
|
||||
|
||||
# 启动会 (前3个中心)
|
||||
if idx < 3:
|
||||
kickoff_date = base_date + timedelta(days=idx * 10 + 60)
|
||||
|
||||
# 检查启动会是否已存在
|
||||
existing = conn.execute(sa.text(
|
||||
"SELECT 1 FROM kickoff_meetings WHERE study_id = :study_id AND site_id = :site_id LIMIT 1"
|
||||
), {"study_id": study_id, "site_id": site_id}).first()
|
||||
|
||||
if not existing:
|
||||
conn.execute(sa.text("""
|
||||
INSERT INTO kickoff_meetings (id, study_id, site_id, kickoff_date, attendees, created_at)
|
||||
VALUES (gen_random_uuid(), :study_id, :site_id, :kickoff_date, '["研究者", "CRA", "PM"]'::jsonb, NOW())
|
||||
"""), {
|
||||
"study_id": study_id,
|
||||
"site_id": site_id,
|
||||
"kickoff_date": kickoff_date
|
||||
})
|
||||
|
||||
# 为前3个中心添加受试者
|
||||
enrollment_counts = [52, 18, 70]
|
||||
start_enrollment = date(2025, 1, 1)
|
||||
|
||||
for idx in range(min(3, len(sites))):
|
||||
site_id = sites[idx][0]
|
||||
count = enrollment_counts[idx] if idx < len(enrollment_counts) else 0
|
||||
|
||||
# 检查该中心是否已有受试者数据
|
||||
existing_count = conn.execute(sa.text(
|
||||
"SELECT COUNT(*) FROM subjects WHERE study_id = :study_id AND site_id = :site_id"
|
||||
), {"study_id": study_id, "site_id": site_id}).scalar()
|
||||
|
||||
# 如果已有数据,跳过该中心
|
||||
if existing_count > 0:
|
||||
print(f" 中心 {idx+1} 已有 {existing_count} 名受试者,跳过")
|
||||
continue
|
||||
|
||||
for i in range(count):
|
||||
enrollment_date = start_enrollment + timedelta(days=i * 4)
|
||||
screening_date = enrollment_date - timedelta(days=7)
|
||||
consent_date = enrollment_date - timedelta(days=3)
|
||||
|
||||
conn.execute(sa.text("""
|
||||
INSERT INTO subjects (id, study_id, site_id, subject_no, status, screening_date, consent_date, enrollment_date, created_at)
|
||||
VALUES (gen_random_uuid(), :study_id, :site_id, :subject_no, 'ENROLLED', :screening_date, :consent_date, :enrollment_date, NOW())
|
||||
"""), {
|
||||
"study_id": study_id,
|
||||
"site_id": site_id,
|
||||
"subject_no": f"S{idx+1:02d}-{i+1:03d}",
|
||||
"screening_date": screening_date,
|
||||
"consent_date": consent_date,
|
||||
"enrollment_date": enrollment_date
|
||||
})
|
||||
|
||||
print("✅ 示例数据注入完成")
|
||||
# This historical revision is intentionally kept as a no-op so fresh
|
||||
# databases no longer receive overview demo data during migration.
|
||||
return
|
||||
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
# 删除示例数据 (可选 - 也可以留空)
|
||||
pass
|
||||
return
|
||||
|
||||
@@ -0,0 +1,397 @@
|
||||
"""cleanup overview demo data
|
||||
|
||||
Revision ID: 20260331_01
|
||||
Revises: 20260302_02
|
||||
Create Date: 2026-03-31 10:45:00.000000
|
||||
|
||||
"""
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
|
||||
revision: str = "20260331_01"
|
||||
down_revision: Union[str, None] = "20260302_02"
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
DEMO_FEASIBILITY_CODES = ("PROJ-001", "PROJ-002", "PROJ-003", "PROJ-004", "PROJ-005")
|
||||
DEMO_ETHICS_CODES = ("EC-001", "EC-002", "EC-003", "EC-004")
|
||||
DEMO_FINANCE_CONTRACT_CODES = ("CT-001", "CT-002", "CT-003")
|
||||
DEMO_STUDY_CODE = "DEMO-CTMS"
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
conn = op.get_bind()
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH candidate_studies AS (
|
||||
SELECT DISTINCT study_id
|
||||
FROM startup_feasibility
|
||||
WHERE project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id
|
||||
FROM startup_ethics
|
||||
WHERE approval_no = ANY(:ethics_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id
|
||||
FROM finance_contracts
|
||||
WHERE contract_no = ANY(:contract_codes)
|
||||
AND amount = 500000.0
|
||||
AND currency = 'CNY'
|
||||
UNION
|
||||
SELECT DISTINCT study_id
|
||||
FROM kickoff_meetings
|
||||
WHERE attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT study_id
|
||||
FROM subjects
|
||||
WHERE subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
),
|
||||
candidate_sites AS (
|
||||
SELECT DISTINCT site_id
|
||||
FROM startup_feasibility
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT site_id
|
||||
FROM startup_ethics
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND (
|
||||
approval_no = ANY(:ethics_codes)
|
||||
OR (
|
||||
approval_no IS NULL
|
||||
AND submit_date = DATE '2025-01-10'
|
||||
AND accept_date = DATE '2025-01-15'
|
||||
AND meeting_date IS NULL
|
||||
AND approved_date IS NULL
|
||||
)
|
||||
OR (
|
||||
approval_no IS NULL
|
||||
AND submit_date = DATE '2025-01-20'
|
||||
AND accept_date = DATE '2025-01-25'
|
||||
AND meeting_date IS NULL
|
||||
AND approved_date IS NULL
|
||||
)
|
||||
OR (
|
||||
approval_no IS NULL
|
||||
AND submit_date = DATE '2025-01-30'
|
||||
AND accept_date = DATE '2025-02-04'
|
||||
AND meeting_date IS NULL
|
||||
AND approved_date IS NULL
|
||||
)
|
||||
)
|
||||
UNION
|
||||
SELECT DISTINCT site_id
|
||||
FROM kickoff_meetings
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT site_id
|
||||
FROM subjects
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
)
|
||||
UPDATE sites
|
||||
SET enrollment_target = 0
|
||||
WHERE id IN (SELECT site_id FROM candidate_sites WHERE site_id IS NOT NULL)
|
||||
AND enrollment_target IN (40, 50, 60, 70, 80)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"feasibility_codes": list(DEMO_FEASIBILITY_CODES),
|
||||
"ethics_codes": list(DEMO_ETHICS_CODES),
|
||||
"contract_codes": list(DEMO_FINANCE_CONTRACT_CODES),
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH candidate_studies AS (
|
||||
SELECT DISTINCT study_id FROM startup_feasibility WHERE project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM startup_ethics WHERE approval_no = ANY(:ethics_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM finance_contracts
|
||||
WHERE contract_no = ANY(:contract_codes) AND amount = 500000.0 AND currency = 'CNY'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM kickoff_meetings WHERE attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM subjects WHERE subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
)
|
||||
DELETE FROM subjects
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
"""
|
||||
),
|
||||
{
|
||||
"feasibility_codes": list(DEMO_FEASIBILITY_CODES),
|
||||
"ethics_codes": list(DEMO_ETHICS_CODES),
|
||||
"contract_codes": list(DEMO_FINANCE_CONTRACT_CODES),
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH candidate_studies AS (
|
||||
SELECT DISTINCT study_id FROM startup_feasibility WHERE project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM startup_ethics WHERE approval_no = ANY(:ethics_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM finance_contracts
|
||||
WHERE contract_no = ANY(:contract_codes) AND amount = 500000.0 AND currency = 'CNY'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM kickoff_meetings WHERE attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM subjects WHERE subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
)
|
||||
DELETE FROM kickoff_meetings
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND attendees::text = '["研究者", "CRA", "PM"]'
|
||||
"""
|
||||
),
|
||||
{
|
||||
"feasibility_codes": list(DEMO_FEASIBILITY_CODES),
|
||||
"ethics_codes": list(DEMO_ETHICS_CODES),
|
||||
"contract_codes": list(DEMO_FINANCE_CONTRACT_CODES),
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH candidate_studies AS (
|
||||
SELECT DISTINCT study_id FROM startup_feasibility WHERE project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM startup_ethics WHERE approval_no = ANY(:ethics_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM finance_contracts
|
||||
WHERE contract_no = ANY(:contract_codes) AND amount = 500000.0 AND currency = 'CNY'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM kickoff_meetings WHERE attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM subjects WHERE subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
)
|
||||
DELETE FROM finance_contracts
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND contract_no = ANY(:contract_codes)
|
||||
AND amount = 500000.0
|
||||
AND currency = 'CNY'
|
||||
"""
|
||||
),
|
||||
{
|
||||
"feasibility_codes": list(DEMO_FEASIBILITY_CODES),
|
||||
"ethics_codes": list(DEMO_ETHICS_CODES),
|
||||
"contract_codes": list(DEMO_FINANCE_CONTRACT_CODES),
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH candidate_studies AS (
|
||||
SELECT DISTINCT study_id FROM startup_feasibility WHERE project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM startup_ethics WHERE approval_no = ANY(:ethics_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM finance_contracts
|
||||
WHERE contract_no = ANY(:contract_codes) AND amount = 500000.0 AND currency = 'CNY'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM kickoff_meetings WHERE attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM subjects WHERE subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
)
|
||||
DELETE FROM startup_ethics
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND (
|
||||
approval_no = ANY(:ethics_codes)
|
||||
OR (
|
||||
approval_no IS NULL
|
||||
AND submit_date = DATE '2025-01-10'
|
||||
AND accept_date = DATE '2025-01-15'
|
||||
AND meeting_date IS NULL
|
||||
AND approved_date IS NULL
|
||||
)
|
||||
OR (
|
||||
approval_no IS NULL
|
||||
AND submit_date = DATE '2025-01-20'
|
||||
AND accept_date = DATE '2025-01-25'
|
||||
AND meeting_date IS NULL
|
||||
AND approved_date IS NULL
|
||||
)
|
||||
OR (
|
||||
approval_no IS NULL
|
||||
AND submit_date = DATE '2025-01-30'
|
||||
AND accept_date = DATE '2025-02-04'
|
||||
AND meeting_date IS NULL
|
||||
AND approved_date IS NULL
|
||||
)
|
||||
)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"feasibility_codes": list(DEMO_FEASIBILITY_CODES),
|
||||
"ethics_codes": list(DEMO_ETHICS_CODES),
|
||||
"contract_codes": list(DEMO_FINANCE_CONTRACT_CODES),
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH candidate_studies AS (
|
||||
SELECT DISTINCT study_id FROM startup_feasibility WHERE project_no = ANY(:feasibility_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM startup_ethics WHERE approval_no = ANY(:ethics_codes)
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM finance_contracts
|
||||
WHERE contract_no = ANY(:contract_codes) AND amount = 500000.0 AND currency = 'CNY'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM kickoff_meetings WHERE attendees::text = '["研究者", "CRA", "PM"]'
|
||||
UNION
|
||||
SELECT DISTINCT study_id FROM subjects WHERE subject_no ~ '^S0[1-3]-[0-9]{3}$'
|
||||
)
|
||||
DELETE FROM startup_feasibility
|
||||
WHERE study_id IN (SELECT study_id FROM candidate_studies)
|
||||
AND project_no = ANY(:feasibility_codes)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"feasibility_codes": list(DEMO_FEASIBILITY_CODES),
|
||||
"ethics_codes": list(DEMO_ETHICS_CODES),
|
||||
"contract_codes": list(DEMO_FINANCE_CONTRACT_CODES),
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH demo_study AS (
|
||||
SELECT id
|
||||
FROM studies
|
||||
WHERE code = :demo_study_code
|
||||
),
|
||||
demo_contract_fees AS (
|
||||
SELECT id
|
||||
FROM contract_fees
|
||||
WHERE project_id IN (SELECT id FROM demo_study)
|
||||
),
|
||||
demo_special_expenses AS (
|
||||
SELECT id
|
||||
FROM special_expenses
|
||||
WHERE project_id IN (SELECT id FROM demo_study)
|
||||
)
|
||||
DELETE FROM fee_attachments
|
||||
WHERE (
|
||||
entity_type = 'contract_fee'
|
||||
AND entity_id IN (SELECT id FROM demo_contract_fees)
|
||||
) OR (
|
||||
entity_type = 'special_expense'
|
||||
AND entity_id IN (SELECT id FROM demo_special_expenses)
|
||||
) OR (
|
||||
url IN (
|
||||
'https://example.com/contract-demo.pdf',
|
||||
'https://example.com/voucher-demo.pdf',
|
||||
'https://example.com/invoice-demo.pdf'
|
||||
)
|
||||
)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"demo_study_code": DEMO_STUDY_CODE,
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
WITH demo_study AS (
|
||||
SELECT id
|
||||
FROM studies
|
||||
WHERE code = :demo_study_code
|
||||
),
|
||||
demo_contract_fees AS (
|
||||
SELECT id
|
||||
FROM contract_fees
|
||||
WHERE project_id IN (SELECT id FROM demo_study)
|
||||
)
|
||||
DELETE FROM contract_fee_payments
|
||||
WHERE contract_fee_id IN (SELECT id FROM demo_contract_fees)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"demo_study_code": DEMO_STUDY_CODE,
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
DELETE FROM contract_fees
|
||||
WHERE project_id IN (
|
||||
SELECT id
|
||||
FROM studies
|
||||
WHERE code = :demo_study_code
|
||||
)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"demo_study_code": DEMO_STUDY_CODE,
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
DELETE FROM special_expenses
|
||||
WHERE project_id IN (
|
||||
SELECT id
|
||||
FROM studies
|
||||
WHERE code = :demo_study_code
|
||||
)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"demo_study_code": DEMO_STUDY_CODE,
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
DELETE FROM sites
|
||||
WHERE study_id IN (
|
||||
SELECT id
|
||||
FROM studies
|
||||
WHERE code = :demo_study_code
|
||||
)
|
||||
AND name LIKE '示例中心 %'
|
||||
"""
|
||||
),
|
||||
{
|
||||
"demo_study_code": DEMO_STUDY_CODE,
|
||||
},
|
||||
)
|
||||
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
DELETE FROM studies
|
||||
WHERE code = :demo_study_code
|
||||
"""
|
||||
),
|
||||
{
|
||||
"demo_study_code": DEMO_STUDY_CODE,
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
return
|
||||
Reference in New Issue
Block a user