def initdb():
'''Init/reset database.'''
if not os.environ.get('PRODUCTION'):
db.connection.drop_database(app.config['MONGODB_SETTINGS']['db'])
user_datastore = MongoEngineUserDatastore(db, User, Role)
admin = user_datastore.create_role(name='admin', description='Admin User')
user = user_datastore.create_user(
email='*****@*****.**',
password=encrypt_password('password')
)
user_datastore.add_role_to_user(user, admin)
logs_repository=logs_repository,
current_user=current_user,
orchestation_interactor=orchestation_interactor,
get_time_line_events=get_time_line_events,
user_messaging=user_messaging,
users_privileges=users_privileges,
get_workers_load_model_status=get_workers_load_model)
app.register_blueprint(dashboard.get_blueprint())
api_dashboard = ApiDashboard(create_ml_model=create_ml_model,
token_verification=token_verification,
worker_repository=worker_repository)
app.register_blueprint(api_dashboard.get_blueprint())
# This snippet of code is user with password create example
if CREATE_ADMIN_USER and not len(user_datastore.user_model.objects()):
logging.info("Creating default admin user.")
with app.app_context():
admin_role = user_datastore.find_or_create_role('admin')
user_datastore.create_user(email='admin',
password=utils.hash_password('admin'),
name='admin',
username='******',
roles=[admin_role])
if __name__ == '__main__':
# Start app
app.run(debug=True, host='0.0.0.0', port=int(SERVICE_PORT))