nci-cidc-api-modules 1.2.54__py3-none-any.whl → 1.2.56__py3-none-any.whl

cidc_api/models/db/stage2/medical_history_orm.py

@@ -2,10 +2,11 @@ from __future__ import annotations
 from typing import List
 
 from pydantic import NonNegativeInt, PositiveFloat
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import TobaccoSmokingStatus
 
 
@@ -13,6 +14,12 @@ class MedicalHistoryORM(BaseORM):
     __tablename__ = "medical_history"
     __repr_attrs__ = ["medical_history_id"]
     __data_category__ = "medical_history"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     medical_history_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/other_clinical_endpoint_orm.py

@@ -1,14 +1,22 @@
 from __future__ import annotations
-from sqlalchemy import ForeignKey, String
+from sqlalchemy import String, ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import YNU, ResponseSystem, ResponseSystemVersion
 
 
 class OtherClinicalEndpointORM(BaseORM):
     __tablename__ = "other_clinical_endpoint"
     __repr_attrs__ = ["other_clinical_endpoint_id", "name", "event"]
+    __data_category__ = "other_clinical_endpoint"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     other_clinical_endpoint_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/other_malignancy_orm.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 from pydantic import NonPositiveInt
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import UberonAnatomicalTerm, ICDO3MorphologicalCode, ICDO3MorphologicalTerm, MalignancyStatus
 
 
@@ -11,6 +12,12 @@ class OtherMalignancyORM(BaseORM):
     __tablename__ = "other_malignancy"
     __repr_attrs__ = ["other_malignancy_id", "primary_disease_site"]
     __data_category__ = "other_malignancy"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     other_malignancy_id: Mapped[int] = mapped_column(primary_key=True)
     medical_history_id: Mapped[int] = mapped_column(

cidc_api/models/db/stage2/participant_orm.py

@@ -1,26 +1,26 @@
 from __future__ import annotations
 from typing import List
+
 from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import OffStudyReason, YNU
 
 
 class ParticipantORM(BaseORM):
     __tablename__ = "participant"
-    __repr_attrs__ = ["native_participant_id", "cimac_participant_id"]
+    __repr_attrs__ = ["participant_id", "native_participant_id", "cimac_participant_id"]
+    __data_category__ = "participant"
     __table_args__ = (
-        ForeignKeyConstraint(
-            ["trial_id", "version"], ["stage2.trial.trial_id", "stage2.trial.version"], ondelete="CASCADE"
-        ),
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
     )
-    __data_category__ = "participant"
 
-    participant_id: Mapped[int] = mapped_column(primary_key=True)
     trial_id: Mapped[str]
     version: Mapped[str]
 
+    participant_id: Mapped[int] = mapped_column(primary_key=True)
     native_participant_id: Mapped[str | None]
     cimac_participant_id: Mapped[str | None]
     consent_group_id: Mapped[int | None] = mapped_column(

cidc_api/models/db/stage2/prior_treatment_orm.py

@@ -2,11 +2,11 @@ from __future__ import annotations
 from typing import List
 
 from pydantic import NonPositiveInt, NegativeInt
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
-from sqlalchemy.types import JSON
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import ConditioningRegimenType, StemCellDonorType
 
 
@@ -14,6 +14,12 @@ class PriorTreatmentORM(BaseORM):
     __tablename__ = "prior_treatment"
     __repr_attrs__ = ["prior_treatment_id", "type"]
     __data_category__ = "prior_treatment"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     prior_treatment_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))
@@ -24,6 +30,6 @@ class PriorTreatmentORM(BaseORM):
     prior_treatment_best_response: Mapped[str | None]
     prior_treatment_conditioning_regimen_type: Mapped[ConditioningRegimenType | None]
     prior_treatment_stem_cell_donor_type: Mapped[StemCellDonorType | None]
