AdvLabDB/advlabdb/models.py

"""
See the file DB.drawio for the design of the database. It can be opened in the internet browser with the website:
https://app.diagrams.net

For more information about the implementation, see the part to Models in the documentation of Flask-SQLAlchemy:
https://flask-sqlalchemy.palletsprojects.com/en/2.x/models/
"""

# Imports
from flask import flash
from flask_security.models.fsqla_v2 import FsRoleMixin, FsUserMixin

# Importing the database instance
from advlabdb import db
from advlabdb.configUtils import getConfig


class Student(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    student_number = db.Column(db.Integer, nullable=False, unique=True)
    first_name = db.Column(db.String(100), nullable=False)
    last_name = db.Column(db.String(100), nullable=False)
    uni_email = db.Column(db.String(200), nullable=False, unique=True)
    contact_email = db.Column(db.String(200), unique=True)
    bachelor_thesis = db.Column(db.String, nullable=True)
    bachelor_thesis_work_group = db.Column(db.String, nullable=True)
    note = db.Column(db.Text, nullable=True)
    part_students = db.relationship("PartStudent", backref="student", lazy=True)

    def __repr__(self):
        return f"<Student: {self.first_name} {self.last_name}>"


class PartStudent(db.Model):
    # A student doing a specific part
    id = db.Column(db.Integer, primary_key=True)
    final_part_mark = db.Column(db.Integer, nullable=True)
    student_id = db.Column(db.Integer, db.ForeignKey("student.id"), nullable=False)
    part_id = db.Column(db.Integer, db.ForeignKey("part.id"), nullable=False)
    group_id = db.Column(db.Integer, db.ForeignKey("group.id"), nullable=True)
    experiment_marks = db.relationship("ExperimentMark", backref="part_student", lazy=True)


class Group(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    number = db.Column(db.Integer, nullable=False)
    part_id = db.Column(db.Integer, db.ForeignKey("part.id"), nullable=False)
    part_students = db.relationship("PartStudent", backref="group", lazy=True)
    group_experiments = db.relationship("GroupExperiment", backref="group", lazy=True)

    def __repr__(self):
        return f"<Group {self.number} {Part.query.get(self.part_id)}>"


class GroupExperiment(db.Model):
    # An experiment specified to a group
    id = db.Column(db.Integer, primary_key=True)
    part_experiment_id = db.Column(db.Integer, db.ForeignKey("part_experiment.id"), nullable=False)
    group_id = db.Column(db.Integer, db.ForeignKey("group.id"), nullable=False)
    appointments = db.relationship("Appointment", backref="group_experiment", lazy=True)
    experiment_marks = db.relationship("ExperimentMark", backref="group_experiment", lazy=True)


class Experiment(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    number = db.Column(db.Integer, nullable=False, unique=True)
    name = db.Column(db.String(200), nullable=False)
    description = db.Column(db.Text, nullable=True)
    room = db.Column(db.String(100), nullable=False)
    building = db.Column(db.String(100), nullable=False)
    responsibility = db.Column(db.String(200), nullable=True)
    duration_in_days = db.Column(db.Integer, nullable=False)
    deprecated = db.Column(db.Boolean, nullable=False, default=False)  # To not be deleted!
    oral_weighting = db.Column(db.Float, nullable=False)
    protocol_weighting = db.Column(db.Float, nullable=False)
    final_weighting = db.Column(db.Float, nullable=False)
    part_experiments = db.relationship("PartExperiment", backref="experiment", lazy=True)

    def __repr__(self):
        return f"<Exp {self.number}>"


# Helper table for the many to many relationship between Assistant and PartExperiment
experiment_assistant = db.Table(
    "experiment_assistant",
    db.Column("part_experiment_id", db.Integer, db.ForeignKey("part_experiment.id"), primary_key=True),
    db.Column("assistant_id", db.Integer, db.ForeignKey("assistant.id"), primary_key=True),
)


class PartExperiment(db.Model):
    # An experiment in a specific part
    id = db.Column(db.Integer, primary_key=True)
    experiment_id = db.Column(db.Integer, db.ForeignKey("experiment.id"), nullable=False)
    part_id = db.Column(db.Integer, db.ForeignKey("part.id"), nullable=False)
    assistants = db.relationship(
        "Assistant", secondary=experiment_assistant, lazy=True, backref=db.backref("part_experiments", lazy=True)
    )
    group_experiments = db.relationship("GroupExperiment", backref="part_experiment", lazy=True)


class Assistant(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    first_name = db.Column(db.String(100), nullable=False)
    last_name = db.Column(db.String(100), nullable=False)
    email = db.Column(db.String(200), nullable=False)
    phone_number = db.Column(db.String(50), nullable=True)
    mobile_phone_number = db.Column(db.String(50), nullable=True)
    room = db.Column(db.String(100), nullable=True)
    building = db.Column(db.String(100), nullable=True)
    user_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
    appointments = db.relationship("Appointment", backref="assistant", lazy=True)
    experiment_marks = db.relationship("ExperimentMark", backref="assistant", lazy=True)

    def __repr__(self):
        return f"<Assistant: {self.first_name} {self.last_name}>"


class Appointment(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    date = db.Column(db.DateTime, nullable=False)  # To be specified with the python package "datetime"
    special = db.Column(db.Boolean, nullable=False)  # In the break or not
    group_experiment_id = db.Column(db.Integer, db.ForeignKey("group_experiment.id"), nullable=False)
    assistant_id = db.Column(db.Integer, db.ForeignKey("assistant.id"), nullable=False)


class Part(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    label = db.Column(db.String(100), nullable=False)  # A/1, A/2, B/1, B/2
    semester_id = db.Column(db.Integer, db.ForeignKey("semester.id"), nullable=False)
    part_experiments = db.relationship("PartExperiment", backref="part", lazy=True)
    part_students = db.relationship("PartStudent", backref="part", lazy=True)
    groups = db.relationship("Group", backref="part", lazy=True)

    def __repr__(self):
        return f"<Part {self.label} {Semester.query.get(self.semester_id).label}>"


class Semester(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    label = db.Column(db.String(100), nullable=False, unique=True)  # WS2122 for example
    parts = db.relationship("Part", backref="semester", lazy=True)

    def __repr__(self):
        return f"<{self.label}>"

    def createParts(self):
        try:
            for partLabel in getConfig("partLabels"):
                db.session.add(Part(label=partLabel, semester=self))
            db.session.commit()
        except Exception as ex:
            flash(ex, "error")

            db.session.rollback()


class ExperimentMark(db.Model):
    # A mark for a student after a specific experiment
    id = db.Column(db.Integer, primary_key=True)
    oral_mark = db.Column(db.Integer, nullable=True)
    protocol_mark = db.Column(db.Integer, nullable=True)
    part_student_id = db.Column(db.Integer, db.ForeignKey("part_student.id"), nullable=False)
    group_experiment_id = db.Column(db.Integer, db.ForeignKey("group_experiment.id"), nullable=False)
    assistant_id = db.Column(
        db.Integer, db.ForeignKey("assistant.id"), nullable=True
    )  # The assistant who gives the mark


class User(db.Model, FsUserMixin):
    assistant = db.relationship("Assistant", backref="user", lazy=True, uselist=False)
    active_semester_id = db.Column(db.Integer, nullable=True)

    def __repr__(self):
        return f"<User {self.email}>"


class Role(db.Model, FsRoleMixin):
    def __repr__(self):
        return f"<{self.name}>"