def test_register_user_failed(self):
usr = UserModel(
username=self.first_user.username,
password=self.first_user.password,
email=self.first_user.email
)
response_body = UserApi.register_user(usr).json()
self.check_error(response_body, 'Username {} already exist'.format(usr.username))
usr.username = random_string(7)
response_body = UserApi.register_user(usr).json()
self.check_error(response_body, 'Email {} already exist'.format(usr.email.lower()))
usr.email = random_string(7)
response_body = UserApi.register_user(usr).json()
self.check_error(response_body, 'Not valid email')
usr.email = "{}@{}.{}".format(random_string(7), random_string(3), random_string(2))
usr.username = random_string(6)
response_body = UserApi.register_user(usr).json()
self.check_error(response_body, "Username must be greater than 6 characters")
usr.username = random_string(7)
usr.password = random_string(6)
response_body = UserApi.register_user(usr).json()
self.check_error(response_body, "Password must be greater than 6 characters")
def register():
if current_user.is_authenticated:
return redirect('/coffeeshops')
form = registerForm()
if form.validate_on_submit():
if request.method == "POST":
username = request.form["username"]
email = request.form["email"]
pw = request.form["password"]
user = UserModel.query.filter_by(email=email).first()
if user is not None:
flash('You are already registered. Proceed to login')
return redirect(url_for('login'))
if user is not None and user.check_password(pw):
# login_user(user)
# return redirect('/coffeeshops')
return redirect('/login')
else:
user = UserModel(email=email)
user.set_password(pw)
user.email = email
user.username = username
db.session.add(user)
db.session.commit()
flash(
'Congratulations, You are now a registered user! Please login'
)
return redirect(url_for('login'))
# return render_template('login.html',form=form)
else:
return render_template('login.html', form=form)
else:
return render_template('login.html', form=form)
def register():
form = registerForm()
if form.validate_on_submit():
if request.method == "POST":
username = request.form["username"]
pw = request.form["password"]
email = request.form["email"]
user = UserModel.query.filter_by(username=username).first()
address = UserModel.query.filter_by(email=email).first()
if user:
flash("Username already exists.")
return redirect('/register')
elif address:
flash("Email already in use.")
return redirect('/register')
else:
new_user = UserModel(username=username)
new_user.set_password(pw)
new_user.email = email
db.session.add(new_user)
db.session.commit()
return redirect('/login')
else:
return render_template('registration.html', form=form)
else:
return render_template('registration.html', form=form)
def create(username: str, email: str, password: str, first_name: str,
last_name: str) -> "UserModel":
""" Create a new user """
user = UserModel()
user.username = username
user.email = email
user.password = password
user.first_name = first_name
user.last_name = last_name
user.save()
return user
def addUser(self, username, email, password):
"""
Adds a new record to the database
:param username: String
:param email: String
:param password: String
:return: None
"""
user = UserModel(username=username)
user.set_password(password)
user.email = email
self.db.session.add(user)
self.db.session.commit()
def signup(request):
if request.method == "POST":
form = UserForm(request.POST)
if form.is_valid():
username = form.cleaned_data['username']
password = form.cleaned_data['password']
email = form.cleaned_data['email']
#save into db
user = UserModel()
user.username = username
user.password = password
user.email = email
user.save()
return render_to_response('account_success.html',{'username':username})
else:
form = UserForm()
return render_to_response('account_signup.html',{'form':form})
