def addAdmin(adminData): admin = models.Admin() for field, data in adminData.items(): admin.__setattr__(field, data) user = models.User(userId=admin.adminId, userType="admin") db.session.add(admin) db.session.add(user) db.session.commit()
def registerStudent(studentData): student = models.Student() for field, data in studentData.items(): student.__setattr__(field, data) user = models.User(userId=student.studentId, userType="student") db.session.add(student) db.session.add(user) db.session.commit()
def addLab(newData): lab = models.Lab() for field, data in newData.items(): lab.__setattr__(field, data) user = models.User(userId=newData['acronym'], userType="lab") db.session.add(user) db.session.add(lab) db.session.commit() return lab.id
def updateLab(id, newData): lab = getLabById(id) if "acronym" in newData: user = models.User.query.filter_by(userId=lab.acronym).first() if not user: user = models.User(userId=newData["acronym"], userType="lab") db.session.add(user) else: user.userId = newData["acronym"] for field, data in newData.items(): lab.__setattr__(field, data) db.session.commit()