def post(self):
if 'user_id' in session:
session.pop('user_id')
if 'comment' in session:
session.pop('comment')
data = {}
if settings.XSRF_COOKIES:
data['csrf_token'] = generate_csrf_token('home')
return self.render_template('home.html', **data)
def post(self):
action = request.form.get('action')
comment = request.form.get('comment')
if action == 'add_comment':
data = {'departments': Department.query.all(), 'comment': comment}
if settings.XSRF_COOKIES:
data['csrf_token'] = generate_csrf_token('register')
return self.render_template('form.html', **data)
else:
"""
if 'user_id' in session:
logging.info('have user')
return redirect(url_for('thanks'))
if not 'comment' in session:
logging.info('not comment')
return redirect(url_for('home'))
"""
#comment = request.form['comment']
"""
if not comment:
logging.error('not comment')
return redirect(url_for('home'))
if len(comment) > 140:
logging.error('not size comment')
return redirect(url_for('home'))
"""
if settings.XSRF_COOKIES:
csrf_protect('register')
form = RegisterForm(request.form)
form.email_exists.data = bool(User.query.filter_by(
email=form.email.data).count())
form.dni_exists.data = bool(User.query.filter_by(
dni=form.dni.data).count())
form.cod_dpto.query = Department.query.all()
#form.comment.data = session.get('comment')
#form.comment.data = comment
if form.validate():
user = User()
form.populate_obj(user)
user.cod_dpto = form.cod_dpto.data.id
user.enabled = True
db_session.add(user)
try:
db_session.commit()
except Exception as exc:
logging.error(exc)
db_session.rollback()
db_session.remove()
return redirect(url_for('thanks'))
else:
db_session.remove()
session['user_id'] = str(user.id)
return redirect(url_for('thanks'))
else:
logging.error(form.errors)
return redirect(url_for('thanks'))
