예제 #1
0
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()
예제 #2
0
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()
예제 #3
0
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
예제 #4
0
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()