X-Git-Url: https://plomlompom.com/repos//%22https:/validator.w3.org/check?a=blobdiff_plain;f=rogue_chat_curses.py;h=1f93dd808b690d488b0cf9f93fd3ed80a72fa50e;hb=0f1e35f5f1a318d6e3e610085babd46946ad3600;hp=db5f9933640aa92081b0797ed5d7518ab9aee4dd;hpb=c99f3476d21b7416d584c33bfe3b1010ada523e2;p=plomrogue2

diff --git a/rogue_chat_curses.py b/rogue_chat_curses.py
index db5f993..1f93dd8 100755
--- a/rogue_chat_curses.py
+++ b/rogue_chat_curses.py
@@ -625,14 +625,15 @@ class TUI:
         elif self.mode.name == 'take_thing':
             self.log_msg('selectable things:')
             player = self.game.get_thing(self.game.player_id)
-            selectables = [t for t in self.game.things
-                           if t != player and t.type_ != 'Player'
-                           and t.position == player.position]
-            if len(selectables) == 0:
+            self.selectables = [t for t in self.game.things
+                                if t != player and t.type_ != 'Player'
+                                and t.position == player.position]
+            if len(self.selectables) == 0:
                 self.log_msg('none')
             else:
-                for t in selectables:
-                    self.log_msg(str(t.id_) + ' ' + self.get_thing_info(t))
+                for i in range(len(self.selectables)):
+                    t = self.selectables[i]
+                    self.log_msg(str(i) + ': ' + self.get_thing_info(t))
         elif self.mode.name == 'command_thing':
             self.send('TASK:COMMAND ' + quote('HELP'))
         elif self.mode.name == 'control_pw_pw':
@@ -1013,7 +1014,14 @@ class TUI:
                 if self.input_ == '':
                     self.log_msg('@ aborted')
                 else:
-                    self.send('TASK:PICK_UP ' + quote(self.input_))
+                    try:
+                        i = int(self.input_)
+                        if i < 0 or i >= len(self.selectables):
+                            self.log_msg('? invalid index, aborted')
+                        else:
+                            self.send('TASK:PICK_UP %s' % self.selectables[i].id_)
+                    except ValueError:
+                        self.log_msg('? invalid index, aborted')
                 self.input_ = ''
                 self.switch_mode('play')
             elif self.mode.name == 'command_thing' and key == '\n':