立项配置页初步优化

This commit is contained in:
Cheng Zhou
2026-02-24 16:53:22 +08:00
parent 0693f09b1d
commit 8f3f717e48
124 changed files with 12519 additions and 2954 deletions
@@ -0,0 +1,36 @@
"""add is_sae and is_susar flags to adverse events
Revision ID: 20260206_01
Revises: 20260116_08
Create Date: 2026-02-06 10:20:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = "20260206_01"
down_revision: Union[str, None] = "20260116_08"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.add_column(
"adverse_events",
sa.Column("is_sae", sa.Boolean(), nullable=False, server_default=sa.false()),
)
op.add_column(
"adverse_events",
sa.Column("is_susar", sa.Boolean(), nullable=False, server_default=sa.false()),
)
op.alter_column("adverse_events", "is_sae", server_default=None)
op.alter_column("adverse_events", "is_susar", server_default=None)
def downgrade() -> None:
op.drop_column("adverse_events", "is_susar")
op.drop_column("adverse_events", "is_sae")
@@ -0,0 +1,56 @@
"""add study_setup_configs table
Revision ID: 20260206_02
Revises: 20260206_01
Create Date: 2026-02-06 19:10:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = "20260206_02"
down_revision: Union[str, None] = "20260206_01"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
if "study_setup_configs" not in inspector.get_table_names():
op.create_table(
"study_setup_configs",
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("study_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("version", sa.Integer(), nullable=False, server_default="1"),
sa.Column("config", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
sa.Column("saved_by", postgresql.UUID(as_uuid=True), nullable=True),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
sa.ForeignKeyConstraint(["saved_by"], ["users.id"]),
sa.ForeignKeyConstraint(["study_id"], ["studies.id"]),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("study_id", name="uq_study_setup_config_study"),
)
indexes = {idx["name"] for idx in inspector.get_indexes("study_setup_configs")}
index_name = op.f("ix_study_setup_configs_study_id")
if index_name not in indexes:
op.create_index(index_name, "study_setup_configs", ["study_id"], unique=False)
uniques = {uq["name"] for uq in inspector.get_unique_constraints("study_setup_configs") if uq.get("name")}
if "uq_study_setup_config_study" not in uniques:
op.create_unique_constraint("uq_study_setup_config_study", "study_setup_configs", ["study_id"])
op.execute("ALTER TABLE study_setup_configs ALTER COLUMN version DROP DEFAULT")
def downgrade() -> None:
op.drop_index(op.f("ix_study_setup_configs_study_id"), table_name="study_setup_configs")
op.drop_table("study_setup_configs")
@@ -0,0 +1,66 @@
"""add setup config publish fields
Revision ID: 20260206_03
Revises: 20260206_02
Create Date: 2026-02-06 19:40:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = "20260206_03"
down_revision: Union[str, None] = "20260206_02"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
columns = {c["name"] for c in inspector.get_columns("study_setup_configs")}
if "publish_status" not in columns:
op.add_column(
"study_setup_configs",
sa.Column("publish_status", sa.String(length=20), nullable=False, server_default="DRAFT"),
)
op.alter_column("study_setup_configs", "publish_status", server_default=None)
if "published_config" not in columns:
op.add_column(
"study_setup_configs",
sa.Column("published_config", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
)
if "published_by" not in columns:
op.add_column(
"study_setup_configs",
sa.Column("published_by", postgresql.UUID(as_uuid=True), nullable=True),
)
op.create_foreign_key(
"study_setup_configs_published_by_fkey",
"study_setup_configs",
"users",
["published_by"],
["id"],
)
if "published_at" not in columns:
op.add_column(
"study_setup_configs",
sa.Column("published_at", sa.DateTime(timezone=True), nullable=True),
)
def downgrade() -> None:
op.drop_column("study_setup_configs", "published_at")
op.drop_constraint("study_setup_configs_published_by_fkey", "study_setup_configs", type_="foreignkey")
op.drop_column("study_setup_configs", "published_by")
op.drop_column("study_setup_configs", "published_config")
op.drop_column("study_setup_configs", "publish_status")
@@ -0,0 +1,75 @@
"""add study setup config versions table
Revision ID: 20260211_01
Revises: 20260206_03
Create Date: 2026-02-11 11:30:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = "20260211_01"
down_revision: Union[str, None] = "20260206_03"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
tables = set(inspector.get_table_names())
if "study_setup_config_versions" not in tables:
op.create_table(
"study_setup_config_versions",
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("study_setup_config_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("study_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("version", sa.Integer(), nullable=False),
sa.Column("source_version", sa.Integer(), nullable=True),
sa.Column("config", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
sa.Column("published_by", postgresql.UUID(as_uuid=True), nullable=True),
sa.Column("published_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
sa.ForeignKeyConstraint(["published_by"], ["users.id"]),
sa.ForeignKeyConstraint(["study_id"], ["studies.id"]),
sa.ForeignKeyConstraint(["study_setup_config_id"], ["study_setup_configs.id"]),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("study_id", "version", name="uq_setup_config_versions_study_version"),
)
existing_indexes = {idx["name"] for idx in inspector.get_indexes("study_setup_config_versions")}
if "ix_study_setup_config_versions_study_setup_config_id" not in existing_indexes:
op.create_index(
"ix_study_setup_config_versions_study_setup_config_id",
"study_setup_config_versions",
["study_setup_config_id"],
unique=False,
)
if "ix_study_setup_config_versions_study_id" not in existing_indexes:
op.create_index(
"ix_study_setup_config_versions_study_id",
"study_setup_config_versions",
["study_id"],
unique=False,
)
def downgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
tables = set(inspector.get_table_names())
if "study_setup_config_versions" not in tables:
return
indexes = {idx["name"] for idx in inspector.get_indexes("study_setup_config_versions")}
if "ix_study_setup_config_versions_study_setup_config_id" in indexes:
op.drop_index("ix_study_setup_config_versions_study_setup_config_id", table_name="study_setup_config_versions")
if "ix_study_setup_config_versions_study_id" in indexes:
op.drop_index("ix_study_setup_config_versions_study_id", table_name="study_setup_config_versions")
op.drop_table("study_setup_config_versions")
@@ -0,0 +1,88 @@
"""add extended profile fields to studies
Revision ID: 20260211_02
Revises: 20260211_01
Create Date: 2026-02-11 14:20:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = "20260211_02"
down_revision: Union[str, None] = "20260211_01"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
columns = {col["name"] for col in inspector.get_columns("studies")}
if "project_full_name" not in columns:
op.add_column("studies", sa.Column("project_full_name", sa.String(length=500), nullable=True))
if "lead_unit" not in columns:
op.add_column("studies", sa.Column("lead_unit", sa.String(length=255), nullable=True))
if "principal_investigator" not in columns:
op.add_column("studies", sa.Column("principal_investigator", sa.String(length=255), nullable=True))
if "main_pm" not in columns:
op.add_column("studies", sa.Column("main_pm", sa.String(length=255), nullable=True))
if "research_analysis" not in columns:
op.add_column("studies", sa.Column("research_analysis", sa.Text(), nullable=True))
if "research_product" not in columns:
op.add_column("studies", sa.Column("research_product", sa.String(length=255), nullable=True))
if "control_product" not in columns:
op.add_column("studies", sa.Column("control_product", sa.String(length=255), nullable=True))
if "indication" not in columns:
op.add_column("studies", sa.Column("indication", sa.String(length=255), nullable=True))
if "research_population" not in columns:
op.add_column("studies", sa.Column("research_population", sa.String(length=255), nullable=True))
if "research_design" not in columns:
op.add_column("studies", sa.Column("research_design", sa.Text(), nullable=True))
if "plan_start_date" not in columns:
op.add_column("studies", sa.Column("plan_start_date", sa.Date(), nullable=True))
if "plan_end_date" not in columns:
op.add_column("studies", sa.Column("plan_end_date", sa.Date(), nullable=True))
if "planned_site_count" not in columns:
op.add_column("studies", sa.Column("planned_site_count", sa.Integer(), nullable=True))
if "planned_enrollment_count" not in columns:
op.add_column("studies", sa.Column("planned_enrollment_count", sa.Integer(), nullable=True))
def downgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
columns = {col["name"] for col in inspector.get_columns("studies")}
if "planned_enrollment_count" in columns:
op.drop_column("studies", "planned_enrollment_count")
if "planned_site_count" in columns:
op.drop_column("studies", "planned_site_count")
if "plan_end_date" in columns:
op.drop_column("studies", "plan_end_date")
if "plan_start_date" in columns:
op.drop_column("studies", "plan_start_date")
if "research_design" in columns:
op.drop_column("studies", "research_design")
if "research_population" in columns:
op.drop_column("studies", "research_population")
if "indication" in columns:
op.drop_column("studies", "indication")
if "control_product" in columns:
op.drop_column("studies", "control_product")
if "research_product" in columns:
op.drop_column("studies", "research_product")
if "research_analysis" in columns:
op.drop_column("studies", "research_analysis")
if "main_pm" in columns:
op.drop_column("studies", "main_pm")
if "principal_investigator" in columns:
op.drop_column("studies", "principal_investigator")
if "lead_unit" in columns:
op.drop_column("studies", "lead_unit")
if "project_full_name" in columns:
op.drop_column("studies", "project_full_name")
@@ -0,0 +1,77 @@
"""add display version to setup config versions
Revision ID: 20260213_01
Revises: 20260211_02
Create Date: 2026-02-13 12:10:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = "20260213_01"
down_revision: Union[str, None] = "20260211_02"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
columns = {col["name"] for col in inspector.get_columns("study_setup_config_versions")}
if "display_version" not in columns:
op.add_column(
"study_setup_config_versions",
sa.Column("display_version", sa.Integer(), nullable=True),
)
op.execute(
sa.text(
"""
WITH ranked AS (
SELECT
id,
ROW_NUMBER() OVER (
PARTITION BY study_id
ORDER BY published_at ASC, version ASC
) AS rn
FROM study_setup_config_versions
)
UPDATE study_setup_config_versions v
SET display_version = ranked.rn
FROM ranked
WHERE v.id = ranked.id
AND v.display_version IS NULL
"""
)
)
op.alter_column("study_setup_config_versions", "display_version", nullable=False)
uniques = {item["name"] for item in inspector.get_unique_constraints("study_setup_config_versions")}
if "uq_setup_config_versions_study_display_version" not in uniques:
op.create_unique_constraint(
"uq_setup_config_versions_study_display_version",
"study_setup_config_versions",
["study_id", "display_version"],
)
def downgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
uniques = {item["name"] for item in inspector.get_unique_constraints("study_setup_config_versions")}
if "uq_setup_config_versions_study_display_version" in uniques:
op.drop_constraint(
"uq_setup_config_versions_study_display_version",
"study_setup_config_versions",
type_="unique",
)
columns = {col["name"] for col in inspector.get_columns("study_setup_config_versions")}
if "display_version" in columns:
op.drop_column("study_setup_config_versions", "display_version")
@@ -0,0 +1,64 @@
"""add setup projection fields to studies and sites
Revision ID: 20260213_02
Revises: 20260213_01
Create Date: 2026-02-13 16:30:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = "20260213_02"
down_revision: Union[str, None] = "20260213_01"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
study_columns = {col["name"] for col in inspector.get_columns("studies")}
if "enrollment_monthly_goal_note" not in study_columns:
op.add_column("studies", sa.Column("enrollment_monthly_goal_note", sa.Text(), nullable=True))
if "enrollment_stage_breakdown" not in study_columns:
op.add_column("studies", sa.Column("enrollment_stage_breakdown", sa.Text(), nullable=True))
if "summary_note" not in study_columns:
op.add_column("studies", sa.Column("summary_note", sa.Text(), nullable=True))
if "objective_note" not in study_columns:
op.add_column("studies", sa.Column("objective_note", sa.Text(), nullable=True))
site_columns = {col["name"] for col in inspector.get_columns("sites")}
if "enrollment_plan_start_date" not in site_columns:
op.add_column("sites", sa.Column("enrollment_plan_start_date", sa.Date(), nullable=True))
if "enrollment_plan_end_date" not in site_columns:
op.add_column("sites", sa.Column("enrollment_plan_end_date", sa.Date(), nullable=True))
if "enrollment_plan_note" not in site_columns:
op.add_column("sites", sa.Column("enrollment_plan_note", sa.Text(), nullable=True))
def downgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
site_columns = {col["name"] for col in inspector.get_columns("sites")}
if "enrollment_plan_note" in site_columns:
op.drop_column("sites", "enrollment_plan_note")
if "enrollment_plan_end_date" in site_columns:
op.drop_column("sites", "enrollment_plan_end_date")
if "enrollment_plan_start_date" in site_columns:
op.drop_column("sites", "enrollment_plan_start_date")
study_columns = {col["name"] for col in inspector.get_columns("studies")}
if "objective_note" in study_columns:
op.drop_column("studies", "objective_note")
if "summary_note" in study_columns:
op.drop_column("studies", "summary_note")
if "enrollment_stage_breakdown" in study_columns:
op.drop_column("studies", "enrollment_stage_breakdown")
if "enrollment_monthly_goal_note" in study_columns:
op.drop_column("studies", "enrollment_monthly_goal_note")
@@ -0,0 +1,88 @@
"""add setup projection entities for monitoring and center confirm
Revision ID: 20260213_03
Revises: 20260213_02
Create Date: 2026-02-13 11:20:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = "20260213_03"
down_revision: Union[str, None] = "20260213_02"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
milestone_columns = {col["name"] for col in inspector.get_columns("milestones")}
if "owner_name" not in milestone_columns:
op.add_column("milestones", sa.Column("owner_name", sa.String(length=255), nullable=True))
if "study_monitoring_strategies" not in inspector.get_table_names():
op.create_table(
"study_monitoring_strategies",
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("study_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("source_setup_item_id", sa.String(length=64), nullable=True),
sa.Column("strategy_type", sa.String(length=50), nullable=False),
sa.Column("detail", sa.Text(), nullable=False),
sa.Column("frequency", sa.String(length=50), nullable=False),
sa.Column("enabled", sa.Boolean(), nullable=False, server_default=sa.text("true")),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
sa.ForeignKeyConstraint(["study_id"], ["studies.id"]),
sa.PrimaryKeyConstraint("id"),
)
op.create_index(
"ix_study_monitoring_strategies_study_id",
"study_monitoring_strategies",
["study_id"],
unique=False,
)
if "study_center_confirms" not in inspector.get_table_names():
op.create_table(
"study_center_confirms",
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("study_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("site_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("source_setup_item_id", sa.String(length=64), nullable=True),
sa.Column("confirmer", sa.String(length=255), nullable=True),
sa.Column("confirm_status", sa.String(length=20), nullable=False),
sa.Column("confirm_date", sa.Date(), nullable=True),
sa.Column("note", sa.Text(), nullable=True),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
sa.ForeignKeyConstraint(["site_id"], ["sites.id"]),
sa.ForeignKeyConstraint(["study_id"], ["studies.id"]),
sa.PrimaryKeyConstraint("id"),
)
op.create_index("ix_study_center_confirms_site_id", "study_center_confirms", ["site_id"], unique=False)
op.create_index("ix_study_center_confirms_study_id", "study_center_confirms", ["study_id"], unique=False)
def downgrade() -> None:
bind = op.get_bind()
inspector = sa.inspect(bind)
if "study_center_confirms" in inspector.get_table_names():
op.drop_index("ix_study_center_confirms_study_id", table_name="study_center_confirms")
op.drop_index("ix_study_center_confirms_site_id", table_name="study_center_confirms")
op.drop_table("study_center_confirms")
if "study_monitoring_strategies" in inspector.get_table_names():
op.drop_index("ix_study_monitoring_strategies_study_id", table_name="study_monitoring_strategies")
op.drop_table("study_monitoring_strategies")
milestone_columns = {col["name"] for col in inspector.get_columns("milestones")}
if "owner_name" in milestone_columns:
op.drop_column("milestones", "owner_name")