client.py

# -*- coding: utf-8 -*-
import sys
if sys.version_info[0] != 3:
    print("You need to run this with Python 3!")
    sys.exit(1)

'''
KTN-project 2013 / 2014
'''
import socket
from Message import *
from MessageWorker import MessageWorker
import threading
import re
import time
from _io import StringIO



class Client(object):
    def __init__(self, host, port):
        self.connection = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        self.connection.connect((host, port))
 
        self.login_response_event = threading.Event()

        # Start the message worker (which listens on the connection and notifies us if it has received a message)
        self.message_worker = MessageWorker(self.connection, self)
        self.message_worker.start()


        self.run = True


    # Main loop
    def start(self):
        self.username = None

        # Log in
        while self.username is None:
            self.login_response_event.clear()

            new_username = self.input("Enter username: ")

            if self.valid_username(new_username):
                loginRequestMessage = LoginRequestMessage()
                loginRequestMessage.set_login_info(new_username)
                self.send(loginRequestMessage)

                self.login_response_event.wait()  # Blocks until message_received receives a LoginResponseMessage

            else:
                self.output("Client: Invalid username!")


        self.output("Logged in as " + self.username)

        while self.run:
            new_message = self.input("> ")

            # Parse as cmd
            cmd = self.get_cmd(new_message)

            if not cmd:
                now_string = time.strftime("%H:%M:%S")
                self.output(self.username + " " + now_string + ": " + new_message)
                chatRequestMessage = ChatRequestMessage()
                chatRequestMessage.set_chat_message(new_message)
                self.send(chatRequestMessage)

            else:
                if cmd == "logout":
                    logoutRequestMessage = LogoutRequestMessage()
                    self.send(logoutRequestMessage)
                    self.run = False

                elif cmd == "listusers":
                    listUsersRequestMessage = ListUsersRequestMessage()
                    self.send(listUsersRequestMessage)

                else:
                    self.output("Unknown command: /" + cmd)

        self.output("Logged out, good bye!")

    def get_cmd(self, text):
        if len(text) > 1 and text[0] == '/':
            return text[1:].lower()
        else:
            return False


    # Message is already decoded from JSON
    def message_received(self, data):
        #print "Message received from server: " + str(data)
        if "response" in data:
            if data["response"] == "login":
                # Success
                if "error" not in data:
                    self.username = data["username"]

                    for message in data["messages"]:
                        self.output(message, True)


                elif data["error"] == "Invalid username!":
                    self.output("Invalid username!")


                elif data["error"] == "Name already taken!":
                    self.output("Name already taken!")

                # Notify main thread
                self.login_response_event.set()

            elif data["response"] == "message":
                if not "error" in data:
                    self.output(data["message"], True)

                # Error: not logged in
                else:
                    self.output(data["error"])

            elif data["response"] == "listUsers":
                users_online_string = "Online users: " + ", ".join(data["users"])
                self.output(users_online_string)

        else:
            self.output("Server makes no sense, me don't understand!")

    def valid_username(self, username):
        match_obj = re.search(u'[A-zæøåÆØÅ_0-9]+', username)
        return match_obj is not None and match_obj.group(0) == username

    # Server closed connection
    def connection_closed(self):
        self.message_worker.join()  # Wait for listener to exit

    # data should be a Message object
    def send(self, data):
        self.connection.sendall(data.get_JSON().encode("UTF-8"))

    def force_disconnect(self):
        self.connection.close()

    # Output this to console
    def output(self, line, newline=False):
        if newline:
            print("\r" + line)
        else:
            print("\r" + line, end="")

    # Get input
    def input(self, prompt):
        print("\r" + prompt)
        return sys.stdin.readline().strip()



if __name__ == "__main__":
    client = Client('www.furic.pw', 9999)
    client.start()