from marshmallow_enum import EnumField from marshmallow_sqlalchemy import SQLAlchemyAutoSchema from sqlalchemy import func from crc import db from crc.models.protocol_builder import ProtocolBuilderStatus class StudyModel(db.Model): __tablename__ = 'study' id = db.Column(db.Integer, primary_key=True) title = db.Column(db.String) last_updated = db.Column(db.DateTime(timezone=True), default=func.now()) protocol_builder_status = db.Column(db.Enum(ProtocolBuilderStatus)) primary_investigator_id = db.Column(db.String, nullable=True) sponsor = db.Column(db.String, nullable=True) hsr_number = db.Column(db.String, nullable=True) ind_number = db.Column(db.String, nullable=True) user_uid = db.Column(db.String, db.ForeignKey('user.uid'), nullable=False) investigator_uids = db.Column(db.ARRAY(db.String), nullable=True) inactive = db.Column(db.Boolean, default=False) requirements = db.Column(db.ARRAY(db.Integer), nullable=True) status_spec_id = db.Column(db.String, db.ForeignKey('workflow_spec.id')) status_spec_version = db.Column(db.String) class StudyModelSchema(SQLAlchemyAutoSchema): class Meta: model = StudyModel load_instance = True include_relationships = True include_fk = True # Includes foreign keys protocol_builder_status = EnumField(ProtocolBuilderStatus)