from operationscore.Behavior import *
from logger import main_log
class ExpandingColorZones(Behavior):
    def behaviorInit(self):
        self.mapping = {'s001':[(132,0),(255,0,0)], 's002':[(400,0), (0,255,0)],
                's003':[(668,0),
            (0,0,255)]}
        self.mappingkey = 'data'
    def processResponse(self, sensorInputs, recursiveInputs):
        ret = []
        for data in sensorInputs:
            print data
            data = dict(data)
            if self.mappingkey in data:
                try:
                    data['Location'], data['Color'] =\
                        self.mapping[data[self.mappingkey]] 
                    ret.append(data)
                except:
                    main_log.warn('Bad mapping key.  Expanding Color Zones.')
        return (ret,[])