Update manager

slimovich · Jul 1, 2020 · 3209c51 · 3209c51
1 parent 382eb11
commit 3209c51
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/core/manage.py b/src/core/manage.py
@@ -24,11 +24,14 @@ def database() -> None:
 
 
 @database.command(help="create database")
-def create() -> int:
- return subprocess.call(["alembic", "upgrade", "head"]) # nosec
+@click.option('-n', '--name', default="fast_api", help='name for database')
+def create(name: str) -> int:
+ click.echo(f"Creating database name: {name}")
+ return subprocess.call(["createdb", f"{name}"]) # nosec
 
 
 @database.command(help="Make migration")
+@click.option('-m', '--msg', default="autogenerate", help='Add a message, default=autogenerate')
 def migration(msg: str) -> int:
  return subprocess.call(["alembic", "revision", "--autogenerate", "-m", f"{msg}"]) # nosec