-    prior_treatment_days_to_prior_transplant: Mapped[NegativeInt | None]
+    prior_treatment_days_from_transplant_to_treatment_initiation: Mapped[NegativeInt | None]
 
     participant: Mapped[ParticipantORM] = relationship(back_populates="prior_treatments", cascade="all, delete")

cidc_api/models/db/stage2/publication_orm.py

@@ -3,22 +3,23 @@ from sqlalchemy import ForeignKeyConstraint, ForeignKey
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import ConditioningRegimenType, StemCellDonorType
 
 
 class PublicationORM(BaseORM):
     __tablename__ = "publication"
     __repr_attrs__ = ["publication_id", "publication_title"]
+    __data_category__ = "publication"
     __table_args__ = (
-        ForeignKeyConstraint(
-            ["trial_id", "version"], ["stage2.trial.trial_id", "stage2.trial.version"], ondelete="CASCADE"
-        ),
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
     )
 
-    publication_id: Mapped[int] = mapped_column(primary_key=True)
     trial_id: Mapped[str]
     version: Mapped[str]
 
+    publication_id: Mapped[int] = mapped_column(primary_key=True)
+
     digital_object_id: Mapped[str]
     pubmed_id: Mapped[str | None]
     publication_title: Mapped[str | None]

cidc_api/models/db/stage2/radiotherapy_dose_orm.py

@@ -1,17 +1,18 @@
 from __future__ import annotations
 from pydantic import NonNegativeInt, NonNegativeFloat, PositiveFloat
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import (
     RadiotherapyProcedure,
     UberonAnatomicalTerm,
     RadiotherapyDoseUnits,
+    RadiationExtent,
     YN,
     YNU,
-    RadiationExtent,
 )
 
 
