safe key getting from the response
This commit is contained in:
@@ -24,14 +24,20 @@ class Changer(object):
|
|||||||
'closed': self.close_in_valve
|
'closed': self.close_in_valve
|
||||||
}
|
}
|
||||||
|
|
||||||
|
def safe_remote_state(key):
|
||||||
|
if key in ['out_valve', 'in_valve']:
|
||||||
|
return self.remote_state.get(key, 'closed')
|
||||||
|
else
|
||||||
|
return self.remote_state.get(key,'');
|
||||||
|
|
||||||
def process_change(self):
|
def process_change(self):
|
||||||
self.validate_states()
|
self.validate_states()
|
||||||
|
|
||||||
out_valve_change = self.out_valve_states.get(self.remote_state['out_valve'], None )
|
out_valve_change = self.out_valve_states.get(self.safe_remote_state('out_valve'), None )
|
||||||
if out_valve_change is not None:
|
if out_valve_change is not None:
|
||||||
out_valve_change()
|
out_valve_change()
|
||||||
|
|
||||||
in_valve_change = self.in_valve_states.get(self.remote_state['in_valve'], None )
|
in_valve_change = self.in_valve_states.get(self.safe_remote_state('in_valve'), None )
|
||||||
if in_valve_change is not None:
|
if in_valve_change is not None:
|
||||||
in_valve_change()
|
in_valve_change()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user