Esempio n. 1
0
def register_manager(manager):
    """Register all manager plugins and default commands with the manager."""
    from six.moves.urllib.parse import urlparse
    managers = RegistryProxy('managers', ModuleAutoDiscoveryRegistry, 'manage')

    with manager.app.app_context():
        for script in find_modules('invenio.base.scripts'):
            manager.add_command(
                script.split('.')[-1], import_string(script + ':manager'))
        for script in managers:
            if script.__name__ == 'invenio.base.manage':
                continue
            manager.add_command(
                script.__name__.split('.')[-2], getattr(script, 'manager'))

    manager.add_command("clean", Clean())
    manager.add_command("show-urls", ShowUrls())
    manager.add_command("shell", Shell())
    parsed_url = urlparse(manager.app.config.get('CFG_SITE_URL'))
    host = manager.app.config.get('SERVER_BIND_ADDRESS', parsed_url.hostname
                                  or '127.0.0.1')
    port = manager.app.config.get('SERVER_BIND_PORT', parsed_url.port or 80)
    runserver = Server(host=host, port=port)
    manager.add_command("runserver", runserver)

    # FIXME separation of concerns is violated here.
    from invenio.ext.collect import collect
    collect.init_script(manager)

    from invenio.ext.assets import command, bower
    manager.add_command("assets", command)
    manager.add_command("bower", bower)
Esempio n. 2
0
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)
Esempio n. 3
0
def main():
    manager.add_command("start", LemurServer())
    manager.add_command("runserver", Server(host='127.0.0.1'))
    manager.add_command("clean", Clean())
    manager.add_command("show_urls", ShowUrls())
    manager.add_command("db", MigrateCommand)
    manager.add_command("init", InitializeApp())
    manager.add_command("create_user", CreateUser())
    manager.add_command("create_role", CreateRole())
    manager.add_command("provision_elb", ProvisionELB())
    manager.run()
Esempio n. 4
0
def main():
    manager.add_command("start", LemurServer())
    manager.add_command("runserver", Server(host='127.0.0.1', threaded=True))
    manager.add_command("clean", Clean())
    manager.add_command("show_urls", ShowUrls())
    manager.add_command("db", MigrateCommand)
    manager.add_command("init", InitializeApp())
    manager.add_command("create_user", CreateUser())
    manager.add_command("reset_password", ResetPassword())
    manager.add_command("create_role", CreateRole())
    manager.add_command("source", source_manager)
    manager.add_command("report", Report())
    manager.add_command("certificate", certificate_manager)
    manager.run()
Esempio n. 5
0
def main():
    # create application
    manager = Manager(inc.create_app)
    # add config, default by dev/dev.py
    manager.add_option('-c', '--config', dest='config', required=False, default='../config/config.py')
    # show the application's url routes
    manager.add_command("urls", ShowUrls())
    # clean .pyc .pyo
    manager.add_command("clean", Clean())
    manager.add_command("init", InitApp())

    try:
        manager.run()
    except KeyboardInterrupt:
        print('KeyboardInterrupt cause quit')
Esempio n. 6
0
def main():
    manager.add_command("start", LemurServer())
    manager.add_command("runserver", Server(host='127.0.0.1', threaded=True))
    manager.add_command("clean", Clean())
    manager.add_command("show_urls", ShowUrls())
    manager.add_command("db", MigrateCommand)
    manager.add_command("init", InitializeApp())
    manager.add_command("create_user", CreateUser())
    manager.add_command("reset_password", ResetPassword())
    manager.add_command("create_role", CreateRole())
    manager.add_command("provision_elb", ProvisionELB())
    manager.add_command("rotate_elbs", RotateELBs())
    manager.add_command("rolling", Rolling())
    manager.add_command("sources", Sources())
    manager.add_command("report", Report())
    manager.run()
Esempio n. 7
0
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  # noqa
    import udata.core.user.commands  # noqa
    import udata.core.dataset.commands  # noqa
    import udata.core.organization.commands  # noqa
    import udata.search.commands  # noqa
    import udata.core.spatial.commands  # noqa
    import udata.core.jobs.commands  # noqa
    import udata.core.badges.commands  # noqa
    import udata.api.commands  # noqa
    import udata.harvest.commands  # noqa
    import udata.features.territories.commands  # noqa
    import udata.linkchecker.commands  # noqa

    # Dynamic module commands loading
    for plugin in manager.app.config['PLUGINS']:
        name = 'udata_{0}.commands'.format(plugin)
        try:
            __import__(name)
        except ImportError as e:
            pass
        except Exception as e:
            log.error('Error during import of %s: %s', name, e)
Esempio n. 8
0
def main():
    manager.add_command("start", LemurServer())
    manager.add_command("runserver", Server(host='127.0.0.1', threaded=True))
    manager.add_command("clean", Clean())
    manager.add_command("show_urls", ShowUrls())
    manager.add_command("db", MigrateCommand)
    manager.add_command("init", InitializeApp())
    manager.add_command("create_user", CreateUser())
    manager.add_command("reset_password", ResetPassword())
    manager.add_command("create_role", CreateRole())
    manager.add_command("source", source_manager)
    manager.add_command("certificate", certificate_manager)
    manager.add_command("notify", notification_manager)
    manager.add_command("endpoint", endpoint_manager)
    manager.add_command("report", report_manager)
    manager.add_command("policy", policy_manager)
    manager.add_command("pending_certs", pending_certificate_manager)
    manager.add_command("dns_providers", dns_provider_manager)
    manager.run()
