Class MsgServer

PATH/core/msg_server.py:17–65

parser.add_argument('--timeout', type=int)

class MsgServer(object):
    def __init__(self, server_ip, server_port):
        self.init_chat()
        self.send('server chat logged in!')
        self.start_server(server_ip, server_port)

    def init_chat(self):
        itchat.auto_login(enableCmdQR=2)

    def send(self, msg, echo=True):
        if echo:
            print(msg)
        itchat.send(msg, toUserName='filehelper')

    def worker_thread(self, conn, addr):
        conn.settimeout(args.timeout)
        if conn is not None:
            self.send('job connected! [{}]'.format(addr))
        else:
            self.send('none connection!')
            return -1

        while(True):
            try:
                recv_data = conn.recv(1024)
            except socket.timeout as e:
                print('no msg...')
            else:
                msg_len = len(recv_data)
                if msg_len == 0:
                    self.send('connection break, waiting for other connections..')
                    break
                self.send(str(recv_data, encoding = 'utf-8'))
        conn.close()

    def start_server(self, server_ip, server_port):
        ip_port = (server_ip, server_port)
        s = socket.socket()
        s.bind(ip_port)

        # dump ip/port info to file
        with open('server.txt', 'w') as f:
            f.write('{} {}\n'.format(server_ip, server_port))

        s.listen()
        self.send('server listening on {}, waiting for job connection...'.format(server_ip))
        while(True):
            conn, addr = s.accept()
            threading.Thread(target=self.worker_thread, args=(conn, addr)).start()

class MsgClient(object):
    def __init__(self, server_ip, server_port):