From afa654af0a2d2e6b34b47af52ed214977b7a3e12 Mon Sep 17 00:00:00 2001
From: Marek Siarkowicz <mareksiarkowicz@gmail.com>
Date: Mon, 15 Jun 2015 13:56:44 +0200
Subject: Refactor. Move.

---
 parrot_zik/model/version2.py | 118 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 118 insertions(+)
 create mode 100644 parrot_zik/model/version2.py

(limited to 'parrot_zik/model/version2.py')

diff --git a/parrot_zik/model/version2.py b/parrot_zik/model/version2.py
new file mode 100644
index 0000000..9ef6267
--- /dev/null
+++ b/parrot_zik/model/version2.py
@@ -0,0 +1,118 @@
+from .base import ParrotZikBase
+from ..resource_manager import Version2ResourceManager
+
+
+class ParrotZikVersion2(ParrotZikBase):
+    def __init__(self, resource_manager):
+        super(ParrotZikVersion2, self).__init__(
+            resource_manager.get_resource_manager(
+                Version2ResourceManager))
+
+    @property
+    def version(self):
+        answer = self.resource_manager.get('/api/software/version')
+        return answer.software['sip6']
+
+    @property
+    def battery_level(self):
+        return self.get_battery_level('percent')
+
+    @property
+    def flight_mode(self):
+        answer = self.resource_manager.get('/api/flight_mode')
+        return self._result_to_bool(answer.flight_mode['enabled'])
+
+    @flight_mode.setter
+    def flight_mode(self, arg):
+        if arg:
+            self.resource_manager.toggle_on('/api/flight_mode')
+        else:
+            self.resource_manager.toggle_off('/api/flight_mode')
+
+    @property
+    def sound_effect(self):
+        answer = self.resource_manager.get('/api/audio/sound_effect/enabled')
+        return self._result_to_bool(answer.audio.sound_effect['enabled'])
+
+    @sound_effect.setter
+    def sound_effect(self, arg):
+        self.resource_manager.set('/api/audio/sound_effect/enabled', arg)
+
+    @property
+    def room(self):
+        answer = self.resource_manager.get('/api/audio/sound_effect/room_size')
+        return answer.audio.sound_effect['room_size']
+
+    @room.setter
+    def room(self, arg):
+        self.resource_manager.set('/api/audio/sound_effect/room_size', arg)
+
+    @property
+    def external_noise(self):
+        answer = self.resource_manager.get('/api/audio/noise')
+        return int(answer.audio.noise['external'])
+
+    @property
+    def internal_noise(self):
+        answer = self.resource_manager.get('/api/audio/noise')
+        return int(answer.audio.noise['internal'])
+
+    @property
+    def angle(self):
+        answer = self.resource_manager.get('/api/audio/sound_effect/angle')
+        return int(answer.audio.sound_effect['angle'])
+
+    @angle.setter
+    def angle(self, arg):
+        self.resource_manager.set('/api/audio/sound_effect/angle', arg)
+
+    @property
+    def noise_control(self):
+        answer = self.resource_manager.get('/api/audio/noise_control')
+        return NoiseControl.from_noise_control(answer.audio.noise_control)
+
+    @noise_control.setter
+    def noise_control(self, arg):
+        pass
+
+    @property
+    def noise_control_enabled(self):
+        answer = self.resource_manager.get('/api/audio/noise_control/enabled')
+        return self._result_to_bool(answer.audio.noise_control['enabled'])
+
+
+class NoiseControl(object):
+    def __init__(self, type, value):
+        self.type = type
+        self.value = value
+
+    @classmethod
+    def from_noise_control(cls, noise_control):
+        return cls(noise_control['type'], int(noise_control['value']))
+
+    def __eq__(self, other):
+        return self.type == other.type and self.value == other.value
+
+    def __str__(self):
+        return '{}++{}'.format(self.type, self.value)
+
+
+class NoiseControlTypes:
+    NOISE_CONTROL_MAX = NoiseControl('anc', 2)
+    NOISE_CONTROL_ON = NoiseControl('anc', 1)
+    NOISE_CONTROL_OFF = NoiseControl('off', 1)
+    STREET_MODE = NoiseControl('aoc', 1)
+    STREET_MODE_MAX = NoiseControl('aoc', 2)
+
+
+class Rooms:
+    CONCERT_HALL = 'concert'
+    JAZZ_CLUB = 'jazz'
+    LIVING_ROOM = 'living'
+    SILENT_ROOM = 'silent'
+    representation = {
+        CONCERT_HALL: 'Concert Hall',
+        JAZZ_CLUB: 'Jazz Club',
+        LIVING_ROOM: 'Living Room',
+        SILENT_ROOM: 'Silent Room',
+    }
\ No newline at end of file
-- 
cgit v1.2.1