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