def add_work_image(work_id):
work = Work.query.get_or_404(work_id)
form = WorkImageForm()
if request.method == 'GET':
return render_template('work_image/add_work_image.html', work=work, form=form)
else:
if form.validate():
# Save image
image = request.files['image']
image_filename = str(uuid.uuid1()) + '.' + image.filename.split('.')[-1]
image.save(config.IMAGE_PATH + image_filename)
work_image = WorkImage(work_id=work_id, user_id=session['user_id'], url=config.IMAGE_URL+image_filename, filename=image_filename)
db.session.add(work_image)
db.session.commit()
return redirect(url_for('work_image', work_image_id=work_image.id))
else:
return render_template('work_image/add_work_image.html', work=work, form=form)
def edit_work_image(work_image_id):
work_image = WorkImage.query.get_or_404(work_image_id)
form = WorkImageForm()
if request.method == 'GET':
return render_template('work_image/edit_work_image.html', work_image=work_image, form=form)
else:
if form.validate():
# Delete old image
if os.path.isfile(config.IMAGE_PATH + work_image.filename):
os.remove(config.IMAGE_PATH + work_image.filename)
# Save new image
image = request.files['image']
image_filename = str(uuid.uuid1()) + '.' + image.filename.split('.')[-1]
image.save(config.IMAGE_PATH + image_filename)
# update image info
work_image.url = config.IMAGE_URL + image_filename
work_image.filename = image_filename
db.session.add(work_image)
db.session.commit()
return redirect(url_for('work_image', work_image_id=work_image_id))
else:
return render_template('work_image/edit_work_image.html', work_image=work_image, form=form)
