X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=client_prototype.py;h=2dd666fb36b0117297c687847d9926f310b37b1c;hb=8bd3d9463906cacf18fc8e91b302f87d6d01c59d;hp=b19f50cb771d96e301a425f78cbb72ce2e576d45;hpb=a553633d2be3971ba3a6bab2b00e74aa6c5c308f;p=plomrogue

diff --git a/client_prototype.py b/client_prototype.py
index b19f50c..2dd666f 100644
--- a/client_prototype.py
+++ b/client_prototype.py
@@ -167,41 +167,38 @@ def draw_screen():
     stdscr.refresh()
 
 
-def ping_test(server_answered):
-    half_wait_time = 5
-    if server_answered:
-        ping_test.sent = False
-    elif ping_test.wait_start + half_wait_time < time.time():
-        if not ping_test.sent:
-            io["file_out"].write("PING\n")
-            io["file_out"].flush()
-            ping_test.sent = True
-            ping_test.wait_start = time.time()
-        elif ping_test.sent:
-            raise SystemExit("Server not answering anymore.")
-ping_test.wait_start = 0
 
 
-def read_into_message_queue(string):
-    if string == "":
-        return
-    new_open_end = False
-    if string[-1] is not "\n":
-        new_open_end = True
-    new_messages = string.splitlines()
-    logfile = open("log", "a")
-    logfile.write(str(new_messages) + "\n")
-    logfile.flush()
-    logfile.close()
-    if message_queue["open_end"]:
-        message_queue["messages"][-1] = message_queue["messages"][-1] + new_messages[0]
-        del new_messages[0]
-    message_queue["messages"] = message_queue["messages"] + new_messages
-    if new_open_end:
-        message_queue["open_end"] = True
+def cursed_main(stdscr):
 
+    def ping_test():
+        half_wait_time = 5
+        if len(new_data_from_server) > 0:
+            ping_test.sent = False
+        elif ping_test.wait_start + half_wait_time < time.time():
+            if not ping_test.sent:
+                io["file_out"].write("PING\n")
+                io["file_out"].flush()
+                ping_test.sent = True
+                ping_test.wait_start = time.time()
+            elif ping_test.sent:
+                raise SystemExit("Server not answering anymore.")
+    ping_test.wait_start = 0
+
+    def read_into_message_queue():
+        if new_data_from_server == "":
+            return
+        new_open_end = False
+        if new_data_from_server[-1] is not "\n":
+            new_open_end = True
+        new_messages = new_data_from_server.splitlines()
+        if message_queue["open_end"]:
+            message_queue["messages"][-1] += new_messages[0]
+            del new_messages[0]
+        message_queue["messages"] += new_messages
+        if new_open_end:
+            message_queue["open_end"] = True
 
-def cursed_main(stdscr):
     curses.noecho()
     curses.curs_set(False)
     # stdscr.keypad(True)
@@ -217,13 +214,12 @@ def cursed_main(stdscr):
             draw_screen()
             cursed_main.redraw = False
         char = stdscr.getch()
-        if (char >= 0):
-            if chr(char) in commands:
-                commands[chr(char)]()
-                cursed_main.redraw = True
+        if char >= 0 and chr(char) in commands:
+            commands[chr(char)]()
+            cursed_main.redraw = True
         new_data_from_server = io["file_in"].read()
-        ping_test(len(new_data_from_server) > 0)
-        read_into_message_queue(new_data_from_server)
+        ping_test()
+        read_into_message_queue()
 
 
 def foo():
@@ -277,7 +273,6 @@ except:
 finally:
     logfile = open("log", "a")
     logfile.write(str(message_queue))
-    logfile.flush()
     logfile.close()
     if "file_out" in io:
         io["file_out"].close()