def main(auto=False):
if not auto:
sys.argv.insert(1, 'mrbob')
flaskxxl.test_request_context().push()
manager.add_command('mrbob', mrbob_manager)
manager.add_command('clean', Clean())
manager.add_command('urls', ShowUrls())
default_cmd = 'mrbob'
if len(sys.argv) > 2:
default_cmd = 'clean'
manager.run(default_command=default_cmd)
def init_manager(app, db, init_for, **kwargs):
"""
Initialise Manager
:param app: Flask app object
:parm db: db instance
:param init_for: init_for function which is normally present in __init__.py of hgapp.
"""
manager.app, manager.db, manager.init_for = app, db, init_for
manager.add_command("db", database)
manager.add_command("clean", Clean())
manager.add_command("showurls", ShowUrls())
manager.add_command("migrate", InitedMigrations())
return manager
def init_manager(app, db, init_for, **kwargs):
"""
Initialise Manager
:param app: Flask app object
:parm db: db instance
:param init_for: init_for function which is normally present in __init__.py of hgapp
:param kwargs: Additional keyword arguments to be made available as shell context
"""
manager.app, manager.db, manager.init_for = app, db, init_for
manager.context = kwargs
manager.add_command("db", database)
manager.add_command("clean", Clean())
manager.add_command("showurls", ShowUrls())
manager.add_command("migrate", InitedMigrations())
return manager
def register_commands(manager):
'''Register all core commands'''
manager.add_command('clean', Clean())
manager.add_command('urls', ShowUrls())
settings = os.environ.get('UDATA_SETTINGS', join(os.getcwd(), 'udata.cfg'))
manager.add_command(
'serve',
Server(port=7000,
use_debugger=True,
use_reloader=True,
threaded=True,
extra_files=[settings]))
# Load all commands submodule
for filename in iglob(join(dirname(__file__), '[!_]*.py')):
module = splitext(basename(filename))[0]
try:
__import__('udata.commands.{0}'.format(module))
except Exception as e:
log.error('Unable to import %s: %s', module, e)
# Load all core modules commands
import udata.core.metrics.commands
import udata.core.user.commands
import udata.core.dataset.commands
import udata.core.organization.commands
import udata.core.search.commands
import udata.core.spatial.commands
import udata.core.jobs.commands
import udata.core.badges.commands
import udata.api.commands
# Dynamic module commands loading
for plugin in manager.app.config['PLUGINS']:
name = 'udata.ext.{0}.commands'.format(plugin)
try:
__import__(name)
except ImportError:
pass
except Exception as e:
log.error('Error importing %s: %s', name, e)
import os
from flask.ext.script import Manager, Server
from flask.ext.script.commands import ShowUrls, Clean
from rpivideo import create_app
from rpivideo.models import db, User, Video
# default to dev config because no one should use this in
# production anyway
env = os.environ.get('APPNAME_ENV', 'dev')
app = create_app('rpivideo.settings.%sConfig' % env.capitalize(), env=env)
manager = Manager(app)
manager.add_command("server", Server())
manager.add_command("show-urls", ShowUrls())
manager.add_command("clean", Clean())
@manager.shell
def make_shell_context():
""" Creates a python REPL with several default imports
in the context of the app
"""
return dict(app=app, db=db, User=User, Video=Video)
@manager.command
def createdb():
""" Creates a database with all of the tables defined in
your SQLAlchemy models
user = User(username='******',
email='*****@*****.**',
password='******')
user.save()
class DB(object):
Model = Model
metadata = Model.metadata
sq = squ
app.test_request_context().push()
engine = Model._engine
if __name__ == '__main__':
manager.add_command('clean', Clean())
manager.add_command('urls', ShowUrls())
manager.add_command('db', alembic_manager)
app.test_request_context().push()
#conn = Model._engine.raw_connection()
#conn.connection.text_factory = str
#manager.add_command('shell', Shell(make_context=lambda:{'app': app, 'db': DB()}))
app.extensions = app.extensions or {}
class O(object):
pass
db = O()
db.db = DB()
app.extensions['sqlalchemy'] = db
manager.run()
def main():
flaskxxl.test_request_context().push()
manager.add_command('mrbob',mrbob_manager)
manager.add_command('clean',Clean())
manager.add_command('urls',ShowUrls())
manager.run(default_command='mrbob')
from flask.ext.script import Manager
from flask.ext.script.commands import Clean
from flask.ext.migrate import MigrateCommand, Migrate
from cheapflight.app import create_app
from cheapflight.ext import db
# import for migrating
import cheapflight.models # noqa
app = create_app()
manager = Manager(app)
migrate = Migrate(app, db)
manager.add_command('db', MigrateCommand)
manager.add_command(Clean())
@manager.command
def runscript(script_path):
'''
Run Function. e.g.:
cheapflight.scripts.rader:main or
cheapflight/scripts/rader.py with main()
'''
if ":" not in script_path:
script_path = os.path.expanduser(os.path.expandvars(script_path))
script_path = os.path.relpath(script_path)
script_path = script_path.rstrip(".py")
script_path = script_path.replace(os.path.sep, ".")
script_path = script_path.lstrip(".")