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

63 lines
2.0 KiB
Python

from django.core.management.base import BaseCommand
from orchestra import settings
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):
valid_codes = (0,1) if optional else (0,)
e = run('service %s %s' % (service, action), valid_codes=valid_codes, display=True)
if e.exit_code == 1:
return False
return True
def flatten(nested, depth=0):
if isinstance(nested, (list, tuple)):
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)
def add_arguments(self, parser):
for service in flatten(self.services):
parser.add_argument(
'--no-%s' % service,
action='store_false',
dest=service,
default=True,
help='Do not %s %s' % (self.action, service)
)
@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 = settings.ORCHESTRA_START_SERVICES
action = 'start'
help = 'Start all related services. Usefull for reload configuration and files.'