def __init__(self):
print "Chat server starting."
# Set up object references.
self.translator = MessageTranslator(self)
ClientHandler.translator = self.translator
# Start thread that always listens for incoming TCP connections.
HOST = ''
PORT = 9999
threadedTCPServer = ThreadedTCPServer((HOST, PORT), ClientHandler)
threadedTCPServerThread = threading.Thread(target=threadedTCPServer.serve_forever)
threadedTCPServerThread.daemon = True
threadedTCPServerThread.start()
print "Chat server running."
class ChatServer():
DEBUG = False
# List of logged in usernames
usernames = []
# List of all chat messages
messageLog = []
# Reference to the message translator
translator = None
def __init__(self):
print "Chat server starting."
# Set up object references.
self.translator = MessageTranslator(self)
ClientHandler.translator = self.translator
# Start thread that always listens for incoming TCP connections.
HOST = ''
PORT = 9999
threadedTCPServer = ThreadedTCPServer((HOST, PORT), ClientHandler)
threadedTCPServerThread = threading.Thread(target=threadedTCPServer.serve_forever)
threadedTCPServerThread.daemon = True
threadedTCPServerThread.start()
print "Chat server running."
# Removes the specified username from the list of usernames.
def removeUsername(self, username):
try:
self.usernames.remove(username)
print "Removed user: "******"Added user: "******"response"] = response
if not error == None:
dic["error"] = error
if not username == None:
dic["username"] = username
if not message == None:
dic["message"] = message
if not messages == None:
dic["messages"] = messages
return dic
# Handles login requests, and sends response back to the message translator.
def requestLogin(self, username):
try:
response = {}
copy = username
if not copy.replace("_", "X").isalnum():
response = self.createMsgDic("login", "Invalid username!", username, None, None)
elif username in self.usernames:
response = self.createMsgDic("login", "Name already taken!", username, None, None)
else:
# Login is accepted.
response = self.createMsgDic("login", None, username, None, self.messageLog)
self.addUsername(username)
return response
except Exception as e:
if self.DEBUG: print "chatServer.requestLogin: Error: " + e.message
# Handles logout requests, and sends response back to the message translator.
def requestLogout(self, username):
try:
response = {}
if not username in self.usernames:
response = self.createMsgDic("logout", "Not logged in!", username, None, None)
else:
# Logout is accepted.
response = self.createMsgDic("logout", None, username, None, None)
self.removeUsername(username)
return response
except Exception as e:
if self.DEBUG: print "chatServer.requestLogout: Error: " + e.message
# Handles requests to broadcast message.
# Sends response back to message translator if there is an error,
# otherwise instructs message translator to send the message to all connections.
def broadcastMessage(self, message, username):
try:
response = None
if not username in self.usernames:
response = self.createMsgDic("message", "Your are not logged in!", None, None, None)
else:
# message is accepted.
loggedMessage = str(datetime.now().hour) + ":" + str(datetime.now().minute) + " " + username + ": " + message
response = self.createMsgDic("message", None, None, loggedMessage, None)
self.messageLog.append(loggedMessage)
print loggedMessage
self.translator.sendMessageToAll(response)
return response
except Exception as e:
if self.DEBUG: print "ChatServer.broadcastMessage: error: " + e.message
