spacepaste

Paste details

reply | raw

  1.  
  2. #!/usr/bin/python
  4. import socket, select
  6. class ChatServer:
  7. def __init__( self, port ):
  8. self.port = port
  10. self.srvsock = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
  11. self.srvsock.setsockopt( socket.SOL_SOCKET, socket.SO_REUSEADDR, 1 )
  12. self.srvsock.bind( ('', port ) )
  13. self.srvsock.listen( 5 )
  15. self.descriptors = [ self.srvsock ]
  16. print 'ChatServer start on port %s' % port
  18. def run( self ):
  19. while 1:
  20. #await an event on a readable socket descriptor
  21. ( sread, swrite, sexec ) = select.select( self.descriptors, [], [] )
  22. #iterate through the tagged read descriptor
  23. for sock in sread:
  24. #received a connect to the server (listening) socket
  25. if sock == self.srvsock:
  26. self.accept_new_connection()
  27. else:
  28. #received something on a client socket
  29. str = sock.recv( 6042 )
  30. #check to see if the peer socket closed
  31. host, port = sock.getpeername()
  32. if str == '':
  33. print 'stopping [%s:%s]\r\n' % ( host, port )
  34. str = '\033[1;31mClient left %s:%s\033[1;m\r\n' % ( host, port )
  35. self.broadcast_string( str, sock )
  36. sock.close()
  37. self.descriptors.remove( sock )
  38. elif '\username' in str:
  39. self.set_username( str, sock, port )
  40. else:
  41. newstr = '\033[1;34m[user:%s] %s\033[1;m' % ( port, str )
  42. self.broadcast_string( newstr, sock )
  44. def broadcast_string( self, str, omit_sock ):
  45. for sock in self.descriptors:
  46. if sock != self.srvsock and sock != omit_sock:
  47. sock.send( str )
  48. print str
  50. def accept_new_connection( self ):
  51. newsock, ( remhost, remport ) = self.srvsock.accept()
  52. self.descriptors.append( newsock )
  54. newsock.send("\033[1;32mYou are now connected to the Poseidon chat server.\r\nPlease set a username by entering the syntax '\username ' followed by your chosen username.\033[1;m\r\n")
  55. str = '\033[1;33mClient joined %s:%s\033[1;m\r\n' % ( remhost, remport )
  56. self.broadcast_string( str, newsock )
  58. def set_username( self, str, sock, port ):
  59. username = str[ str.find(' ')+1: ]
  60. sock.append( {'username': username} )
  61. str = "[user:%s] now known as %s" % ( port, username )
  62. sock.send( str )
  63. self.broadcast_string( str )
  65. if __name__ == "__main__":
  66. myServer = ChatServer( 2626 ).run()
  67.  
Powered by spacepaste. Paste removal.