diff --git a/pyproject.toml b/pyproject.toml index c9ae63e6..68c48950 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -57,6 +57,7 @@ Changelog = "https://github.com/sparckles/robyn/blob/main/CHANGELOG.md" [tool.poetry.dependencies] python = "^3.7" +inquirerpy = "0.3.4" maturin = "0.14.12" watchdog = "2.2.1" multiprocess = "0.70.14" diff --git a/robyn/__main__.py b/robyn/__main__.py index 37fc27f4..c84d55a2 100644 --- a/robyn/__main__.py +++ b/robyn/__main__.py @@ -1,22 +1,26 @@ import os import webbrowser - +from InquirerPy import prompt +from InquirerPy.base.control import Choice from .argument_parser import Config -def check(value, input_name): - while value not in ["Y", "N"]: - print("Invalid input. Please enter Y or N") - value = input(f"Need {input_name}? (Y/N) ") - return value - - def create_robyn_app(): - project_dir = input("Enter the name of the project directory: ") - docker = input("Need Docker? (Y/N) ") - - # Initailize a new Robyn project - docker = check(docker, "Docker") + questions = [ + {"type": "input", "message": "Enter the name of the project directory:"}, + { + "type": "list", + "message": "Need Docker? (Y/N)", + "choices": [ + Choice("Y", name="Y"), + Choice("N", name="N"), + ], + "default": None, + }, + ] + result = prompt(questions=questions) + project_dir = result[0] + docker = result[1] print(f"Creating a new Robyn project '{project_dir}'...") @@ -74,7 +78,8 @@ def index(): def docs(): print("Opening Robyn documentation... | Offline docs coming soon!") - webbrowser.open("https://sansyrox.github.io/robyn/#/") + webbrowser.open("https://sparckles.github.io/robyn/#/") + if __name__ == "__main__":