#!/usr/bin/python """ This is a simple websocket relay server that facilitates realtime visualization of GraphStates. It doesn't do much except echo websocket messages to all connected clients, and serve some static content over HTTP. """ from websocket_server import WebsocketServer from SimpleHTTPServer import SimpleHTTPRequestHandler from BaseHTTPServer import HTTPServer from SocketServer import ThreadingMixIn import os, sys, threading import webbrowser import argparse import abp import json from pkg_resources import resource_filename clients = [] def new_message(client, server, message): if message.startswith("edit"): edit = json.loads(message[5:]) print "Received update from javascript." print edit else: print "Received update from python {}.".format(client["id"]) server.send_message_to_all(message) def new_client(client, server): print "Client {} connected.".format(client["id"]) clients.append(client) def client_left(client, server): print "Client {} disconnected.".format(client["id"]) clients.remove(client) class ThreadedHTTPServer(ThreadingMixIn, HTTPServer): """ Handle requests in a separate thread """ if __name__ == '__main__': parser = argparse.ArgumentParser(description = "ABP websocket server") parser.add_argument("-v", action="store_true", help="Launch browser") args = parser.parse_args() # Change to the right working dir where = os.path.abspath(resource_filename("abp.static", "")) os.chdir(where) # Start the HTTP server httpserver = ThreadedHTTPServer(('', 5001), SimpleHTTPRequestHandler) thread = threading.Thread(target = httpserver.serve_forever) thread.daemon = True thread.start() if args.v: webbrowser.open("http://localhost:5001/") # Start the websocket server server = WebsocketServer(5000) server.set_fn_new_client(new_client) server.set_fn_message_received(new_message) server.set_fn_client_left(client_left) server.run_forever() httpserver.shutdown()