1 files changed, 36 insertions, 0 deletions
diff --git a/behaviors/SwitchBehavior.py b/behaviors/SwitchBehavior.py
new file mode 100644
index 0000000..1377b42
--- /dev/null
+++ b/behaviors/SwitchBehavior.py
@@ -0,0 +1,36 @@
+from operationscore.Behavior import *
+import util.ComponentRegistry as compReg
+import json
+
+class SwitchBehavior(Behavior):
+    """
+    SwitchBehavior is a behavior that transform into different behaviors base on the input data.
+    The behavior expects a JSON formatted argument 'PrefixToBehavior' that maps prefixes to behaviors. The behavior detects the prefix on the data and use the corresponding Behavior to process the data and return the outputs.
+    In Config file, include:
+      <PrefixToBehavior>JSON format dict with prefix keys and behavior ID values</PrefixToBehavior>
+      <DefaultBehavior>Default behavior's ID</DefaultBehavior>
+    An example config excerpt:
+      <Behavior>
+        <Class>behaviors.SwitchBehavior</Class>
+          <Args>
+            <Id>switch</Id>
+            <PrefixToBehavior>{'@':'game1', '#':'game2', '$':'game3'}</PrefixToBehavior>
+            <DefaultBehavior>game1</DefaultBehavior>
+          </Args>
+      </Behavior>
+    """
+    def behaviorInit(self):
+        self.defaultBehavior = compReg.getComponent(self['DefaultBehavior'])
+        self.prefixDict = json.loads(self['PrefixToBehavior'])
+        self.currBehavior = None
+        self.setBehavior(self.defaultBehavior)
+        
+    def processResponse(self, sInputs, rInputs):
+        dataStr = sInputs[-1]['Data']
+        if dataStr[0] in self.prefixDict:
+            self.setBehavior(compReg.getComponent(self.prefixDict[dataStr[0]]))
+            sInputs[-1]['Data'] = sInputs[-1]['Data'][1:] # remove prefix
+        return self.currBehavior.processResponse(sInputs, rInputs)
+    
+    def setBehavior(self, behavior):
+        self.currBehavior = behavior