def setUp(self): self.app = create_app('TESTING') self.ctx = self.app.app_context() self.ctx.push() self.client = self.app.test_client() db.create_all() self.init_data()
def create_app(self): """ create the application for flask-testing """ return create_app('TESTING')
import os from innovation_center.app import create_app from innovation_center.app.extensions import db from innovation_center.app.models import User, NewsArticle from innovation_center.app.constants import ADMIN from flask.ext.script import Manager, Shell app = create_app(os.environ.get('FLASK_CONFIG') or 'DEFAULT') manager = Manager(app) @manager.command def initdemo(): db.drop_all() db.create_all() admin = User( first_name='Alex', last_name='Frazer', email='*****@*****.**', role_code=ADMIN, username='******', password='******' ) db.session.add(admin) db.session.commit() User.add_fake_users() NewsArticle.populate_news() if __name__ == '__main__': manager.run()
from innovation_center.app import create_app from innovation_center.app.extensions import db from innovation_center.app.auth.models import User from innovation_center.app.auth.roles import ADMIN from flask.ext.script import Manager, Shell app = create_app('DEVELOPMENT') manager = Manager(app) @manager.command def initdb(): db.drop_all() db.create_all() admin = User( first_name='Alex', last_name='Frazer', email='*****@*****.**', role_code=ADMIN, username='******', password='******' ) db.session.add(admin) db.session.commit() if __name__ == '__main__': manager.run()