def handle_request(): if request.method == 'POST': if request.files: username = '******' filename = images.save(request.files['image']) url = images.url(filename) #--------------------- result = predict(filename) #--------------------- user = User.query.filter_by(username=username).first() if user is None: user = User(username=username) report = Report(user=user, data=json.dumps(result)) image = Image(report=report) image.image_filename = filename image.image_url = url db.session.add(user) db.session.add(report) db.session.add(image) db.session.commit() return 'Report Generated' return 'No Files Recieved' return 'Method not POST'
def upload(): form = UploadForm() if request.method == 'POST': if form.validate_on_submit(): username = form.username.data filename = images.save(request.files['image']) url = images.url(filename) #--------------------- result = predict(filename) #--------------------- user = User.query.filter_by(username=username).first() if user is None: user = User(username=username) report = Report(user=user, data=json.dumps(result)) image = Image(report=report) image.image_filename = filename image.image_url = url db.session.add(user) db.session.add(report) db.session.add(image) db.session.commit() flash( 'User Report for username: {} generated. '.format( form.username.data), 'success') return redirect(url_for('report', report_id=report.id)) else: flash_errors(form) flash('ERROR! Report was not generated.', 'error') return render_template('upload.html', title='Upload', form=form)
def upload_images(): images = {} idx = 0 if len(request.files) > 0: file_list = request.files.getlist('images') for file in file_list: if not check_file(file.filename): return {'errors': ['Unaccepted file extension, must be JPG, PNG, or JPEG']}, 400 for file in file_list: image = Image() image.user_id=current_user.id image.type_id=request.form.get("type_id") image.type=request.form.get("type") file_url = upload_file_to_s3(file, Config.S3_BUCKET) image.image_url = file_url db.session.add(image) db.session.commit() images[idx] = image.to_dict() idx += 1 return images, 201 else: return {"errors":"no files"}, 400