django-orchestra-test/orchestra/management/commands/startservices.py

60 lines
2.0 KiB
Python

from optparse import make_option
from django.core.management.base import BaseCommand
from orchestra.settings import ORCHESTRA_START_SERVICES
from orchestra.utils.sys import run, check_root
def run_tuple(services, action, options, optional=False):
if isinstance(services, str):
services = [services]
for service in services:
if options.get(service):
error_codes = [0,1] if optional else [0]
e = run('service %s %s' % (service, action), error_codes=error_codes)
if e.return_code == 1:
return False
return True
def flatten(nested, depth=0):
if hasattr(nested, '__iter__'):
for sublist in nested:
for element in flatten(sublist, depth+1):
yield element
else:
yield nested
class ManageServiceCommand(BaseCommand):
def __init__(self, *args, **kwargs):
super(ManageServiceCommand, self).__init__(*args, **kwargs)
self.option_list = BaseCommand.option_list + tuple(
make_option('--no-%s' % service, action='store_false', dest=service, default=True,
help='Do not %s %s' % (self.action, service)) for service in flatten(self.services)
)
@check_root
def handle(self, *args, **options):
for service in self.services:
if isinstance(service, str):
run_tuple(service, self.action, options)
else:
failure = True
for opt_service in service:
if run_tuple(opt_service, self.action, options, optional=True):
failure = False
break
if failure:
str_services = [ str(s) for s in service ]
self.stderr.write('Error %sing %s' % (self.action, ' OR '.join(str_services)))
class Command(ManageServiceCommand):
services = ORCHESTRA_START_SERVICES
action = 'start'
option_list = BaseCommand.option_list
help = 'Start all related services. Usefull for reload configuration and files.'