1
0
Fork 0
mirror of https://codeberg.org/Mo8it/AdvLabDB.git synced 2024-09-19 18:31:16 +00:00
AdvLabDB/testDB.py

97 lines
3.4 KiB
Python

from datetime import datetime
from flask_security import hash_password
from advlabdb import app, db, user_datastore
from advlabdb.models import *
with app.app_context():
db.drop_all()
db.create_all()
sem1 = Semester(label="WS2021")
sem2 = Semester(label="SS21")
db.session.add(sem1)
db.session.add(sem2)
sem1.createParts()
sem2.createParts()
parta1 = sem2.parts[0]
partb2 = sem2.parts[1]
student1 = Student(student_number=123, first_name="Mo", last_name="Bit", uni_email="m@test.com")
student2 = Student(student_number=1232, first_name="Mo2", last_name="Bit", uni_email="m2@test.com")
student3 = Student(student_number=1233, first_name="Mo3", last_name="Bit3", uni_email="m3@test.com")
db.session.add(student1)
db.session.add(student2)
db.session.add(student3)
g1 = Group(number=1, part=parta1)
g2 = Group(number=2, part=partb2)
db.session.add(g1)
db.session.add(g2)
ps1 = PartStudent(student=student1, part=parta1, group=g1)
ps2 = PartStudent(student=student2, part=parta1, group=g1)
ps3 = PartStudent(student=student3, part=partb2, group=g2)
db.session.add(ps1)
db.session.add(ps2)
db.session.add(ps3)
ex1 = Experiment(number=1, name="exp", room="123", building="phy", responsibility="none", duration_in_days=2,
oral_weighting=0.5, protocol_weighting=0.5, final_weighting=1)
ex2 = Experiment(number=2, name="exp2", room="123", building="phy", responsibility="none", duration_in_days=2,
oral_weighting=0.5, protocol_weighting=0.5, final_weighting=1)
db.session.add(ex1)
db.session.add(ex2)
px1 = PartExperiment(experiment=ex1, part=parta1)
px2 = PartExperiment(experiment=ex2, part=partb2)
db.session.add(px1)
db.session.add(px2)
gx1 = GroupExperiment(part_experiment=px1, group=g1)
gx2 = GroupExperiment(part_experiment=px2, group=g2)
db.session.add(gx1)
db.session.add(gx2)
user_datastore.create_role(name="admin")
user_datastore.create_role(name="assistant")
user_datastore.create_user(email="admin@protonmail.com", password=hash_password("admin"), roles=["admin"])
us1 = user_datastore.create_user(email="test@protonmail.com", password=hash_password("h1"), roles=["assistant"])
us2 = user_datastore.create_user(email="test2@protonmail.com", password=hash_password("h2"), roles=["assistant"])
as1 = Assistant(first_name="As1", last_name="l", email="test@test.com", phone_number="012333212",
mobile_phone_number="012334123", user=us1)
as2 = Assistant(first_name="As2", last_name="l", email="test2@test.com", user=us2)
as1.part_experiments.append(px1)
as2.part_experiments.append(px2)
db.session.add(as1)
db.session.add(as2)
ap1 = Appointment(date=datetime(2021, 3, 21), special=True, group_experiment=gx1, assistant=as1)
ap2 = Appointment(date=datetime(2021, 3, 22), special=True, group_experiment=gx2, assistant=as2)
db.session.add(ap1)
db.session.add(ap2)
em1 = ExperimentMark(oral_mark=11, protocol_mark=14, part_student=ps1, group_experiment=gx1, assistant=as1)
em2 = ExperimentMark(oral_mark=12, protocol_mark=14, part_student=ps2, group_experiment=gx1, assistant=as1)
em3 = ExperimentMark(oral_mark=13, protocol_mark=14, part_student=ps3, group_experiment=gx2, assistant=as2)
db.session.add(em1)
db.session.add(em2)
db.session.add(em3)
db.session.commit()