def add_lesson():
"""
Add a new lesson.
"""
# Create empty error object.
error = None
# Create form.
add_lesson_form = AddLessonForm()
# Add the rooms.
add_lesson_form.lesson_room_id.choices = [
(room.room_id, room.get_location()) for room in Room.query.all()
]
# Select all users.
all_users = select_users_by_roles(('STU', 'TUT', 'STA'))
# Update the form choices.
add_lesson_form.users.choices = [
(user.user_id, user.get_full_name() + " (" + user.get_role(pretty=True) + ")") for user in all_users
]
if request.method == 'POST' and add_lesson_form.validate_on_submit():
# Create a new lesson object.
new_lesson = Lesson()
# Create the datetime object.
lesson_datetime = datetime.combine(
add_lesson_form.lesson_date.data,
time(
add_lesson_form.lesson_hour.data,
add_lesson_form.lesson_minute.data
)
)
# Update the lesson details.
new_lesson.update_lesson_details(
lesson_datetime=lesson_datetime,
lesson_duration=add_lesson_form.lesson_duration.data*60,
lesson_notes=add_lesson_form.lesson_notes.data,
room_id=add_lesson_form.lesson_room_id.data
)
# Iterate through the users.
for user_id in add_lesson_form.users.data:
# Select the user object.
user_object = select_user(user_id)
# Append it to the lessons users.
new_lesson.users.append(user_object)
# Send an update.
if app.config['UPDATE_ON_NEW_LESSON']:
# Send an email update.
html = 'A new lesson has been created on: ' + new_lesson.get_lesson_date()
# Send a lesson update.
send_lesson_update(
user_object, html,
url_for('student.lessons', _external=True)
)
# Flash a success message.
flash("Successfully added new lesson.")
# Add the lesson to the db.
db.session.add(new_lesson)
# Commit changes.
db.session.commit()
return redirect(url_for('staff.add_lesson'))
return render_template(
'staff/add_lesson.html', add_lesson_form=add_lesson_form, error=error
)
