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)
