|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- '''
- The MIT License (MIT)
- Copyright (c) 2013 Dave P.
- '''
-
- import signal
- import sys
- import ssl
- from SimpleWebSocketServer import WebSocket, SimpleWebSocketServer, SimpleSSLWebSocketServer
- from optparse import OptionParser
-
- class SimpleEcho(WebSocket):
-
- def handleMessage(self):
- print self.data
- self.sendMessage(self.data)
-
- def handleConnected(self):
- pass
-
- def handleClose(self):
- pass
-
- clients = []
- class SimpleChat(WebSocket):
-
- def handleMessage(self):
- for client in clients:
- if client != self:
- client.sendMessage(self.address[0] + u' - ' + self.data)
-
- def handleConnected(self):
- print (self.address, 'connected')
- for client in clients:
- client.sendMessage(self.address[0] + u' - connected')
- clients.append(self)
-
- def handleClose(self):
- clients.remove(self)
- print (self.address, 'closed')
- for client in clients:
- client.sendMessage(self.address[0] + u' - disconnected')
-
-
- if __name__ == "__main__":
-
- parser = OptionParser(usage="usage: %prog [options]", version="%prog 1.0")
- parser.add_option("--host", default='', type='string', action="store", dest="host", help="hostname (localhost)")
- parser.add_option("--port", default=8000, type='int', action="store", dest="port", help="port (8000)")
- parser.add_option("--example", default='echo', type='string', action="store", dest="example", help="echo, chat")
- parser.add_option("--ssl", default=0, type='int', action="store", dest="ssl", help="ssl (1: on, 0: off (default))")
- parser.add_option("--cert", default='./cert.pem', type='string', action="store", dest="cert", help="cert (./cert.pem)")
- parser.add_option("--ver", default=ssl.PROTOCOL_TLSv1, type=int, action="store", dest="ver", help="ssl version")
-
- (options, args) = parser.parse_args()
-
- cls = SimpleEcho
- if options.example == 'chat':
- cls = SimpleChat
-
- if options.ssl == 1:
- server = SimpleSSLWebSocketServer(options.host, options.port, cls, options.cert, options.cert, version=options.ver)
- else:
- server = SimpleWebSocketServer(options.host, options.port, cls)
-
- def close_sig_handler(signal, frame):
- server.close()
- sys.exit()
-
- signal.signal(signal.SIGINT, close_sig_handler)
-
- server.serveforever()
|