def init_database(dbengine, session):
"""initialize database"""
Base.metadata.create_all(dbengine)
session.expire_all()
# init roles and test users
# this should go somewhere else later
for role in ['Admin','Instrument','Time Series', 'API']:
if not session.query(Role).filter_by(name=role).count():
r = Role(name=role)
session.add(r)
session.commit()
if not session.query(User).filter_by(email='*****@*****.**').count():
u = User(
first_name='Test', last_name='Admin',
email='*****@*****.**', username='******',
password=user_manager.hash_password('12345678'),
is_enabled=True)
r = session.query(Role).filter_by(name='Admin').first()
u.roles.append(r)
session.add(u)
session.commit()
if not session.query(User).filter_by(email='*****@*****.**').count():
u = User(
first_name='Test', last_name='User',
email='*****@*****.**', username='******',
password=user_manager.hash_password('12345678'),
is_enabled=True)
session.add(u)
session.commit()
def set_password(instid):
data = json.loads(request.data)
user = session.query(User).filter_by(id=instid).first()
if not user:
return "User not found", 404
# should eventually perform check in password complexity
if data.has_key('password') and data['password']:
user.password = user_manager.hash_password(data['password'])
session.commit()
return "password updated for user %s" % user.email, 200
else:
return "missing password", 400
from oii.ifcb2.orm import User, Role
from oii.ifcb2.dashboard.flasksetup import user_manager
if __name__=='__main__':
print 'Searching for default admin user in database ...'
user = session.query(User).filter_by(email='*****@*****.**').first()
if user is not None:
print 'Found user. Deleting ...'
session.delete(user)
session.commit()
else:
print 'Not found.'
print 'Creating recovery user/password ...'
username = '******'
password = gen_id()[:8]
u = User(
first_name='Admin',
last_name='User',
email=username,
username=username,
password=user_manager.hash_password(password),
is_enabled=True
)
r = session.query(Role).filter_by(name='Admin').first()
u.roles.append(r)
session.add(u)
session.commit()
print 'User %s created with password %s' % (username, password)
from oii.ifcb2.session import dbengine, session
from oii.ifcb2.orm import User, Role
from oii.ifcb2.dashboard.flasksetup import user_manager
if __name__ == '__main__':
try:
password = sys.argv[1]
except IndexError:
password = gen_id()[:8]
try:
email = sys.argv[2]
except IndexError:
email = '*****@*****.**'
user = session.query(User).filter_by(email=email).first()
if user is not None:
session.delete(user)
session.commit()
u = User(first_name='Admin',
last_name='User',
email=email,
username=email,
password=user_manager.hash_password(password),
is_enabled=True)
r = session.query(Role).filter_by(name='Admin').first()
u.roles.append(r)
session.add(u)
session.commit()
print "%s|%s" % (email, password)
