53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from marshmallow import fields
|
|
from sqlalchemy import func
|
|
|
|
from crc import db, ma
|
|
from crc.models.study import StudyModel
|
|
|
|
|
|
class EmailModel(db.Model):
|
|
__tablename__ = 'email'
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
subject = db.Column(db.String)
|
|
sender = db.Column(db.String)
|
|
recipients = db.Column(db.String)
|
|
cc = db.Column(db.String, nullable=True)
|
|
bcc = db.Column(db.String, nullable=True)
|
|
content = db.Column(db.String)
|
|
content_html = db.Column(db.String)
|
|
study_id = db.Column(db.Integer, db.ForeignKey(StudyModel.id), nullable=True)
|
|
timestamp = db.Column(db.DateTime(timezone=True), default=func.now())
|
|
workflow_spec_id = db.Column(db.String, nullable=True)
|
|
study = db.relationship(StudyModel)
|
|
name = db.Column(db.String)
|
|
doc_codes = db.relationship("EmailDocCodesModel")
|
|
|
|
|
|
class EmailDocCodesModel(db.Model):
|
|
__tablename__ = 'email_doc_codes'
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
email_id = db.Column(db.Integer, db.ForeignKey(EmailModel.id))
|
|
doc_code = db.Column(db.String)
|
|
|
|
|
|
class EmailDocCodesSchema(ma.Schema):
|
|
class Meta:
|
|
model = EmailDocCodesModel
|
|
fields = ["doc_code"]
|
|
|
|
|
|
class EmailModelSchema(ma.Schema):
|
|
|
|
class Meta:
|
|
model = EmailModel
|
|
fields = ["id", "subject", "sender", "recipients", "cc", "bcc", "content",
|
|
"study_id", "timestamp", "workflow_spec_id", "name", "doc_codes"]
|
|
doc_codes = fields.Method('get_doc_codes', dump_only=True)
|
|
|
|
@staticmethod
|
|
def get_doc_codes(email):
|
|
doc_codes = []
|
|
for doc_code in email.doc_codes:
|
|
doc_codes.append(doc_code.doc_code)
|
|
return doc_codes
|