X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=new%2Fexample_client.py;h=4e61c51aaf8679403cbfa34e6837ec7f9bae4413;hb=d78993c5566a9b206ab1ed835ebf7d4d19c66335;hp=3f31780e17332ffa74a768d47ee2416829a99201;hpb=8746c041c61dad2feb5f7c8cf3f103869dc69dbd;p=plomrogue2-experiments

diff --git a/new/example_client.py b/new/example_client.py
index 3f31780..4e61c51 100755
--- a/new/example_client.py
+++ b/new/example_client.py
@@ -433,19 +433,43 @@ class TUI:
         self.item_pointer = 0
         curses.wrapper(self.loop)
 
-    def setup_screen(self, stdscr):
-        self.stdscr = stdscr
-        self.stdscr.refresh()  # will be called by getkey else, clearing screen
-        self.stdscr.timeout(10)
+    def loop(self, stdscr):
 
-    def switch_widgets(self, widget_1, widget_2):
-        widget_1.visible = False
-        widget_2.visible = True
-        x = widget_2.check_updates[0]
-        self.to_update[x] = True
+        def setup_screen(stdscr):
+            self.stdscr = stdscr
+            self.stdscr.refresh()  # will be called by getkey else, clearing screen
+            self.stdscr.timeout(10)
+
+        def switch_widgets(widget_1, widget_2):
+            widget_1.visible = False
+            widget_2.visible = True
+            trigger = widget_2.check_updates[0]
+            self.to_update[trigger] = True
+
+        def pick_or_drop_menu(action_key, widget, selectables, task,
+                              bonus_command=None):
+            if len(selectables) < self.item_pointer + 1 and\
+               self.item_pointer > 0:
+                self.item_pointer = len(selectables) - 1
+            if key == 'c':
+                switch_widgets(widget, map_widget)
+            elif key == 'j':
+                self.item_pointer += 1
+            elif key == 'k' and self.item_pointer > 0:
+                self.item_pointer -= 1
+            elif key == action_key and len(selectables) > 0:
+                id_ = selectables[self.item_pointer]
+                self.socket.send('TASK:%s %s' % (task, id_))
+                if bonus_command:
+                    self.socket.send(bonus_command)
+                if self.item_pointer > 0:
+                    self.item_pointer -= 1
+            else:
+                return
+            trigger = widget.check_updates[0]
+            self.to_update[trigger] = True
 
-    def loop(self, stdscr):
-        self.setup_screen(stdscr)
+        setup_screen(stdscr)
         curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_RED)
         curses.init_pair(2, curses.COLOR_BLACK, curses.COLOR_GREEN)
         curses.init_pair(3, curses.COLOR_BLACK, curses.COLOR_BLUE)
@@ -490,7 +514,7 @@ class TUI:
                 key = self.stdscr.getkey()
                 if key == 'KEY_RESIZE':
                     curses.endwin()
-                    self.setup_screen(curses.initscr())
+                    setup_screen(curses.initscr())
                     for w in top_widgets:
                         w.size = w.size_def
                         w.ensure_freshness(True)
@@ -508,6 +532,16 @@ class TUI:
                         self.socket.send(''.join(self.to_send))
                         self.to_send[:] = []
                         self.to_update['edit'] = True
+                elif key == 't':
+                    if not popup_widget.visible:
+                        self.to_update['popup'] = True
+                        popup_widget.visible = True
+                        popup_widget.reconfigure()
+                        draw_popup_if_visible = True
+                    else:
+                        popup_widget.visible = False
+                        for w in top_widgets:
+                            w.ensure_freshness(True)
                 elif map_widget.visible:
                     if key == 'w':
                         self.socket.send('TASK:MOVE UPLEFT')
@@ -521,64 +555,21 @@ class TUI:
                         self.socket.send('TASK:MOVE DOWNLEFT')
                     elif key == 'c':
                         self.socket.send('TASK:MOVE DOWNRIGHT')
-                    elif key == 't':
-                        if not popup_widget.visible:
-                            self.to_update['popup'] = True
-                            popup_widget.visible = True
-                            popup_widget.reconfigure()
-                            draw_popup_if_visible = True
-                        else:
-                            popup_widget.visible = False
-                            for w in top_widgets:
-                                w.ensure_freshness(True)
                     elif key == 'p':
                         self.socket.send('GET_PICKABLE_ITEMS')
                         self.item_pointer = 0
-                        self.switch_widgets(map_widget, pickable_items_widget)
+                        switch_widgets(map_widget, pickable_items_widget)
                     elif key == 'i':
                         self.item_pointer = 0
-                        self.switch_widgets(map_widget, inventory_widget)
+                        switch_widgets(map_widget, inventory_widget)
                 elif pickable_items_widget.visible:
-                    if len(self.game.world.pickable_items) < self.item_pointer + 1\
-                       and self.item_pointer > 0:
-                        self.item_pointer = len(self.game.world.pickable_items) - 1
-                    while len(self.game.world.pickable_items) <= self.item_pointer:
-                        self.item_pointer -= 1
-                    if key == 'c':
-                        self.switch_widgets(pickable_items_widget, map_widget)
-                    elif key == 'j':
-                        self.item_pointer += 1
-                    elif key == 'k' and self.item_pointer > 0:
-                        self.item_pointer -= 1
-                    elif key == 'p' and \
-                         len(self.game.world.pickable_items) > 0:
-                        id_ = self.game.world.pickable_items[self.item_pointer]
-                        self.socket.send('TASK:PICKUP %s' % id_)
-                        self.socket.send('GET_PICKABLE_ITEMS')
-                        if self.item_pointer > 0:
-                            self.item_pointer -= 1
-                    else:
-                        continue
-                    self.to_update['pickable_items'] = True
+                    pick_or_drop_menu('p', pickable_items_widget,
+                                      self.game.world.pickable_items,
+                                      'PICKUP', 'GET_PICKABLE_ITEMS')
                 elif inventory_widget.visible:
-                    if len(self.game.world.player_inventory) < self.item_pointer + 1\
-                       and self.item_pointer > 0:
-                        self.item_pointer = len(self.game.world.player_inventory) - 1
-                    if key == 'c':
-                        self.switch_widgets(inventory_widget, map_widget)
-                    elif key == 'j':
-                        self.item_pointer += 1
-                    elif key == 'k' and self.item_pointer > 0:
-                        self.item_pointer -= 1
-                    elif key == 'd' and \
-                         len(self.game.world.player_inventory) > 0:
-                        id_ = self.game.world.player_inventory[self.item_pointer]
-                        self.socket.send('TASK:DROP %s' % id_)
-                        if self.item_pointer > 0:
-                            self.item_pointer -= 1
-                    else:
-                        continue
-                    self.to_update['inventory'] = True
+                    pick_or_drop_menu('d', inventory_widget,
+                                      self.game.world.player_inventory,
+                                      'DROP')
             except curses.error:
                 pass
             if self.game.do_quit: