def setUp(self): app.config['SECRET_KEY'] = '08424fa094ead1b004ff459bbd6e2750' app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///test.db' self.app = app.test_client() db.create_all() hashed_password = bcrypt.generate_password_hash('12345').decode( 'utf-8') u1 = User(username='******', email='*****@*****.**', password=hashed_password) u2 = User(username='******', email='*****@*****.**', password=hashed_password) db.session.add(u1) db.session.add(u2) db.session.commit()
def create_user(username, password): pw_hash = hash_password(password) user = User(username=username, password=pw_hash) try: db.session.add(user) db.session.commit() except: print('An error occurred, please try again with a different username')
def test_is_vote_success(self): u1 = User(username='******', email='*****@*****.**', password='******', preference='1') db.session.add(u1) db.session.commit() u = User.query.get(1) self.assertTrue(u.preference == 1) self.assertTrue(u.username == 'user1')
def register(): if current_user.is_authenticated: return redirect(url_for('home')) form = RegistrationForm() if form.validate_on_submit(): hashed_password = bcrypt.generate_password_hash(form.password.data).decode("utf-8") user = User(username=form.username.data, email=form.email.data, password=hashed_password) db.session.add(user) db.session.commit() flash(f"Account created for {form.username.data}, Log in now!", "success") return redirect(url_for('login')) return render_template("register.html", title="Register", form = form)
def create_user(username, password): if current_user.username == 'admin': pw_hash = hash_password(password) user = User(username=username, password=pw_hash) try: db.session.add(user) db.session.commit() flash(f'Account created for {username}', 'success') except: flash('Something went wrong...', 'failure') else: flash('Only the admin user can create new accounts', 'info')
def reset_token(token): if current_user.is_authenticated: flash(f'Please log out before reset the password') return redirect(url_for('home')) user = User.verify_reset_token(token) if user is None: flash(f"That is an invalid or expired token", "warning") return redirect(url_for('reset_request')) form = ResetPasswordForm() if form.validate_on_submit(): hashed_password = bcrypt.generate_password_hash(form.password.data).decode("utf-8") user.password = hashed_password db.session.commit() flash(f"Your password has been updated", "success") return redirect(url_for('login')) return render_template('reset_token.html', title='Reset Password', form=form)
def test_invalid_registration_duplicates(self): u1 = User(username="******", email="*****@*****.**", password="******", preference="1") db.session.add(u1) db.session.commit() # duplicate username response = self.register("existing_user", "*****@*****.**", "password", "password", "1") self.assertEqual(response.status_code, 200) # duplicate email response = self.register("user", "*****@*****.**", "password", "password", "1") self.assertEqual(response.status_code, 200)
def register(): if current_user.is_authenticated: return redirect(url_for('index')) form = RegistrationForm() form.preference.choices = [(candidate.id, candidate.name) for candidate in Candidate.query.all()] if form.validate_on_submit(): hashed_password = bcrypt.generate_password_hash( form.password.data).decode("utf-8") user = User(username=form.username.data, email=form.email.data, password=hashed_password, preference=form.preference.data) db.session.add(user) db.session.commit() flash(f'Welcome {form.username.data}! You can Log in now!', 'success') return redirect(url_for('login')) return render_template('signup.html', form=form)
def register(): login = request.form.get('login') fullname = request.form.get('fullname') password = request.form.get('password') password2 = request.form.get('password2') next_page = request.args.get('next') if request.method == 'POST': if not (login or password or password2): flash('Заполните все поля') elif password != password2: flash('Пароли не совпадают') else: hash_pwd = generate_password_hash(password) new_user = User(login=login, fullname=fullname, password=hash_pwd) db.session.add(new_user) db.session.commit() return redirect(url_for('login_page') + '?next=' + next_page) return render_template('register.html', next=next_page)
def register(): if current_user.is_authenticated: return redirect(url_for('profile')) form = RegisterForm(request.form) if request.method == 'POST': email = request.form['email'] username = request.form['username'] fname = request.form['fname'] lname = request.form['lname'] password = request.form['password'] if form.validate_on_submit(): hashed_pw = bcrypt.generate_password_hash(password).decode('utf-8') user = User(email=email, username=username, fname=fname, lname=fname, password=hashed_pw) db.session.add(user) db.session.commit() flash(f'Account Created for', 'sucess') return redirect(url_for('login')) return render_template('register.html', form=form)