Skip to content

Commit

Permalink
Adds skip repository and skip model when generating module without da…
Browse files Browse the repository at this point in the history
…tabase connection
  • Loading branch information
cicekhayri committed Dec 13, 2023
1 parent 6fe08d2 commit 313c409
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 8 deletions.
25 changes: 17 additions & 8 deletions pyblaze/cli/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,21 +21,30 @@ def new():

@new.command()
@click.argument("name")
def module(name):
@click.option("--skip-repository", "skip_repository", is_flag=True, required=False)
@click.option("--skip-model", "skip_model", is_flag=True, required=False)
def module(name, skip_repository, skip_model):
if not name:
click.echo("Please provide a name for the module")
return
if not database_file_exists():
click.echo(
"Database file doesn't exists, please generate one before generating modules"
)

if not (skip_model or skip_repository) and not database_file_exists():
click.echo("Database file doesn't exist. Please generate one before generating modules")
return

create_module_files(name, skip_repository, skip_model)
add_route_to_app(name)


def create_module_files(name, skip_repository, skip_model):
create_src_directory()
create_controller_file(name)
generate_model_file(name)
generate_repository_file(name)
add_route_to_app(name)

if not skip_model:
generate_model_file(name)

if not skip_repository:
generate_repository_file(name)


@new.command()
Expand Down
1 change: 1 addition & 0 deletions tests/test_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ def test_resource_command(
):
runner = CliRunner()
result = runner.invoke(cli, ["new", "module", "example"])
print(result.output)
assert result.exit_code == 0
assert "Module 'example' created successfully." in result.output

Expand Down

0 comments on commit 313c409

Please sign in to comment.