diff --git a/libtoorchat.py b/libtoorchat.py index a87f147..0a9dca3 100644 --- a/libtoorchat.py +++ b/libtoorchat.py @@ -10,7 +10,6 @@ class ToorChatProtocol(): ''' This is used to send a simple message over the toorchat protocol ''' msg = ToorChatMessage(message, user) self.device.RFxmit(msg.to_string()) - print msg.to_string() @classmethod def parse_message(cls, raw_message): diff --git a/toorchat.py b/toorchat.py index 9117d95..527945d 100755 --- a/toorchat.py +++ b/toorchat.py @@ -33,6 +33,7 @@ class Visualizer(): self.badge.setModeRX() self.protocol = ToorChatProtocol(self.badge) self.message_queue = [] + self.user = None #This when set to True will kill the thread self.exit = False @@ -53,10 +54,11 @@ class Visualizer(): last_message_index = 0 while True: self.screen_max_y, self.screen_max_x = self.screen.getmaxyx() - self.screen.addstr(0, 1, "[S] Send Message ") + self.screen.addstr(0, 1, "[S] Send Message [U] Set User Name") if len(self.message_queue) > 0: - message = time.ctime()+":"+ str(self.message_queue[len(self.message_queue)-1].data) - self.screen.addstr(last_message_index+3,1, message + " "*(self.screen_max_x-(2+len(message)))) + message = self.message_queue[len(self.message_queue)-1] + message_string = str(message.user)+":"+ str(message.data) + self.screen.addstr(last_message_index+3,1, message_string + " "*(self.screen_max_x-(2+len(message_string)))) print self.screen_max_x last_message_index +=1 self.message_queue.pop() @@ -68,10 +70,15 @@ class Visualizer(): if entry == ord('s'): self.screen.nodelay(0) user_input = self.screen.getstr(1, 1, 60) - self.protocol.send_message(user_input) + self.protocol.send_message(user_input, self.user) self.screen.nodelay(1) - self.__draw_frame__() - self.screen.refresh() + self.screen.addstr(1,1," "*(self.screen_max_x-3)) + if entry == ord('u'): + self.screen.nodelay(0) + user_input = self.screen.getstr(1, 1, 60) + self.user = user_input + self.screen.nodelay(1) + self.screen.addstr(1,1," "*(self.screen_max_x-3)) except KeyboardInterrupt: self.exit = True