def test_set_password(self): new_user = User(username='******', email='*****@*****.**') new_user.set_password('mypassword') db.session.add(new_user) db.session.commit() user = User.query.filter_by(username='******').first() self.assertTrue(user.check_password('mypassword'))
def register(): """ Registration for PhaME :return: """ if current_user.is_authenticated: # logging.debug('current user is authenticated') return redirect(url_for('phame.index')) form = RegistrationForm() # logging.debug(f'form {form.__dict__}') # logging.debug(f'username {form.username.data}') response_object = {'status': 'fail', 'message': 'Invalid payload'} if form.validate_on_submit(): try: # logging.debug(f'form email {form.email.data}') user = User.query.filter_by(email=form.email.data).first() # logging.debug(f'register user {user}') if not user: user = User(username=form.username.data, email=form.email.data) user.set_password(form.password.data) db.session.add(user) db.session.commit() user = User.query.filter_by(email=form.email.data).first() # logging.debug(f'user query {user.email}') flash('Congratulations, you are now a registered user!') if not os.path.exists( os.path.join(current_app.config['PROJECT_DIRECTORY'], user.username)): os.makedirs( os.path.join(current_app.config['PROJECT_DIRECTORY'], user.username)) if not os.path.exists( os.path.join(current_app.config['UPLOAD_DIRECTORY'], user.username)): os.makedirs( os.path.join(current_app.config['UPLOAD_DIRECTORY'], user.username)) response_object = { 'status': 'success', 'message': f'{form.email.data} was added' } return redirect(url_for('users.login')) else: # logging.debug(f'response_object {response_object}') response_object['message'] = \ 'Sorry. That email already exists.' return jsonify(response_object), 400 except IntegrityError: db.session.rollback() return jsonify(response_object), 400 return render_template('register.html', title='Register', form=form)
def seed_db(): """Seeds the database.""" user_1 = User(username='******', email="*****@*****.**") user_1.set_password('admin') db.session.add(user_1) user_2 = User(username='******', email="*****@*****.**") user_2.set_password('admin') db.session.add(user_2) db.session.commit() click.echo('Database for User Initialized with some data!')
def add_user(self, username=None, email=None, password=None, is_admin=False): username = username if username else 'mark' eml = email if email else '*****@*****.**' pssword = password if password else 'test1' pssword2 = password if password else 'test1' user = User(username=username, email=eml, is_admin=is_admin) user.set_password(pssword) db.session.add(user) db.session.commit() return user
def seed_db(): user1 = User(username=os.environ.get('USER'), email=os.environ.get('EMAIL'), is_admin=False) print(os.environ.get('USER')) user1.set_password(os.environ.get('PASSWORD')) db.session.add(user1) print(os.environ.get('ADMIN')) admin = User(username=os.environ.get('ADMIN'), email=os.environ.get('ADMIN_EMAIL'), is_admin=True) admin.set_password(os.environ.get('ADMIN_PASSWORD')) db.session.add(admin) public = User(username=os.environ.get('PUBLIC'), email=os.environ.get('PUBLIC_EMAIL'), is_admin=False) public.set_password(os.environ.get('PUBLIC_PASSWORD')) db.session.add(public) db.session.commit()
def add_user(username, email, password, is_admin=False): user = User(username=username, email=email, is_admin=is_admin) user.set_password(password) db.session.add(user) db.session.commit() return user