refactor: shorten cli code

jckdotim · Jun 10, 2019 · 2ff7115 · 2ff7115
1 parent 79ec718
commit 2ff7115
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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()