diff --git a/README.md b/README.md index c55239170aa98d7a2bf8536a45c1a03a29bf74ae..cdd5bb5d57c8fbb5b0605ddb2b36ee9e9a43c4eb 100644 --- a/README.md +++ b/README.md @@ -56,7 +56,7 @@ $ python -m fastapi_template --help usage: FastAPI template [-h] [--version] [--name PROJECT_NAME] [--description PROJECT_DESCRIPTION] [--db {none,sqlite,mysql,postgresql}] - [--orm {none,ormar,sqlalchemy,tortoise}] + [--orm {ormar,sqlalchemy,tortoise}] [--ci {none,gitlab,github}] [--redis] [--migrations] [--kube] [--dummy] [--routers] [--swagger] [--force] @@ -68,7 +68,7 @@ optional arguments: Project description --db {none,sqlite,mysql,postgresql} Database - --orm {none,ormar,sqlalchemy,tortoise} + --orm {ormar,sqlalchemy,tortoise} ORM --ci {none,gitlab,github} Choose CI support diff --git a/fastapi_template/cli.py b/fastapi_template/cli.py index c0f0fded043ce8033684c7891eaca9fc5250de77..b0933e1890ab604c6171e33a587f3dd49fff204d 100644 --- a/fastapi_template/cli.py +++ b/fastapi_template/cli.py @@ -56,7 +56,7 @@ def parse_args(): "--orm", help="ORM", type=str, - choices=list(map(attrgetter("value"), ORM)), + choices=[orm.value for orm in ORM if orm != ORM.none], default=None, dest="orm", ) @@ -65,7 +65,7 @@ def parse_args(): help="Choose CI support", default=None, type=str, - choices=list(map(attrgetter("value"), CIType)), + choices=[ci.value for ci in CIType], dest="ci_type", ) parser.add_argument( @@ -192,7 +192,7 @@ def read_user_input(current_context: BuilderContext) -> BuilderContext: current_context.orm = radiolist_dialog( "ORM", text="Which ORM do you want?", - values=[(orm, orm.value) for orm in list(ORM)], + values=[(orm, orm.value) for orm in list(ORM) if orm != ORM.none], ).run() if current_context.orm is None: raise KeyboardInterrupt()