CLI - refactoring

fittrackee/migrations/commands.py

@@ -0,0 +1,44 @@
+import os
+import shutil
+
+import click
+from flask_migrate import upgrade
+
+from fittrackee import db
+from fittrackee.cli.app import app
+
+BASEDIR = os.path.abspath(os.path.dirname(__file__))
+app_settings = os.getenv('APP_SETTINGS', 'fittrackee.config.ProductionConfig')
+
+
+@click.group(name='db')
+def db_cli() -> None:
+    """Manage database."""
+    pass
+
+
+@db_cli.command('upgrade')
+def upgrade_db() -> None:
+    """Apply migrations."""
+    with app.app_context():
+        upgrade(directory=BASEDIR)
+
+
+@db_cli.command('drop')
+def drop_db() -> None:
+    """Empty database and delete uploaded files for dev environments."""
+    with app.app_context():
+        if app_settings == 'fittrackee.config.ProductionConfig':
+            click.echo(
+                click.style(
+                    'This is a production server, aborting!', bold=True
+                ),
+                err=True,
+            )
+            return
+        db.engine.execute("DROP TABLE IF EXISTS alembic_version;")
+        db.drop_all()
+        db.session.commit()
+        click.echo('Database dropped.')
+        shutil.rmtree(app.config['UPLOAD_FOLDER'], ignore_errors=True)
+        click.echo('Uploaded files deleted.')