46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import os
|
|
import shutil
|
|
|
|
import click
|
|
from flask_migrate import upgrade
|
|
from sqlalchemy.sql import text
|
|
|
|
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.session.execute(text("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.')
|