manage.py

#!/usr/bin/env python
# coding: utf-8

"""
    manage
    ~~~~~~

    Set of some useful management commands.

    :copyright: (c) 2015 by Roman Zaiev.
    :license: BSD, see LICENSE for more details.
"""

import subprocess

from flask_script import Shell, Manager

from app import app
from base import User
from ext import db


manager = Manager(app)


@manager.command
def clean_pyc():
    """Removes all *.pyc files from the project folder"""
    clean_command = "find . -name *.pyc -delete".split()
    subprocess.call(clean_command)


@manager.command
def init_data():
    """Fish data for project"""
    db.drop_all()
    db.create_all()

    user = User(username='John Doe', email='john@doe.com', password='test')
    user.save()


manager.add_command('shell', Shell(make_context=lambda:{'app': app, 'db': db}))


if __name__ == '__main__':
    manager.run()