示例#1
0
 def create_app(self):
     app = create_app(TestConfig)
     with app.app_context():
         db.create_all()
     return app
示例#2
0
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import sys
import subprocess
from flask.ext.script import Manager, Shell, Server
from flask.ext.migrate import MigrateCommand

from autoposter.app import create_app
from autoposter.user.models import User, Post, DaysOfWeek
from autoposter.settings import DevConfig, ProdConfig
from autoposter.database import db

if os.environ.get("AUTOPOSTER_ENV") == 'prod':
    app = create_app(ProdConfig)
else:
    app = create_app(DevConfig)

manager = Manager(app)
TEST_CMD = "nosetests"


def _make_context():
    '''Return context dict for a shell session so you can access
    app, db, and the User model by default.
    '''
    return {'app': app, 'db': db, 'User': User,
            'Post': Post, 'DaysOfWeek': DaysOfWeek}


@manager.command