def colored(s, color=None):
if not color:
color = 'yellow'
color = COLOR_MAP[color]
return color + '{}\033[0m'.format(s)
class CommandCanceledException(Exception):
pass
CONFIG = yaml_resources.get('CONFIG')
DB = MySQLHelper(logging)
CACHE_DB = RedisHelper(logging)
ADMIN_USER_ID = 'u-admin'
DB_UPGRADE_SEQ_ID = 'upgrade.db.upgradeSeq'
COMMAND_FUNCS = {}
def command(F):
COMMAND_FUNCS[F.__name__] = F
return F
def confirm():
# -*- coding: utf-8 -*-
# Builtin Modules
import time
# Project Modules
from worker import app
from worker.tasks import BaseTask
from worker.utils import yaml_resources
CONFIG = yaml_resources.get('CONFIG')
IMAGE_INFO = yaml_resources.get('IMAGE_INFO')
@app.task(name='Internal.Shutdown',
bind=True,
base=BaseTask,
ignore_result=False)
def shutdown(self, *args, **kwargs):
'''
Shutdown all workers
'''
return app.control.broadcast('shutdown')
@app.task(name='Internal.Ping', bind=True, base=BaseTask, ignore_result=False)
def ping(self, *args, **kwargs):
'''
Ping workers
'''
return app.control.ping(timeout=0.5)
