AIL-framework/var/www/create_default_user.py

#!/usr/bin/env python3
# -*-coding:UTF-8 -*

import os
import sys
import redis

sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib/'))
import ConfigLoader

sys.path.append(os.path.join(os.environ['AIL_FLASK'], 'modules'))

from Role_Manager import create_user_db, edit_user_db, get_default_admin_token, gen_password

config_loader = ConfigLoader.ConfigLoader()

r_serv = config_loader.get_redis_conn("ARDB_DB")
config_loader = None

if __name__ == "__main__":

    # create role_list
    if not r_serv.exists('ail:all_role'):
        r_serv.zadd('ail:all_role', 1, 'admin')
        r_serv.zadd('ail:all_role', 2, 'analyst')
        r_serv.zadd('ail:all_role', 3, 'user')
        r_serv.zadd('ail:all_role', 4, 'user_no_api')
        r_serv.zadd('ail:all_role', 5, 'read_only')

    username = 'admin@admin.test'
    password = gen_password()
    if r_serv.exists('user_metadata:admin@admin.test'):
        edit_user_db(username, password=password, role='admin')
    else:
        create_user_db(username, password, role='admin', default=True)
    token = get_default_admin_token()

    default_passwd_file = os.path.join(os.environ['AIL_HOME'], 'DEFAULT_PASSWORD')
    to_write_str = '# Password Generated by default\n# This file is deleted after the first login\n#\nemail=admin@admin.test\npassword='
    to_write_str = to_write_str + password + '\nAPI_Key=' + token
    with open(default_passwd_file, 'w') as f:
        f.write(to_write_str)

    print('new user created: {}'.format(username))
    print('password: {}'.format(password))
    print('token: {}'.format(token))