def new_advert(): ''' Create new advert ''' form = AdvertForm() if form.validate_on_submit(): my_advert = Advert() form.populate_obj(my_advert) db.session.add(my_advert) db.session.commit() flash('Advert created correctly', 'success') try: # User info desc = my_advert.description return redirect(url_for('tweet',desc=desc)) except: db.session.rollback() flash('Error generating advert.', 'danger') return render_template('web/new_advert.html', form=form)
def edit_advert(id): ''' Edit advert :param id: Id from advert ''' my_advert = Advert.query.filter_by(id=id).first() my_advert.date = datetime.strptime(my_advert.date, '%Y-%m-%d') date = my_advert.date form = AdvertForm(obj=my_advert) if form.validate_on_submit(): try: # Update contact form.populate_obj(my_advert) db.session.add(my_advert) db.session.commit() # User info flash('Saved successfully', 'success') except: db.session.rollback() flash('Error update advert.', 'danger') return render_template( 'web/edit_advert.html',date = date, form=form)