63 lines
2 KiB
Python
63 lines
2 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.'
|