diff --git a/db_create.py b/db_create.py index 06a1783..8d2eed8 100755 --- a/db_create.py +++ b/db_create.py @@ -55,6 +55,6 @@ if __name__ == '__main__': if database_empty: print('Remember to run the following to mark the database up-to-date for Alembic:') - print('./db_migrate.py db stamp head') + print('./db_migrate.py stamp head') # Technically we should be able to do this here, but when you have # Flask-Migrate and Flask-SQA and everything... I didn't get it working. \ No newline at end of file diff --git a/db_migrate.py b/db_migrate.py index 1f480e7..07989d1 100755 --- a/db_migrate.py +++ b/db_migrate.py @@ -1,5 +1,6 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- +import sys from nyaa import app, db from flask_script import Manager from flask_migrate import Migrate, MigrateCommand @@ -10,4 +11,12 @@ manager = Manager(app) manager.add_command("db", MigrateCommand) if __name__ == "__main__": + # Patch sys.argv to default to 'db' + argv_contents = sys.argv[:] + + sys.argv.clear() + sys.argv.append(argv_contents[0]) + sys.argv.append('db') + sys.argv.extend(argv_contents[1:]) + manager.run()