diff options
| -rw-r--r-- | resource_manager.py | 50 |
1 files changed, 27 insertions, 23 deletions
diff --git a/resource_manager.py b/resource_manager.py index 1a6e224..b343807 100644 --- a/resource_manager.py +++ b/resource_manager.py @@ -19,6 +19,7 @@ class ResourceManagerBase(object): def fetch(self, resource): assert resource in self.resources, 'Unknown resource {}'.format(resource) + assert 'get' in self.resources[resource], 'Unhandled method' message = ParrotProtocol.getRequest(resource + '/get') result = self.send_message(message) self.resource_values[resource] = result @@ -26,18 +27,21 @@ class ResourceManagerBase(object): def toggle_on(self, resource): assert resource in self.resources, 'Unknown resource {}'.format(resource) + assert 'enable' in self.resources[resource], 'Unhandled method' message = ParrotProtocol.getRequest(resource + '/enable') self.send_message(message) self.fetch(resource) def toggle_off(self, resource): assert resource in self.resources, 'Unknown resource {}'.format(resource) + assert 'disable' in self.resources[resource], 'Unhandled method' message = ParrotProtocol.getRequest(resource + '/disable') self.send_message(message) self.fetch(resource) def set(self, resource, arg): assert resource in self.resources, 'Unknown resource {}'.format(resource) + assert 'set' in self.resources[resource], 'Unhandled method' message = ParrotProtocol.setRequest(resource + '/set', str(arg).lower()) self.send_message(message) self.fetch(resource) @@ -78,7 +82,7 @@ class ResourceManagerBase(object): class GenericResourceManager(ResourceManagerBase): resources = { - '/api/software/version', + '/api/software/version': ['get'], } def __init__(self, sock): @@ -105,31 +109,31 @@ class GenericResourceManager(ResourceManagerBase): class Version1ResourceManager(ResourceManagerBase): resources = { - '/api/software/version', - '/api/system/battery', - '/api/bluetooth/friendlyname', - '/api/system/auto_connection/enabled', - '/api/system/anc_phone_mode/enabled', - '/api/audio/specific_mode/enabled', - '/api/audio/sound_effect/enabled', - '/api/audio/noise_cancellation/enabled', + '/api/software/version': ['get'], + '/api/system/battery': ['get'], + '/api/bluetooth/friendlyname': ['get'], + '/api/system/auto_connection/enabled': ['get', 'set'], + '/api/system/anc_phone_mode/enabled': ['get', 'set'], + '/api/audio/specific_mode/enabled': ['get', 'set'], + '/api/audio/sound_effect/enabled': ['get', 'set'], + '/api/audio/noise_cancellation/enabled': ['get', 'set'], } class Version2ResourceManager(ResourceManagerBase): resources = { - '/api/software/version', - '/api/system/battery', - '/api/system/pi', - '/api/bluetooth/friendlyname', - '/api/system/auto_connection/enabled', - '/api/system/anc_phone_mode/enabled', - '/api/flight_mode', - '/api/audio/sound_effect/enabled', - '/api/audio/sound_effect/room_size', - '/api/audio/sound_effect/angle', - '/api/audio/noise', - '/api/audio/noise_control', - '/api/audio/noise_control/enabled', - '/api/audio/track/metadata', + '/api/software/version': ['get'], + '/api/system/battery': ['get'], + '/api/system/pi': ['get'], + '/api/bluetooth/friendlyname': ['get'], + '/api/system/auto_connection/enabled': ['get', 'set'], + '/api/system/anc_phone_mode/enabled': ['get', 'set'], + '/api/flight_mode': ['get', 'enable', 'disable'], + '/api/audio/sound_effect/enabled': ['get', 'set'], + '/api/audio/sound_effect/room_size': ['get', 'set'], + '/api/audio/sound_effect/angle': ['get', 'set'], + '/api/audio/noise': ['get'], + '/api/audio/noise_control': ['get'], + '/api/audio/noise_control/enabled': ['get', 'set'], + '/api/audio/track/metadata': ['get'], } |