diff --git a/app.py b/app.py index f6e4d28..26a2fc4 100644 --- a/app.py +++ b/app.py @@ -9,6 +9,7 @@ from routes.create import addasset_bp from routes.update import update_bp from routes.delete import delete_bp from routes.upload import upload_bp +from routes.confirm_save import confirm_save_bp app = Flask(__name__) app.secret_key = 'your_secret_key' # Required for flashing messages and session @@ -23,6 +24,7 @@ app.register_blueprint(addasset_bp) app.register_blueprint(update_bp) app.register_blueprint(delete_bp) app.register_blueprint(upload_bp) +app.register_blueprint(confirm_save_bp) with app.app_context(): db.create_all() diff --git a/routes/confirm_save.py b/routes/confirm_save.py new file mode 100644 index 0000000..93f2754 --- /dev/null +++ b/routes/confirm_save.py @@ -0,0 +1,21 @@ +from flask import Blueprint, redirect, session +from models import Asset, db + +confirm_save_bp = Blueprint('confirm_save', __name__) + +# When confirmed, write the csv data to the database +@confirm_save_bp.route('/confirm_save', methods=['POST']) +def confirm_save(): + assets = session.get('assets', []) # Retrieve assets from session + for asset_data in assets: + asset = Asset( + assettag=asset_data['assettag'], + hostname=asset_data['hostname'], + warrantyfrom=asset_data['warrantyfrom'], + status=asset_data['status'], + staffnum=asset_data['staffnum'] + ) + db.session.add(asset) + db.session.commit() + session.pop('assets', None) # Clear session data + return redirect('/viewall') \ No newline at end of file diff --git a/routes/upload.py b/routes/upload.py index da2e117..6b98d1b 100644 --- a/routes/upload.py +++ b/routes/upload.py @@ -35,20 +35,3 @@ def upload_file(): # Render the upload page for GET requests return render_template('upload.html') - -# When confirmed, write the csv data to the database -@upload_bp.route('/confirm_save', methods=['POST']) -def confirm_save(): - assets = session.get('assets', []) # Retrieve assets from session - for asset_data in assets: - asset = Asset( - assettag=asset_data['assettag'], - hostname=asset_data['hostname'], - warrantyfrom=asset_data['warrantyfrom'], - status=asset_data['status'], - staffnum=asset_data['staffnum'] - ) - db.session.add(asset) - db.session.commit() - session.pop('assets', None) # Clear session data - return redirect('/viewall') \ No newline at end of file