model.py

from app import db
from datetime import datetime

class Team(db.Model):
    team_id = db.Column(db.Integer, primary_key=True)
    team_name = db.Column(db.String(120))
    jobs = db.relationship('Job', backref='team', lazy='dynamic')

    def __init__(self, team_name):
        self.team_name = team_name

    def __repr__(self):
        return '<Team %r>' % self.team_id


class Job(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    team_id = db.Column(db.Integer, db.ForeignKey('team.team_id'))
    patient_id = db.Column(db.String(120))
    urgency = db.Column(db.Integer)
    creator_comment = db.Column(db.String(120))
    doctor_comment = db.Column(db.String(120))
    bed = db.Column(db.String(120))
    ward = db.Column(db.String(120))
    location = db.Column(db.String(120))
    creator_id = db.Column(db.String(120))
    creator_name = db.Column(db.String(120))
    acknowledged = db.Column(db.DateTime)
    done = db.Column(db.DateTime)
    created_at = db.Column(db.DateTime)
    audits = db.relationship('Audit', backref='job', lazy='dynamic')

    def __init__(self, team_id, patient_id, urgency, creator_comment, doctor_comment, bed, ward, location, creator_name):
        self.team_id = team_id
        self.patient_id = patient_id
        self.urgency = urgency
        self.creator_comment = creator_comment
        self.doctor_comment = doctor_comment
        self.bed = bed
        self.ward = ward
        self.location = location
        self.creator_name = creator_name
        self.created_at = datetime.now()

    def __repr__(self):
        return '<Job %r>' % self.id


class Audit(db.Model):
    audit_id = db.Column(db.Integer, primary_key=True)
    job_id = db.Column(db.Integer, db.ForeignKey('job.id'))
    action = db.Column(db.String(120))
    old_val = db.Column(db.String(120))
    new_val = db.Column(db.String(120))
    audit_time = db.Column(db.DateTime)
    creator_id = db.Column(db.Integer)
    creator_name = db.Column(db.String(120))

    def __init__(self, job_id, action, old_val, new_val, creator_id, creator_name):
        self.job_id = job_id
        self.action = action
        self.old_val = old_val
        self.new_val = new_val
        self.creator_id = creator_id
        self.creator_name = creator_name

    def __repr__(self):
        return '<Audit %r>' % self.job_id


class Doctor(db.Model):
    doctor_id = db.Column(db.Integer, primary_key=True)
    doctor_name = db.Column(db.String(120))

    def __init__(self, doctor_name):
        self.doctor_name = doctor_name

    def __repr__(self):
        return '<Doctor %r>' % self.doctor_id