def register(): if current_user.is_authenticated(): return redirect(url_for('editprofile')) if request.method == 'GET': return render_template('register.html', user=User(), errors={}) user = User( request.form['username'], request.form['password'], request.form['password2'], request.form['email'], request.form['fullname'] ) if user.validate(): user.enabled = True current_app.db_session.add(user) current_app.db_session.commit() login_user(user) flash(u'Du har nĂ¥ opprettet en bruker!') return redirect(request.args.get('next') or url_for('index')) else: return render_template( 'register.html', errors=user.validation_errors, user=user )
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'))