main.py 文件源码

def initialize():
    from werkzeug.security import generate_password_hash
    import mongo

    def create_rabbit_init_user(username, password):
        number_of_tries = 100
        while number_of_tries:
            try:
                number_of_tries -= 1
                user.update_rabbitmq_user(username, password)
                break
            except urllib.error.URLError:
                sleep(5)
        else:
            raise Exception()

    mongo.ZimfarmDatabase().initialize()
    users = mongo.UsersCollection()
    if users.find_one() is None:
        username = getenv('INIT_USERNAME', 'admin')
        password = getenv('INIT_PASSWORD', 'admin_pass')
        users.insert_one({
            'username': username,
            'password_hash': generate_password_hash(password),
            'scope': {'admin': True}
        })
        create_rabbit_init_user(username, password)