added 'conditions', set message when pirate is drunk (the end)

npc/pirate/actions.py

@@ -65,7 +65,7 @@ def enter(self):
     def update(self, time):
         self.drunkness += 1
         if self.drunkness >= 3:
-            print "DRUNK!"
+            self.parent.set_condition('drunk', True)
             self.finish()
 
 exported_actions = []
@@ -120,7 +120,7 @@ class drink_rum(ActionBuilder):
     """
     def get_actions(self, caller, memory):
         for pct in memory.of_class(PositionPrecept):
-            print "looking for rum", pct
+            #print "looking for rum", pct
             if pct.position[0] == 'self' and pct.entity.name == "rum":
                 action = DrinkRumAction(caller)
                 action.effects.append(SimpleGoal(is_drunk=True))

pygoap/environment.py

@@ -29,13 +29,23 @@ class for objects that agents can interact with
 
     def __init__(self, name='noname'):
         self.name = name
+        self._condition = {}
 
     def get_actions(self, other):
         """
         generate a list of actions that could be used with this object
         """
         return []        
 
+    def condition(self, name):
+        try:
+            return self._condition[name]
+        except KeyError:
+            return False
+
+    def set_condition(self, name, value):
+        self._condition[name] = value
+
     def __repr__(self):
         return "<Object: {}>".format(self.name)
 

test.py

@@ -95,6 +95,11 @@ def run_once():
             wench = Human("Female", "wench")
             formosa.add(wench)
 
+        if time >= 1:
+            if pirate.condition('drunk'):
+                print "YAY!  A drunk pirate is a happy pirate!"
+                print "Test concluded"
+
         screen_buf.fill((0,128,255))
         formosa.render(screen_buf)
         pygame.transform.scale2x(screen_buf, screen)