mirror of https://github.com/CIRCL/AIL-framework
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*-coding:UTF-8 -*
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import redis
|
||
|
|
import configparser
|
||
|
|
|
||
|
|
import bcrypt
|
||
|
|
import secrets
|
||
|
|
|
||
|
|
# Import config
|
||
|
|
sys.path.append('./modules/')
|
||
|
|
|
||
|
|
def hashing_password(bytes_password):
|
||
|
|
hashed = bcrypt.hashpw(bytes_password, bcrypt.gensalt())
|
||
|
|
return hashed
|
||
|
|
|
||
|
|
def create_user_db(username_id , password, default=False):
|
||
|
|
password = password.encode()
|
||
|
|
password_hash = hashing_password(password)
|
||
|
|
r_serv_db.hset('user:all', username_id, password_hash)
|
||
|
|
if default:
|
||
|
|
r_serv_db.set('user:request_password_change', username_id)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
configfile = os.path.join(os.environ['AIL_BIN'], 'packages/config.cfg')
|
||
|
|
if not os.path.exists(configfile):
|
||
|
|
raise Exception('Unable to find the configuration file. \
|
||
|
|
Did you set environment variables? \
|
||
|
|
Or activate the virtualenv.')
|
||
|
|
|
||
|
|
cfg = configparser.ConfigParser()
|
||
|
|
cfg.read(configfile)
|
||
|
|
|
||
|
|
r_serv_db = redis.StrictRedis(
|
||
|
|
host=cfg.get("ARDB_DB", "host"),
|
||
|
|
port=cfg.getint("ARDB_DB", "port"),
|
||
|
|
db=cfg.getint("ARDB_DB", "db"),
|
||
|
|
decode_responses=True)
|
||
|
|
|
||
|
|
username = 'admin@admin.test'
|
||
|
|
# # TODO: create random password
|
||
|
|
password = 'admin'
|
||
|
|
create_user_db(username, password, default=True)
|
||
|
|
|
||
|
|
# create user token
|
||
|
|
token = secrets.token_urlsafe(41)
|
||
|
|
r_serv_db.hset('user:tokens', token, username)
|
||
|
|
|
||
|
|
print('new user created: {}'.format(username))
|
||
|
|
print('password: {}'.format(password))
|