candifloss/flask_crud_app
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use config in update.py

Browse Source
This commit is contained in:
Candifloss 2025-02-08 00:04:43 +05:30
parent 200810d793
commit fc93e6b716
1 changed files with 28 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
routes/update.py
View File

@ -1,39 +1,44 @@
from flask import Blueprint, request, render_template, redirect from flask import Blueprint, request, render_template, redirect
from models import Asset, db from models import Asset, db
from config import item_attributes from config import item_attributes
from sqlalchemy import exc # Import exc for database exceptions
update_bp = Blueprint('editasset', __name__) update_bp = Blueprint('editasset', __name__)
@update_bp.route('/update/<string:assettag>/', methods=['GET', 'POST']) @update_bp.route('/update/<string:assettag>/', methods=['GET', 'POST'])
def update(assettag): def update(assettag):
item = Asset.query.filter_by(assettag=assettag).first() item = Asset.query.filter_by(assettag=assettag).first()
if not item:
return f"Asset {assettag} is not found"
if request.method == 'GET':
return render_template('update.html', item=item, item_attributes=item_attributes)
if request.method == 'POST': if request.method == 'POST':
if item: # Dynamically get form data using item_attributes
assettag = request.form['assettag'] form_data = {attrib: request.form[attrib] for attrib in item_attributes}
hostname = request.form['hostname']
warrantyfrom = request.form['warrantyfrom'] # Validate status (if it's an option field)
status = request.form['status'] if 'status' in item_attributes and item_attributes['status'].html_input_type == "select":
if status not in ['Active', 'Inactive']: if form_data['status'] not in item_attributes['status'].options:
return render_template('update.html', item=item, exc='status', item_attributes=item_attributes) # Ensure status is valid return render_template('update.html', item=item, exc='status', item_attributes=item_attributes)
# Convert staffnum to int (if it's a number field)
if 'staffnum' in item_attributes and item_attributes['staffnum'].html_input_type == "number":
try: try:
staffnum = int(request.form['staffnum']) form_data['staffnum'] = int(form_data['staffnum'])
except ValueError: except ValueError:
return render_template('update.html', item=item, exc='staffnum', item_attributes=item_attributes) return render_template('update.html', item=item, exc='staffnum', item_attributes=item_attributes)
try: # Update the item with the new data
setattr(item, 'assettag', assettag) for attrib, value in form_data.items():
setattr(item, 'hostname', hostname) setattr(item, attrib, value)
setattr(item, 'warrantyfrom', warrantyfrom)
setattr(item, 'status', status)
setattr(item, 'staffnum', staffnum)
db.session.commit() try:
except exc.IntegrityError: db.session.commit()
return render_template('update.html', item=item, exc='integrity', item_attributes=item_attributes) except exc.IntegrityError:
except (exc.StatementError, exc.InvalidRequestError) as e: return render_template('update.html', item=item, exc='integrity', item_attributes=item_attributes)
return render_template('update.html', item=item, exc='status', item_attributes=item_attributes) except (exc.StatementError, exc.InvalidRequestError):
return render_template('update.html', item=item, exc='status', item_attributes=item_attributes)
return redirect(f'/viewall/') return redirect('/viewall')
return f"Asset {assettag} is not found"
return render_template('update.html', item=item, item_attributes=item_attributes)