def create_submanager(**opts):
def sub_opts(**kwargs):
pass
mgr = Manager(sub_opts, **opts)
mgr.__doc__ = opts.get('description')
return mgr
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)
