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'))