def create_submanager(**opts): def sub_opts(**kwargs): pass mgr = Manager(sub_opts, **opts) mgr.__doc__ = opts.get('description') return mgr
""" Management commands for account related activities (including oauth) """ import datetime from adsws.modules.oauth2server.models import OAuthToken, OAuthClient from adsws.core.users import User from adsws.core import db from adsws.accounts import create_app from sqlalchemy import or_, exc, and_ from flask.ext.script import Manager accounts_manager = Manager(create_app()) accounts_manager.__doc__ = __doc__ # Overwrite default docstring def parse_timedelta(s): """ Helper function which converts a string formatted timedelta into a datetime.timedelta object :param s: string formatted timedelta (e.g. "days=1") :return: parsed timedelta :rtype: datetime.timedelta """ td = {s.split('=')[0]: float(s.split('=')[1])} return datetime.timedelta(**td)
""" Management commands for account related activities (including oauth) """ import datetime from adsws.modules.oauth2server.models import OAuthToken, OAuthClient from adsws.core.users import User from adsws.core import db from adsws.accounts import create_app from sqlalchemy import or_, exc from flask.ext.script import Manager accounts_manager = Manager(create_app()) accounts_manager.__doc__ = __doc__ # Overwrite default docstring def parse_timedelta(s): """ Helper function which converts a string formatted timedelta into a datetime.timedelta object :param s: string formatted timedelta (e.g. "days=1") :return: parsed timedelta :rtype: datetime.timedelta """ td = {s.split('=')[0]: float(s.split('=')[1])} return datetime.timedelta(**td) @accounts_manager.command
def say_ok(): print('\x1b[1m\x1b[32mOK\x1b[0m') def quoted_path(): from pipes import quote return quote(os.environ['PATH']) def _celery_opts(app, **kw): pass celery_manager = Manager(_celery_opts) celery_manager.__doc__ = ('Celery sub commands. Runs `{0} celery` ' 'for more help.').format(sys.argv[0]) manager.add_command('celery', celery_manager) def celery_debug(): """Starting a celery worker in console mode""" g = _celery() p = sh.celery.worker(app=g.APP, loglevel=g.DEBUG_LOGLEVEL, pool=g.POOL, _bg=True, **_shkw) safe_wait(p) celery_debug.__name__ = 'debug' celery_manager.command(celery_debug)
import logging from flask.ext.migrate import upgrade from flask.ext.script import Manager from spendb.core import db from spendb.model import Dataset log = logging.getLogger(__name__) manager = Manager() manager.__doc__ = 'Database operations' @manager.command def drop(): """ Drop database """ log.warn("Dropping database") db.metadata.reflect() db.metadata.drop_all() @manager.command def drop_dataset(name): """ Drop a dataset from the database """ log.warn("Dropping dataset '%s'", name) dataset = db.session.query(Dataset).filter_by(name=name).first() if dataset is None: raise Exception("Dataset does not exist: '%s'" % name) dataset.drop() db.session.delete(dataset)