Esempio n. 9
0
def init_manager(app, db, **kwargs):
    """
    Initialise Manager

    :param app: Flask app object
    :parm db: db instance
    :param kwargs: Additional keyword arguments to be made available as shell context
    """
    manager.app = app
    manager.db = db
    manager.context = kwargs
    manager.add_command('db', MigrateCommand)
    manager.add_command('clean', Clean())
    manager.add_command('showurls', ShowUrls())
    manager.add_command('shell', Shell(make_context=shell_context))
    manager.add_command(
        'plainshell',
        Shell(make_context=shell_context, use_ipython=False,
              use_bpython=False))
    return manager
Esempio n. 10
0
#!/usr/bin/env python
# -*- coding: utf-8 -*-

from flask_script import Manager, Server
from flask_script.commands import Clean, ShowUrls

from rentomatic.app import create_app


app = create_app()
manager = Manager(app)

manager.add_command('server', Server())
manager.add_command('urls', ShowUrls())
manager.add_command('clean', Clean())

if __name__ == '__main__':
    manager.run()
Esempio n. 11
0
# Local Imports
# -----------------------------------------------------------------------------
from webapp.app import app, create_app
from webapp.settings import DevConfig, ProdConfig

HERE = os.path.abspath(os.path.dirname(__file__))
TEST_PATH = os.path.join(HERE, 'tests')

app = create_app(app, DevConfig)

manager = Manager(app)


@manager.command
def test():
    """Run the tests."""
    import pytest
    exit_code = pytest.main([TEST_PATH, '--verbose'])
    return exit_code


# register commands with app manager
# now to run app type:
# $ python manage.py runserver
manager.add_command("runserver", Server())
manager.add_command("show-urls", ShowUrls())
manager.add_command("clean", Clean())

if __name__ == "__main__":
    manager.run()
Esempio n. 12
0
from flask_migrate import Migrate, MigrateCommand
from flask_script import Manager, Shell, Server, Command
from flask_script.commands import Clean, ShowUrls
from app import create_app, db

# app,celery = create_app()
app = create_app()
manager = Manager(app)
migrate = Migrate(app, db)


def make_shell_context():
    from app.modules.admin.model import Admin
    return dict(app=app, db=db, Admin=Admin)


# Get BaoAI version and URL # 获取BaoAI版本及官方URL
@manager.command
def baoai():
    print('BaoAI v2.0.0 - http://www.baoai.co')


manager.add_command("runserver", Server(host='0.0.0.0', port=5000))
manager.add_command("shell", Shell(make_context=make_shell_context))
manager.add_command("db", MigrateCommand)  # Database Manage # 数据库管理
manager.add_command("clean", Clean())  # Clean Cache File # 清理缓存文件
manager.add_command("url", ShowUrls())  # Print All URL # 打印所有URL

if __name__ == "__main__":
    manager.run()
Esempio n. 13
0
from __future__ import absolute_import, print_function

import sys

from flask_script import Manager, prompt_pass
from flask_script.commands import ShowUrls, Clean

from .app import create_app
from .models.auth import User

manager = Manager(create_app())
manager.add_command(ShowUrls())
manager.add_command(Clean())


@manager.command
def initadmin():
    """Creates an initial user."""
    admin_user = User.get_by_name('admin')
    if admin_user:
        print('The user "admin" exists', file=sys.stderr)
        sys.exit(1)

    password = None
    while not password:
        password = prompt_pass('Password', default='').strip()
    admin_user = User.create_normal('admin', password, is_active=True)
    admin_user.grant_admin()


def main():
Esempio n. 14
0
from webtools.tests.script import RunTests


def _make_context():
    return dict(app=app, db=app.db, models=models)


if __name__ == '__main__':
    migrate = Migrate(app, app.db)
    manager = Manager(app, with_default_commands=False)

    manager.add_command('shell', Shell(make_context=_make_context))
    manager.add_command('run_dev_server',
                        Server(use_reloader=True, threaded=True))
    manager.add_command('show_urls', ShowUrls())
    manager.add_command('clean_pyc', Clean())

    manager.add_command('db', MigrateCommand)
    manager.add_command('db_reset', ResetDb)
    manager.add_command('db_command', DatabaseCommand)
    manager.add_command('add_imdb', AddImageDB)
    manager.add_command('clean_models', CleanModels)
    manager.add_command('get_samples', GetSamples)
    manager.add_command('reset_check', ResetCheckedNotAnnotated)

    manager.add_command('clean_waste_images', CleanWasteImages)
    manager.add_command('clean_cache', CleanCache)
    manager.add_command('clean_samples', CleanSamples)

    manager.add_command('create_user', CreateUserCommand)
    manager.add_command('add_role', AddRoleCommand)
Esempio n. 15
0
    my_role = Role(name='admin')
    my_role.add_abilities('create_users', 'delete_users')

    user.add_roles('admin', 'superadmin')

    db.session.add(user)
    db.session.add(my_role)
    db.session.commit()


@manager.command
def upgrade():
    """
    Upgrades the DB
    """
    from {{cookiecutter.app_name}}.upgrade import upgrade
    upgrade()


manager.add_command("server", Server())
manager.add_command("show-urls", ShowUrls())
manager.add_command("clean", Clean())
manager.add_command('shell', Shell(make_context=_make_context))
manager.add_command('db', MigrateCommand)
manager.add_command('urls', ShowUrls())
manager.add_command('clean', Clean())
manager.add_command('lint', Lint())

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