コード例 #1
0
ファイル: conftest.py プロジェクト: oii/ogre
def _flask_app(request, app_config):
    if request.config.getoption("--only-client"):
        yield None
        return

    from ogreserver.factory import create_app, make_celery, configure_extensions, register_blueprints, register_signals
    from ogreserver.extensions.celery import register_tasks

    app = create_app(app_config)
    app.testing = True
    app.celery = make_celery(app)
    register_tasks(app)
    configure_extensions(app)
    register_blueprints(app)
    register_signals(app)
    yield app
    if os.path.exists(app_config["WHOOSH_BASE"]):
        shutil.rmtree(app_config["WHOOSH_BASE"])
コード例 #2
0
ファイル: manage.py プロジェクト: oii/ogre
import sys

import boto

from flask.ext.script import Manager
from sqlalchemy.exc import IntegrityError, ProgrammingError

import rethinkdb as r
from rethinkdb.errors import RqlRuntimeError

from ogreserver.factory import create_app, make_celery, register_signals
from ogreserver.extensions.celery import register_tasks
from ogreserver.extensions.database import setup_db_session, create_tables, setup_roles
from ogreserver.utils import connect_s3, decode_rql_dates

app = create_app()
manager = Manager(app)


@manager.command
def cleardb():
    if app.debug is False:
        print 'You cannot run cleardb when not in DEBUG!!'
        return
    conn = r.connect("localhost", 28015, db='ogreserver').repl()
    r.table('ebooks').delete().run()
    r.table('versions').delete().run()
    r.table('formats').delete().run()
    r.table('sync_events').delete().run()
    if os.path.exists('search.db'):
        import shutil, subprocess