def validate_user(): stuid, name = auth() try: user = get_user(stuid) except: type = "admin" if stuid in options.admins else "student" user = User(stuid=stuid, name=name, type=type) session.add(user) session.commit() if user.type == "admin" and user.proxy: user = session.query(User).get(user.proxy) return user
def add_homework(): admin = validate_admin() name = request.form['name'] start_date = datetime.strptime(request.form['start_date'], "%m/%d/%Y %H:%M:%S") due_date = datetime.strptime(request.form['due_date'], "%m/%d/%Y %H:%M:%S") category_id = request.form['category_id'] homework = Homework(name=name, start_date=start_date, due_date=due_date, category_id=category_id) session.add(homework) session.commit() return "%s added successfully!" % name
def update_category(): admin = validate_admin() name = request.form['name'] weight = float(request.form['weight']) drops = int(request.form['drops']) try: category_id = int(request.form['id']) category = session.query(Category).get(category_id) category.name = name category.weight = weight category.drops = drops except: session.add(Category(name=name, weight=weight, drops=drops)) session.commit() return "Category %s successfully added/updated." % name