@@ -19,6 +20,12 @@ class RadiotherapyDoseORM(BaseORM):
     __tablename__ = "radiotherapy_dose"
     __repr_attrs__ = ["radiotherapy_dose_id", "procedure"]
     __data_category__ = "radiotherapy_dose"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     radiotherapy_dose_id: Mapped[int] = mapped_column(primary_key=True)
     treatment_id: Mapped[int] = mapped_column(ForeignKey("stage2.treatment.treatment_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/response_by_system_orm.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 from pydantic import PositiveInt
-from sqlalchemy import ForeignKey, String
+from sqlalchemy import String, ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import ResponseSystem, ResponseSystemVersion, BestOverallResponse, YNUNA, YN
 
 
@@ -11,6 +12,12 @@ class ResponseBySystemORM(BaseORM):
     __tablename__ = "response_by_system"
     __repr_attrs__ = ["response_by_system_id", "participant_id"]
     __data_category__ = "response_by_system"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     response_by_system_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/response_orm.py

@@ -1,17 +1,24 @@
 from __future__ import annotations
 from pydantic import NonNegativeInt
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
-from cidc_api.models.types import SurvivalStatus, YNUNA, YN, CauseOfDeath
+from cidc_api.models.db.stage2.trial_orm import TrialORM
+from cidc_api.models.types import SurvivalStatus, CauseOfDeath, YNUNA, YN
 
 
 class ResponseORM(BaseORM):
     __tablename__ = "response"
     __repr_attrs__ = ["response_id", "participant_id"]
     __data_category__ = "response"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     response_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/shipment_orm.py

@@ -6,6 +6,7 @@ from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import AssayPriority, AssayType, Courier, ShipmentCondition, ShipmentQuality
 
 
@@ -13,16 +14,15 @@ class ShipmentORM(BaseORM):
     __tablename__ = "shipment"
     __repr_attrs__ = ["shipment_id", "institution_id", "trial_id"]
     __table_args__ = (
-        ForeignKeyConstraint(
-            ["trial_id", "version"], ["stage2.trial.trial_id", "stage2.trial.version"], ondelete="CASCADE"
-        ),
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
     )
 
-    shipment_id: Mapped[int] = mapped_column(primary_key=True)
-    institution_id: Mapped[int] = mapped_column(ForeignKey("stage2.institution.institution_id", ondelete="CASCADE"))
     trial_id: Mapped[str]
     version: Mapped[str]
 
+    shipment_id: Mapped[int] = mapped_column(primary_key=True)
+    institution_id: Mapped[int] = mapped_column(ForeignKey("stage2.institution.institution_id", ondelete="CASCADE"))
+
     manifest_id: Mapped[str]
     assay_priority: Mapped[AssayPriority | None]
     assay_type: Mapped[AssayType | None]

cidc_api/models/db/stage2/shipment_specimen_orm.py

@@ -1,15 +1,21 @@
 from __future__ import annotations
-from datetime import datetime
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 
 
 class ShipmentSpecimenORM(BaseORM):
     __tablename__ = "shipment_specimen"
     __repr_attrs__ = ["specimen_id", "shipment_id"]
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     specimen_id: Mapped[int] = mapped_column(
         ForeignKey("stage2.specimen.specimen_id", ondelete="CASCADE"), primary_key=True

cidc_api/models/db/stage2/specimen_orm.py

@@ -1,97 +1,31 @@
 from __future__ import annotations
-from datetime import datetime
-from typing import List
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import (
     UberonAnatomicalTerm,
-    ICDO3MorphologicalCode,
-    SpecimenType,
-    SpecimenDescription,
-    TumorType,
-    CollectionProcedure,
-    FixationStabilizationType,
-    PrimaryContainerType,
-    VolumeUnits,
-    ProcessedType,
-    ConcentrationUnits,
-    DerivativeType,
-    PBMCRestingPeriodUsed,
-    MaterialUnits,
-    MaterialStorageCondition,
-    QCCondition,
-    ReplacementRequested,
-    ResidualUse,
-    DiagnosisVerification,
-    AssayType,
 )
 
 
 class SpecimenORM(BaseORM):
     __tablename__ = "specimen"
     __repr_attrs__ = ["specimen_id", "participant_id", "cimac_id"]
+    __data_category__ = "specimen"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     specimen_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))
 
     cimac_id: Mapped[str]
-    surgical_pathology_report_id: Mapped[str | None]
-    clinical_report_id: Mapped[str | None]
-    parent_specimen_id: Mapped[str | None]
-    processed_specimen_id: Mapped[str | None]
-    organ_site_of_collection: Mapped[UberonAnatomicalTerm | None]
-    histology_behavior: Mapped[ICDO3MorphologicalCode | None]
-    histology_behavior_description: Mapped[str | None]
     collection_event_name: Mapped[str]
-    specimen_type: Mapped[SpecimenType | None]
-    specimen_type_other: Mapped[str | None]
-    specimen_description: Mapped[SpecimenDescription | None]
-    tumor_type: Mapped[TumorType | None]
-    collection_procedure: Mapped[CollectionProcedure | None]
-    collection_procedure_other: Mapped[str | None]
-    core_number: Mapped[str | None]
-    fixation_stabilization_type: Mapped[FixationStabilizationType | None]
-    primary_container_type: Mapped[PrimaryContainerType | None]
-    primary_container_type_other: Mapped[str | None]
-    volume: Mapped[float | None]
-    volume_units: Mapped[VolumeUnits | None]
-    processed_type: Mapped[ProcessedType | None]
-    processed_volume: Mapped[float | None]
-    processed_volume_units: Mapped[VolumeUnits | None]
-    processed_concentration: Mapped[float | None]
-    processed_concentration_units: Mapped[ConcentrationUnits | None]
-    processed_quantity: Mapped[float | None]
-    derivative_type: Mapped[DerivativeType | None]
-    derivative_volume: Mapped[float | None]
-    derivative_volume_units: Mapped[VolumeUnits | None]
-    derivative_concentration: Mapped[float | None]
-    derivative_concentration_units: Mapped[ConcentrationUnits | None]
-    tumor_tissue_total_area_percentage: Mapped[float | None]
-    viable_tumor_area_percentage: Mapped[float | None]
-    viable_stroma_area_percentage: Mapped[float | None]
-    necrosis_area_percentage: Mapped[float | None]
-    fibrosis_area_percentage: Mapped[float | None]
-    din: Mapped[float | None]
-    a260_a280: Mapped[float | None]
-    a260_a230: Mapped[float | None]
-    pbmc_viability: Mapped[float | None]
-    pbmc_recovery: Mapped[float | None]
-    pbmc_resting_period_used: Mapped[PBMCRestingPeriodUsed | None]
-    material_used: Mapped[float | None]
-    material_used_units: Mapped[MaterialUnits | None]
-    material_remaining: Mapped[float | None]
-    material_remaining_units: Mapped[MaterialUnits | None]
-    material_storage_condition: Mapped[MaterialStorageCondition | None]
-    qc_condition: Mapped[QCCondition | None]
-    replacement_requested: Mapped[ReplacementRequested | None]
-    residual_use: Mapped[ResidualUse | None]
-    comments: Mapped[str | None]
-    diagnosis_verification: Mapped[DiagnosisVerification | None]
-    intended_assay: Mapped[AssayType | None]
-    date_ingested: Mapped[datetime | None]
     days_to_specimen_collection: Mapped[int]
     organ_site_of_collection: Mapped[UberonAnatomicalTerm]
 

cidc_api/models/db/stage2/stem_cell_transplant_orm.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 from pydantic import NonNegativeInt
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import StemCellDonorType, AllogeneicDonorType, StemCellSource, ConditioningRegimenType
 
 
@@ -12,6 +13,12 @@ class StemCellTransplantORM(BaseORM):
     __tablename__ = "stem_cell_transplant"
     __repr_attrs__ = ["stem_cell_transplant_id"]
     __data_category__ = "stem_cell_transplant"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     stem_cell_transplant_id: Mapped[int] = mapped_column(primary_key=True)
     treatment_id: Mapped[int] = mapped_column(ForeignKey("stage2.treatment.treatment_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/surgery_orm.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 from pydantic import NonNegativeInt
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import SurgicalProcedure, UberonAnatomicalTerm, YNU
 
 
@@ -12,6 +13,12 @@ class SurgeryORM(BaseORM):
     __tablename__ = "surgery"
     __repr_attrs__ = ["surgery_id", "procedure"]
     __data_category__ = "surgery"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     surgery_id: Mapped[int] = mapped_column(primary_key=True)
     treatment_id: Mapped[int] = mapped_column(ForeignKey("stage2.treatment.treatment_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/therapy_agent_dose_orm.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 from pydantic import NonNegativeInt, NonNegativeFloat, PositiveFloat
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import TherapyAgentDoseUnits, YNU
 
 
@@ -12,6 +13,12 @@ class TherapyAgentDoseORM(BaseORM):
     __tablename__ = "therapy_agent_dose"
     __repr_attrs__ = ["therapy_agent_dose_id", "therapy_agent_name"]
     __data_category__ = "therapy_agent_dose"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     therapy_agent_dose_id: Mapped[int] = mapped_column(primary_key=True)
     treatment_id: Mapped[int] = mapped_column(ForeignKey("stage2.treatment.treatment_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/treatment_orm.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 from typing import List
 
-from sqlalchemy import ForeignKey
+from sqlalchemy import ForeignKey, ForeignKeyConstraint
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from cidc_api.models.db.stage2.base_orm import BaseORM
+from cidc_api.models.db.stage2.trial_orm import TrialORM
 from cidc_api.models.types import YNU, OffTreatmentReason
 
 
@@ -12,6 +13,12 @@ class TreatmentORM(BaseORM):
     __tablename__ = "treatment"
     __repr_attrs__ = ["treatment_id", "participant_id", "treatment_description"]
     __data_category__ = "treatment"
+    __table_args__ = (
+        ForeignKeyConstraint(["trial_id", "version"], [TrialORM.trial_id, TrialORM.version], ondelete="CASCADE"),
+    )
+
+    trial_id: Mapped[str]
+    version: Mapped[str]
 
     treatment_id: Mapped[int] = mapped_column(primary_key=True)
     participant_id: Mapped[int] = mapped_column(ForeignKey("stage2.participant.participant_id", ondelete="CASCADE"))

cidc_api/models/db/stage2/trial_orm.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
-from datetime import datetime
 from typing import List
+from datetime import datetime
 
-from sqlalchemy import ForeignKey
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 from sqlalchemy.types import JSON
 

cidc_api/models/pydantic/stage1/response_by_system.py

@@ -176,6 +176,19 @@ class ResponseBySystem(Base):
                     loc="days_to_best_response",
                 )
 
+    @forced_validator
+    @classmethod
+    def validate_progression_free_survival_disease_progression_chronology(cls, data, info) -> None:
+        progression = data.get("progression", None)
+        days_to_disease_progression = data.get("days_to_disease_progression", None)
+        progression_free_survival = data.get("progression_free_survival", None)
+
+        if progression == "Yes" and progression_free_survival != days_to_disease_progression:
+            raise ValueLocError(
+                'Violate "progression_free_survival" = "days_to_disease_progression" when progression is PFS event',
+                loc="progression_free_survival",
+            )
+
     @model_validator(mode="after")
     def validate_days_to_last_vital_status_chronology(self) -> Self:
         if not self.response:
@@ -218,3 +231,52 @@ class ResponseBySystem(Base):
                 loc="days_to_death,days_to_first_response,days_to_best_response,days_to_disease_progression",
             )
         return self
+
+    @model_validator(mode="after")
+    def validate_progression_free_survival_death_chronology(self) -> Self:
+        if not self.response:
+            return self
+
+        if (
+            self.progression_free_survival_event == "Yes"
+            and self.progression == "No"
+            and self.progression_free_survival != self.response.days_to_death
+        ):
+            raise ValueLocError(
+                'Violate "progression_free_survival" = "days_to_death" when death is PFS event',
+                loc="progression_free_survival",
+            )
+        return self
+
+    @model_validator(mode="after")
+    def validate_progression_free_survival_event_no_or_unknown(self) -> Self:
+        if not self.response:
+            return self
+
+        pfs_event = self.progression_free_survival_event
+        if pfs_event in ["No", "Unknown"] and (self.days_to_disease_progression or self.response.days_to_death):
+            raise ValueLocError(
+                'Violate "days_to_death" and "days_to_disease_progression" should be blank when PFS event is "No" or "Unknown"',
+                loc="progression_free_survival_event",
+            )
+        return self
+
+    @model_validator(mode="after")
+    def validate_progression_free_survival_event_consistent(self) -> Self:
+        if not self.response:
+            return self
+
+        progressed_or_dead = self.progression == "Yes" or self.response.survival_status == "Dead"
+        no_progression_and_alive = self.progression == "No" and self.response.survival_status == "Alive"
+        pfs_event = self.progression_free_survival_event
+        if (
+            (pfs_event == "Yes" and not progressed_or_dead)
+            or (progressed_or_dead and pfs_event != "Yes")
+            or (pfs_event == "No" and not no_progression_and_alive)
+            or (no_progression_and_alive and pfs_event != "No")
+        ):
+            raise ValueLocError(
+                'Inconsistent flags for "progression_free_survival_event","progression" and "survival_status"',
+                loc="progression_free_survival_event",
+            )
+        return self

cidc_api/models/pydantic/stage1/trial.py

@@ -3,7 +3,7 @@ from pydantic import BeforeValidator
 from typing import List, Annotated
 
 from cidc_api.models.pydantic.base import Base
-from cidc_api.models.types import PrimaryPurposeType, AgeGroup
+from cidc_api.models.types import TrialOrganization, TrialFundingAgency, AssayType, AgeGroup, PrimaryPurposeType
 
 
 class Trial(Base):