def add_lesson():
form = request.form
lesson = Lesson(name=form['name'])
lesson.save()
company = Company.objects(id=form['company_id']).first()
company.insert_lesson(lesson)
company.save()
return redirect(url_for('.write_point', lesson_id=lesson.id))
def create():
# Check if user exists and signed in
jwt_user = get_jwt_identity()
user = User.get_or_none(User.name == jwt_user)
if not user:
return error_401("Unauthorized action")
# Retrieve data from json
title = request.form.get("title")
description = request.form.get("description")
if (str(request.form.get("teach")) == "true"):
teach = True
elif (str(request.form.get("teach")) == "false"):
teach = False
else:
return error_401("Invalid teach input")
skill = request.form.get("skill")
# Retrieve image from json
image_valid = False
if "image" in request.files:
image_file = request.files['image']
if image_file and allowed_file(image_file.filename):
image_file.filename = secure_filename(image_file.filename)
output = upload_file_to_s3(image_file)
image_valid = True
# Check title and description and then create new lesson
if title and description:
if image_valid:
lesson = Lesson(title=title, description=description, teach=teach, owner=user, skill=skill, image=image_file.filename)
else:
lesson = Lesson(title=title, description=description, teach=teach, owner=user, skill=skill)
if lesson.save():
lesson = {
'id': lesson.id,
'title': lesson.title,
'description': lesson.description,
'rating': lesson.rating,
'teach': lesson.teach,
'owner_id': lesson.owner_id,
'owner_name': lesson.owner.name,
'skill_id': lesson.skill_id,
'skill_name': lesson.skill.name,
'image_url': lesson.image_url
}
return success_201("New lesson created successfully", lesson)
else:
return error_401("Create lesson failed")
else:
return error_401("Invalid title or description")
