diff --git a/requirements.txt b/requirements.txt index 1bdc081..856a201 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,3 +10,4 @@ requests celery psycopg2 redis +requests diff --git a/wotstats/views/home.py b/wotstats/views/home.py index 0a9a3e4..a48c068 100644 --- a/wotstats/views/home.py +++ b/wotstats/views/home.py @@ -2,6 +2,7 @@ import re import requests +import validator from flask import ( g, Blueprint, render_template, abort, current_app, redirect, redirect, request, url_for, session, flash @@ -73,19 +74,19 @@ def create_profile_post(): email = request.form['email'] if not name: flash(u'Error: you have to provide a name') - elif '@' not in email: + if not validator.email(email): flash(u'Error: you have to enter a valid email address') return redirect(url_for('home.create_profile')) - else: - flash(u'Profile successfully created') - u = User(email) - u.name = name - u.openid = session['openid'] - u.password = '' - db.session.add(u) - db.session.commit() - return redirect(oid.get_next_url()) + flash(u'Profile successfully created') + u = User(email) + u.name = name + u.openid = session['openid'] + u.password = '' + + db.session.add(u) + db.session.commit() + return redirect(oid.get_next_url()) @pages_home.route('/logout') @@ -176,6 +177,8 @@ def create_or_login(resp): # flash(u'Successfully signed in') g.user = user + print session if not session['token']: + print('not token') return redirect(url_for('pages_home.token')) return redirect(oid.get_next_url())