break upload logic calls out to seperate file
This commit is contained in:
@@ -5,10 +5,10 @@ from flask import(
|
|||||||
Blueprint, current_app, 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 flaskfdx.auth import login_required
|
from flaskfdx.auth import login_required
|
||||||
from flaskfdx.db import get_db
|
from flaskfdx.db import get_db
|
||||||
|
from flaskfdx.upload import process_file
|
||||||
|
|
||||||
bp = Blueprint('dashboard', __name__)
|
bp = Blueprint('dashboard', __name__)
|
||||||
|
|
||||||
@@ -67,7 +67,12 @@ def upload():
|
|||||||
flash('No selected file')
|
flash('No selected file')
|
||||||
return redirect(request.url)
|
return redirect(request.url)
|
||||||
if file and allowed_file(file.filename):
|
if file and allowed_file(file.filename):
|
||||||
filename = secure_filename(file.filename)
|
# todo validity check before processing
|
||||||
|
filename, error = process_file(filetype, file)
|
||||||
|
if error:
|
||||||
|
print(error)
|
||||||
|
else:
|
||||||
file.save(os.path.join(current_app.config['UPLOAD_FOLDER'], filename))
|
file.save(os.path.join(current_app.config['UPLOAD_FOLDER'], filename))
|
||||||
|
|
||||||
return redirect(url_for('index'))
|
return redirect(url_for('index'))
|
||||||
return render_template('dashboard/upload.html')
|
return render_template('dashboard/upload.html')
|
||||||
|
|||||||
@@ -112,16 +112,11 @@ def processDsw(file, database):
|
|||||||
|
|
||||||
reportDate = getDate(srcSheet)
|
reportDate = getDate(srcSheet)
|
||||||
|
|
||||||
#open the database
|
|
||||||
conn = openDatabase(database)
|
|
||||||
if conn == None:
|
|
||||||
sys.exit('database could not be opened, does it exist')
|
|
||||||
|
|
||||||
# check if records already exist and then add records if not
|
# check if records already exist and then add records if not
|
||||||
if checkDate(conn, reportDate):
|
if checkDate(database, reportDate):
|
||||||
print("we have already processed " + reportDate)
|
print("we have already processed " + reportDate)
|
||||||
else:
|
else:
|
||||||
extractDsw(srcSheet, conn, reportDate)
|
extractDsw(srcSheet, database, reportDate)
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
@@ -132,8 +127,13 @@ def main():
|
|||||||
parser.add_argument("database")
|
parser.add_argument("database")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
#open the database
|
||||||
|
conn = openDatabase(args.database)
|
||||||
|
if conn == None:
|
||||||
|
sys.exit('database could not be opened, does it exist')
|
||||||
|
|
||||||
if args.type == 'dsw':
|
if args.type == 'dsw':
|
||||||
processDsw(args.inFile, args.database)
|
processDsw(args.inFile, conn)
|
||||||
else:
|
else:
|
||||||
print('type not found or not entered')
|
print('type not found or not entered')
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user