diff --git a/cli.py b/cli.py index 184d7e0..36c1b0d 100644 --- a/cli.py +++ b/cli.py @@ -7,19 +7,15 @@ def cli(): pass -@click.command() +@cli.command() def createdb(): db.create_all() -@click.command() +@cli.command() def dropdb(): db.drop_all() -cli.add_command(createdb) -cli.add_command(dropdb) - - if __name__ == '__main__': cli()