mirror of https://github.com/D4-project/d4-core
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
# -*-coding:UTF-8 -*
|
||
|
|
||
|
import os
|
||
|
import sys
|
||
|
import redis
|
||
|
import configparser
|
||
|
|
||
|
sys.path.append(os.path.join(os.environ['D4_HOME'], 'lib'))
|
||
|
|
||
|
from Role_Manager import create_user_db, edit_user_db, get_default_admin_token, gen_password
|
||
|
|
||
|
host_redis_metadata = os.getenv('D4_REDIS_METADATA_HOST', "localhost")
|
||
|
port_redis_metadata = int(os.getenv('D4_REDIS_METADATA_HOST', 6380))
|
||
|
|
||
|
r_serv = redis.StrictRedis(
|
||
|
host=host_redis_metadata,
|
||
|
port=port_redis_metadata,
|
||
|
db=1,
|
||
|
decode_responses=True)
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
|
||
|
# create role_list
|
||
|
if not r_serv.exists('d4:all_role'):
|
||
|
role_dict = {'admin': 1, 'user': 1, 'sensor_register': 20}
|
||
|
r_serv.zadd('d4:all_role', role_dict)
|
||
|
|
||
|
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['D4_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))
|