brainiak-cloud/bin/server/serve at master · brainiak/brainiak-cloud

Repository navigation

#!/usr/bin/env python

import sys

import json

import click

import signal

import subprocess

import platform

from rtcloud import Server

def signal_handler(signal, frame):

print('You pressed Ctrl-C!')

rmq_stop = ['sudo', 'rabbitmqctl', 'stop']

if platform.system() == 'Darwin':

rmq_stop = ['brew', 'services', 'stop', 'rabbitmq']

subprocess.call(rmq_stop)

sys.exit(0)

@click.command(context_settings=dict(help_option_names=["-h", "--help"]))

@click.argument('conf')

def serve(conf):

opts = {}

with open(conf) as json_data:

# TODO: Error handling

# TODO: Check schema and print usage if wrong

opts = json.load(json_data)

signal.signal(signal.SIGINT, signal_handler)

# Assume rabbitmq-server is installed

rmq = ['sudo', 'service', 'rabbitmq-server', 'start']

if platform.system() == 'Darwin':

rmq = ['brew', 'services', 'start', 'rabbitmq']

subprocess.Popen(rmq)

server = Server(opts)

if __name__ == '__main__':

serve()