Add wearable hats.

[plomrogue2] / plomrogue / tasks.py

diff --git a/plomrogue/tasks.py b/plomrogue/tasks.py
index 9165cd44b63d0adece3c1da95366cdbd7b5fae4a..9fccbc33450f2c50fe5caf6269d0ba171bc7e644 100644 (file)
--- a/plomrogue/tasks.py
+++ b/plomrogue/tasks.py
@@ -202,3 +202,29 @@ class Task_INSTALL(Task):
         else:
             self._get_uninstallables()[0].uninstall()
             self.thing.send_msg('CHAT "You uninstall the thing here."')
+
+
+
+class Task_WEAR(Task):
+
+    def check(self):
+        if self.thing.name in self.thing.game.hats:
+            return
+        if not self.thing.carrying:
+            raise PlayError('carrying nothing to wear')
+        if self.thing.name in self.thing.game.hats:
+            raise PlayError('already wearing a hat')
+        if self.thing.carrying.type_ != 'Hat':
+            raise PlayError('can only wear a hat')
+
+    def do(self):
+        if self.thing.name in self.thing.game.hats:
+            t = self.thing.game.thing_types['Hat'](self.thing.game,
+                                                   position=self.thing.position)
+            self.thing.game.things += [t]
+            t.design = self.thing.game.hats[self.thing.name]
+            del self.thing.game.hats[self.thing.name]
+        else:
+            self.thing.game.hats[self.thing.name] = self.thing.carrying.design
+            self.thing.game.things.remove(self.thing.carrying)
+            self.thing.carrying = None