client.Client.fetch_new_messages()

Complexity

Conditions

7

Size

Total Lines	38
Code Lines	29

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

import errno
    def fetch_new_messages(self):
        if len(self.send_message):
            message = self.send_message.encode("utf-8")
            message_header = f"{len(message) :< {HEADER_LENGTH}}".encode("utf-8")
            self.socket.send(message_header + message)
            self.send_message = ''

        new_messages = []

        try:
            while True:
                username_header = self.socket.recv(HEADER_LENGTH)
                if not len(username_header):
                    print("Connection closed by the server")
                    sys.exit()

                username_length = int(username_header.decode("utf-8").strip())
                username = self.socket.recv(username_length).decode("utf-8")

                message_header = self.socket.recv(HEADER_LENGTH)
                message_length = int(message_header.decode("utf-8").strip())
                message = self.socket.recv(message_length).decode("utf-8")

                new_messages.append({
                    'author': username,
                    'content': message
                })

        except IOError as e:
            if e.errno not in [errno.EAGAIN, errno.EWOULDBLOCK]:
                print("reading error", str(e))
                sys.exit()

            return new_messages

        except Exception as e:
            print("general error", str(e))
            sys.exit()