uva-covid19-testing-communicator/communicator/models/sample.py

from datetime import datetime

import marshmallow
from marshmallow import EXCLUDE
from marshmallow_sqlalchemy import SQLAlchemyAutoSchema

from communicator import db
from communicator.models.notification import Notification


class Sample(db.Model):
    barcode = db.Column(db.String, primary_key=True)
    student_id = db.Column(db.Integer)
    computing_id = db.Column(db.String)
    date = db.Column(db.DateTime)
    created_on = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
    last_modified = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.now, nullable=False)
    location = db.Column(db.Integer)
    station = db.Column(db.Integer)
    phone = db.Column(db.String)
    email = db.Column(db.String)
    result_code = db.Column(db.String)
    ivy_file = db.Column(db.String)
    in_firebase = db.Column(db.Boolean, default=False)  # Does this record exist in Firebase?
    in_ivy = db.Column(db.Boolean, default=False)  # Has this record come in from the IVY?
    email_notified = db.Column(db.Boolean, default=False)
    text_notified = db.Column(db.Boolean, default=False)
    notifications = db.relationship(Notification, back_populates="sample",
                                    cascade="all, delete, delete-orphan",
                                    order_by=Notification.date.desc)

    def last_failure_by_type(self, notification_type):
        notifications = list(filter(lambda x: x.type == notification_type, self.notifications))
        if len(notifications) == 0:
            return  # No notifications yet.
        elif notifications[0].successful:
            return  # Last notification was successful.
        else:
            return notifications[0]

    def merge(self, sample):
        if sample.computing_id:
            self.computing_id = sample.computing_id
        if sample.phone:
            self.phone = sample.phone
        if sample.email:
            self.email = sample.email
        if sample.result_code:
            self.result_code = sample.result_code
        if sample.in_firebase:
            self.in_firebase = True
        if sample.in_ivy:
            self.in_ivy = True
        if sample.ivy_file:
            self.ivy_file = sample.ivy_file


class NotificationSchema(SQLAlchemyAutoSchema):
    class Meta:
        model = Notification
        load_instance = True
        include_relationships = True
        include_fk = True  # Includes foreign keys
        unknown = EXCLUDE

class SampleSchema(SQLAlchemyAutoSchema):
    class Meta:
        model = Sample
        load_instance = True
        include_relationships = True
        include_fk = True  # Includes foreign keys
        unknown = EXCLUDE
        exclude = ["in_firebase", "in_ivy"]

    notifications = marshmallow.fields.List(marshmallow.fields.Nested(NotificationSchema, dump_only=True,
                                                                      exclude=['sample', 'sample_barcode']))