21 lines
764 B
Python
21 lines
764 B
Python
|
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')
|