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
