Chat_V1.0.py

import socket
import sys
import threading
import subprocess
import struct
import pickle

#test branch

class Chat:
    def __init__(self):
        self.__pseudo = input("enter username: \n")
        self.__port = None
        self.__ip = None
        self._lsUser = {}
        self._handlers = {
            '/exit': self._exit,
            '/quit': self._quit,
            '/join': self._join,
            '/send': self._send,
            '/address': self._client,
            '/user': self._user,
            '/server': self._server,
        }

    def run(self):
        self.__address = None
        self.__running = True
        threading.Thread(target=self._receive).start()

        while self.__running:
            line = sys.stdin.readline().rstrip() + ' '
            # Extract the command and the param
            command = line[:line.index(' ')]
            param = line[line.index(' ') + 1:].rstrip()
            # Call the command handler
            if command in self._handlers:
                try:
                    self._handlers[command]() if param == '' else self._handlers[command](param)
                except:
                    print("Erreur lors de l'exécution de la commande.")
            else:
                print('Command inconnue:', command)

    def _chat(self, host=socket.gethostname(), port=5000):
        s = socket.socket(type=socket.SOCK_DGRAM)
        s.settimeout(0.5)
        s.bind((host, port))
        self.__s = s
        print('Écoute sur {}:{}'.format(host, port))

    def _exit(self):
        self._send("disconnect")
        self.__running = False
        self.__address = None
        self.__s.close()

    def _quit(self):
        self.__address = None

    def _join(self, param):
        tokens = param.split(' ')
        if len(tokens) == 2:
            try:
                self.__address = (socket.gethostbyaddr(tokens[0])[0], int(tokens[1]))
                print('Connecté à {}:{}'.format(*self.__address))
            except OSError:
                print("Erreur lors de l'envoi du message.")

    def _send(self, param):
        if self.__address is not None:
            try:
                message = param.encode()
                totalsent = 0
                while totalsent < len(message):
                    sent = self.__s.sendto(message[totalsent:], self.__address)
                    totalsent += sent
            except OSError:
                print('Erreur lors de la réception du message.')

    def _receive(self):
        while self.__running:
            try:
                data, address = self.__s.recvfrom(1024)
                print(data.decode())
            except socket.timeout:
                pass
            except OSError:
                return

    def _client(self):
        print(self.__address)

    def _user(self):
        self.__user = subprocess.Popen(["whoami"], stdout=subprocess.PIPE)
        print(self.__user.communicate()[0].decode().rstrip())

if __name__ == '__main__':
    if len(sys.argv) == 3:
        Chat().run()
    else:
        Chat().run()