blob: 1377b428c3eb9b342fd05f5f7640557457bd240b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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
|