Moved confirm_save to new module

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()

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')

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')