39 lines
1.8 KiB
Python
39 lines
1.8 KiB
Python
from flask import Blueprint, request, render_template, redirect
|
|
from models import Asset, db
|
|
from config import item_attributes
|
|
|
|
update_bp = Blueprint('editasset', __name__)
|
|
|
|
@update_bp.route('/update/<string:assettag>/', methods=['GET', 'POST'])
|
|
def update(assettag):
|
|
item = Asset.query.filter_by(assettag=assettag).first()
|
|
if request.method == 'POST':
|
|
if item:
|
|
assettag = request.form['assettag']
|
|
hostname = request.form['hostname']
|
|
warrantyfrom = request.form['warrantyfrom']
|
|
status = request.form['status']
|
|
if status not in ['Active', 'Inactive']:
|
|
return render_template('update.html', item=item, exc='status', item_attributes=item_attributes) # Ensure status is valid
|
|
try:
|
|
staffnum = int(request.form['staffnum'])
|
|
except ValueError:
|
|
return render_template('update.html', item=item, exc='staffnum', item_attributes=item_attributes)
|
|
|
|
try:
|
|
setattr(item, 'assettag', assettag)
|
|
setattr(item, 'hostname', hostname)
|
|
setattr(item, 'warrantyfrom', warrantyfrom)
|
|
setattr(item, 'status', status)
|
|
setattr(item, 'staffnum', staffnum)
|
|
|
|
db.session.commit()
|
|
except exc.IntegrityError:
|
|
return render_template('update.html', item=item, exc='integrity', item_attributes=item_attributes)
|
|
except (exc.StatementError, exc.InvalidRequestError) as e:
|
|
return render_template('update.html', item=item, exc='status', item_attributes=item_attributes)
|
|
|
|
return redirect(f'/viewall/')
|
|
return f"Asset {assettag} is not found"
|
|
|
|
return render_template('update.html', item=item, item_attributes=item_attributes) |