X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D/edit?a=blobdiff_plain;f=plomrogue%2Ftasks.py;h=105002316193b544448cdecabdc2a6410a5b41ef;hb=1bb87f7a1151a6f244974c99cd7c392eeb0a8a35;hp=423758d8f5959ec7ebec088abfa18d7800d28bc1;hpb=6354f0d85ce38a5450142b2bc775e49f0abfc7b8;p=plomrogue2

diff --git a/plomrogue/tasks.py b/plomrogue/tasks.py
index 423758d..1050023 100644
--- a/plomrogue/tasks.py
+++ b/plomrogue/tasks.py
@@ -76,24 +76,25 @@ class Task_FLATTEN_SURROUNDINGS(Task):
 
 
 class Task_PICK_UP(Task):
+    argtypes = 'int:nonneg'
 
     def check(self):
         if self.thing.carrying:
             raise PlayError('already carrying something')
-        nothing_to_pick_up = True
-        for t in [t for t in self.thing.game.things
-                  if t.portable
-                  and t != self.thing and t.position == self.thing.position and
-                  t.type_ != 'Player']:
-            nothing_to_pick_up = False
-            break
-        if nothing_to_pick_up:
-            raise PlayError('nothing to pick up')
+        to_pick_up = self.thing.game.get_thing(self.args[0])
+        if to_pick_up is None:
+            raise PlayError('no thing of ID %s exists %s' % self.args[0])
+        elif to_pick_up == self.thing:
+            raise PlayError('cannot pick up oneself')
+        elif to_pick_up.type_ == 'Player':
+            raise PlayError('cannot pick up player')
+        elif to_pick_up.position != self.thing.position:
+            raise PlayError('thing of ID %s not in reach' % self.args[0])
+        elif not to_pick_up.portable:
+            raise PlayError('thing of ID %s not portable' % self.args[0])
 
     def do(self):
-        to_pick_up = [t for t in self.thing.game.things
-                      if t.portable
-                      and t != self.thing and t.position == self.thing.position][0]
+        to_pick_up = self.thing.game.get_thing(self.args[0])
         self.thing.carrying = to_pick_up
 
 
@@ -148,6 +149,7 @@ class Task_INTOXICATE(Task):
 
     def do(self):
         self.thing.carrying.full = False
+        self.thing.carrying.empty()
         for c_id in self.thing.game.sessions:
             if self.thing.game.sessions[c_id]['thing_id'] == self.thing.id_:
                 self.thing.game.io.send('RANDOM_COLORS', c_id)