From 2990aeade26e3f1e3120ca935d636ae44c861c88 Mon Sep 17 00:00:00 2001 From: sb745 Date: Wed, 5 Mar 2025 01:05:46 +0200 Subject: [PATCH] Added back db_migrate.py --- db_migrate.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 db_migrate.py diff --git a/db_migrate.py b/db_migrate.py new file mode 100644 index 0000000..928ea1d --- /dev/null +++ b/db_migrate.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +""" +Database migration script for Nyaa. +Compatible with Python 3.13 and Flask-Migrate 4.0. +""" +import sys +from typing import List + +from flask_migrate import Migrate +from flask.cli import FlaskGroup + +from nyaa import create_app +from nyaa.extensions import db + +app = create_app('config') +migrate = Migrate(app, db) + +def create_cli_app(): + return app + +cli = FlaskGroup(create_app=create_cli_app) + +if __name__ == "__main__": + # Patch sys.argv to default to 'db' + if len(sys.argv) > 1 and sys.argv[1] not in ['--help', '-h']: + if sys.argv[1] not in ['db', 'routes', 'shell', 'run']: + args: List[str] = sys.argv.copy() + args.insert(1, 'db') + sys.argv = args + + cli()