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)