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)