def new_asset():
'''
Create new asset
'''
form = AssetForm()
if form.validate_on_submit():
my_asset = Asset()
form.populate_obj(my_asset)
db.session.add(my_asset)
try:
db.session.commit()
# User info
flash('Asset created correctly', 'success')
return redirect(url_for('assets'))
except:
db.session.rollback()
flash('Error generating asset.', 'danger')
return render_template('web/new_asset.html', form=form)
def edit_asset(id):
'''
Edit asset
:param id: Id from asset
'''
my_asset = Asset.query.filter_by(id=id).first()
form = AssetForm(obj=my_asset)
if form.validate_on_submit():
try:
# Update asset
form.populate_obj(my_asset)
db.session.add(my_asset)
db.session.commit()
# User info
flash('Saved successfully', 'success')
except:
db.session.rollback()
flash('Error update asset.', 'danger')
return render_template('web/edit_asset.html', form=form)
