Changeset 32:0e49648f8d74

Timestamp:

Sep 1, 2013, 2:39:13 PM (7 years ago)

Author:

Jeremy Thurgood <firxen@…>

Branch:

default

Message:

Arbitrary function condition.

Location:

nagslang

Files:

• environment.py (modified) (6 diffs)
• tests/test_environment.py (modified) (1 diff)

nagslang/environment.py

@@ -10 +10 @@
 
 class AllConditions(ProtagonistCondition):
+    """Condition that is met if all provided conditions are met.
+
+    Conditions are evaluated lazily, so the first condition to fail stops
+    processing.
+    """
     def __init__(self, *conditions):
         self.conditions = conditions
@@ -21 +26 @@
 
 class AnyCondition(ProtagonistCondition):
+    """Condition that is met if any provided condition is met.
+
+    Conditions are evaluated lazily, so the first condition to pass stops
+    processing.
+    """
     def __init__(self, *conditions):
         self.conditions = conditions
@@ -32 +42 @@
 
 class WolfFormCondition(ProtagonistCondition):
+    """Condition that is met if the protagonist is in wolf form.
+    """
     def check(self, protagonist):
         return protagonist.in_wolf_form()
@@ -37 +49 @@
 
 class HumanFormCondition(ProtagonistCondition):
+    """Condition that is met if the protagonist is in human form.
+    """
     def check(self, protagonist):
         return protagonist.in_human_form()
@@ -42 +56 @@
 
 class ItemRequiredCondition(ProtagonistCondition):
+    """Condition that is met if the protagonist has the required item.
+    """
     def __init__(self, required_item):
         self.required_item = required_item
@@ -47 +63 @@
     def check(self, protagonist):
         return protagonist.has_item(self.required_item)
+
+
+class FunctionCondition(ProtagonistCondition):
+    """Condition that is met if the provided function returns `True`.
+    """
+    def __init__(self, func):
+        self.func = func
+
+    def check(self, protagonist):
+        return self.func(protagonist)
 
 

nagslang/tests/test_environment.py

@@ -89 +89 @@
         self.assert_not_met(item, FakeProtagonist(item=False))
 
+    def test_function_condition(self):
+        condition = environment.FunctionCondition(lambda p: p.startswith('x'))
+        self.assert_met(condition, 'xxx')
+        self.assert_not_met(condition, 'yyy')
+
 
 class TestActions(TestCase):