Documentation fixes and blerg

1 files changed, 9 insertions, 0 deletions

diff --git a/migrate.py b/migrate.py
index c386a0e..1de40da 100755
--- a/migrate.py
+++ b/migrate.py
@@ -24,6 +24,7 @@ def cli():
 
 @cli.command
 def install():
+    """Installs a new database"""
     database.create_tables([NewUser, Post, Faccet])
 
     username = click.prompt("Enter a user account name:", default="user", type=str)
@@ -46,5 +47,13 @@ def install():
     click.echo(styled)
 
 
+@cli.command
+def upgrade_schema():
+    """Upgrade the database if needed"""
+    stub_msg = click.style("You are on the latest database", fg="white", bold=True)
+    click.echo(stub_msg)
+    exit(0)
+
+
 if __name__ == "__main__":
     cli()