scripts/generate_registration_files.py

# ---------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------

import json


def prompt_yes_no(prompt: str) -> bool:
    response = "INVALID"
    while response not in ["y", "n"]:
        response = input(prompt + " (y/n)")
        response = response.lower()
    return response == "y"


def create_workspace_config():
    JSON_FILE = "config.json"
    config_dict = dict()
    config_dict["subscription_id"] = input("Enter subscription id: ")
    config_dict["resource_group"] = input("Enter resource group name: ")
    config_dict["workspace_name"] = input("Enter workspace name: ")

    with open(JSON_FILE, "w") as jf:
        json.dump(config_dict, jf)
    print("Written {0}".format(JSON_FILE))


def create_component_config(desired_version: int):
    JSON_FILE = "component_config.json"
    config_dict = dict()
    config_dict["version"] = desired_version

    with open(JSON_FILE, "w") as jf:
        json.dump(config_dict, jf)
    print("Written {0}".format(JSON_FILE))


def user_specified_component_config():
    user_version = int(input("Enter version: "))
    create_component_config(desired_version=user_version)


def default_component_config():
    create_component_config(desired_version=1)


def main():
    default_component_config()
    if prompt_yes_no("Create workspace config.json? "):
        create_workspace_config()
    print("Completed")


if __name__ == "__main__":
    main()