ewpt3ch/flaskfdx
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

uploads working

Browse Source
This commit is contained in:
Eric Phillips
2021-10-20 19:36:59 -06:00
parent 82134419b7
commit 3e19bba268
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flaskfdx/dashboard.py
+7 -5
View File
@@ -2,7 +2,7 @@ import functools
import os import os
from flask import( from flask import(
Blueprint, flash, g, redirect, render_template, request, session, url_for Blueprint, current_app, flash, g, redirect, render_template, request, session, url_for
) )
from werkzeug.exceptions import abort from werkzeug.exceptions import abort
from werkzeug.utils import secure_filename from werkzeug.utils import secure_filename
@@ -50,6 +50,10 @@ def index():
return render_template('dashboard/index.html', lastseven=lastseven) return render_template('dashboard/index.html', lastseven=lastseven)
def allowed_file(filename):
ALLOWED_EXTENSIONS = {'xls', 'xlsx'}
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@bp.route('/upload', methods=('GET', 'POST')) @bp.route('/upload', methods=('GET', 'POST'))
@login_required @login_required
def upload(): def upload():
@@ -62,10 +66,8 @@ def upload():
if file.filename == '': if file.filename == '':
flash('No selected file') flash('No selected file')
return redirect(request.url) return redirect(request.url)
return redirect(url_for('index'))
if file and allowed_file(file.filename): if file and allowed_file(file.filename):
filename = secure_filename(file.filename) filename = secure_filename(file.filename)
file file.save(os.path.join(current_app.config['UPLOAD_FOLDER'], filename))
return redirect(url_for('index'))
return render_template('dashboard/upload.html') return render_template('dashboard/upload.html')