# -*- coding: utf-8 -*-
from gevent import monkey; monkey.patch_all()
import argparse
import random
import os
import gevent
import gevent.pywsgi
from ws4py.server.geventserver import WebSocketWSGIApplication, \
WebSocketWSGIHandler, WSGIServer
from ws4py.websocket import EchoWebSocket
class BroadcastWebSocket(EchoWebSocket):
def opened(self):
app = self.environ['ws4py.app']
app.clients.append(self)
def received_message(self, m):
# self.clients is set from within the server
# and holds the list of all connected servers
# we can dispatch to
app = self.environ['ws4py.app']
for client in app.clients:
client.send(m)
def closed(self, code, reason="A client left the room without a proper explanation."):
app = self.environ.pop('ws4py.app')
if self in app.clients:
app.clients.remove(self)
for client in app.clients:
try:
client.send(reason)
except:
pass
class EchoWebSocketApplication(object):
def __init__(self, host, port):
self.host = host
self.port = port
self.ws = WebSocketWSGIApplication(handler_cls=BroadcastWebSocket)
# keep track of connected websocket clients
# so that we can brodcasts messages sent by one
# to all of them. Aren't we cool?
self.clients = []
def __call__(self, environ, start_response):
"""
Good ol' WSGI application. This is a simple demo
so I tried to stay away from dependencies.
"""
if environ['PATH_INFO'] == '/favicon.ico':
return self.favicon(environ, start_response)
if environ['PATH_INFO'] == '/ws':
environ['ws4py.app'] = self
return self.ws(environ, start_response)
return self.webapp(environ, start_response)
def favicon(self, environ, start_response):
"""
Don't care about favicon, let's send nothing.
"""
status = '200 OK'
headers = [('Content-type', 'text/plain')]
start_response(status, headers)
return ""
def webapp(self, environ, start_response):
"""
Our main webapp that'll display the chat form
"""
status = '200 OK'
headers = [('Content-type', 'text/html')]
start_response(status, headers)
return """
""" % {'username': "User%d" % random.randint(0, 100),
'host': self.host,
'port': self.port}
if __name__ == '__main__':
from ws4py import configure_logger
configure_logger()
parser = argparse.ArgumentParser(description='Echo gevent Server')
parser.add_argument('--host', default='127.0.0.1')
parser.add_argument('-p', '--port', default=9000, type=int)
args = parser.parse_args()
server = WSGIServer((args.host, args.port), EchoWebSocketApplication(args.host, args.port))
server.serve_forever()