initial commit

Author: zhelev

afsapi/__init__.py

@@ -0,0 +1,371 @@
+"""
+Implements an asynchronous interface for a Frontier Silicon device.
+For example internet radios from: Medion, Hama, Auna, ...
+"""
+import asyncio
+import aiohttp
+from lxml import objectify
+
+ # pylint: disable=R0904
+class AFSAPI():
+    """Builds the interface to a Frontier Silicon device."""
+
+    # states
+    PLAY_STATES = {
+        0: 'stopped',
+        1: 'unknown',
+        2: 'playing',
+        3: 'paused',
+    }
+
+    # implemented API calls
+    API = {
+        # sys
+        'friendly_name': 'netRemote.sys.info.friendlyName',
+        'power': 'netRemote.sys.power',
+        'mode': 'netRemote.sys.mode',
+        'valid_modes': 'netRemote.sys.caps.validModes',
+        'equalisers': 'netRemote.sys.caps.eqPresets',
+        'sleep': 'netRemote.sys.sleep',
+        # volume
+        'volume_steps': 'netRemote.sys.caps.volumeSteps',
+        'volume': 'netRemote.sys.audio.volume',
+        'mute': 'netRemote.sys.audio.mute',
+        # play
+        'status': 'netRemote.play.status',
+        'name': 'netRemote.play.info.name',
+        'control': 'netRemote.play.control',
+        # info
+        'text': 'netRemote.play.info.text',
+        'artist': 'netRemote.play.info.artist',
+        'album': 'netRemote.play.info.album',
+        'graphic_uri': 'netRemote.play.info.graphicUri',
+        'duration': 'netRemote.play.info.duration',
+    }
+
+    def __init__(self, fsapi_device_url, pin):
+        self.fsapi_device_url = fsapi_device_url
+        self.pin = pin
+        self.sid = None
+        self.__webfsapi = None
+        self.__modes = None
+        self.__volume_steps = None
+        self.__equalisers = None
+        self.__session = aiohttp.ClientSession()
+
+    def __del__(self):
+        self.call('DELETE_SESSION')
+        self.__session.close()
+
+    # http request helpers
+
+    @asyncio.coroutine
+    def get_fsapi_endpoint(self):
+        """Parse the fsapi endpoint from the device url."""
+        endpoint = yield from self.__session.get(self.fsapi_device_url)
+        text = yield from endpoint.text(encoding='utf-8')
+        doc = objectify.fromstring(text)
+        return doc.webfsapi.text
+
+    @asyncio.coroutine
+    def create_session(self):
+        """Create a session on the frontier silicon device."""
+        req_url = '%s/%s' % (self.__webfsapi, 'CREATE_SESSION')
+        sid = yield from self.__session.get(req_url, params=dict(pin=self.pin))
+        text = yield from sid.text(encoding='utf-8')
+        doc = objectify.fromstring(text)
+        return doc.sessionId.text
+
+    @asyncio.coroutine
+    def call(self, path, extra=None):
+        """Execute a frontier silicon API call."""
+        if not self.__webfsapi:
+            self.__webfsapi = yield from self.get_fsapi_endpoint()
+
+        if not self.sid:
+            self.sid = yield from self.create_session()
+
+        if not isinstance(extra, dict):
+            extra = dict()
+
+        params = dict(pin=self.pin, sid=self.sid)
+        params.update(**extra)
+
+        req_url = ('%s/%s' % (self.__webfsapi, path))
+        result = yield from self.__session.get(req_url, params=params)
+        if result.status == 200:
+            text = yield from result.text(encoding='utf-8')
+        else:
+            self.sid = yield from self.create_session()
+            params = dict(pin=self.pin, sid=self.sid)
+            params.update(**extra)
+            result = yield from self.__session.get(req_url, params=params)
+            text = yield from result.text(encoding='utf-8')
+
+        return objectify.fromstring(text)
+
+    # Helper methods
+
+    # Handlers
+    @asyncio.coroutine
+    def handle_get(self, item):
+        """Helper method for reading a value by using the fsapi API."""
+        res = yield from self.call('GET/{}'.format(item))
+        return res
+
+    @asyncio.coroutine
+    def handle_set(self, item, value):
+        """Helper method for setting a value by using the fsapi API."""
+        doc = yield from self.call('SET/{}'.format(item), dict(value=value))
+        if doc is None:
+            return None
+
+        return doc.status == 'FS_OK'
+
+    @asyncio.coroutine
+    def handle_text(self, item):
+        """Helper method for fetching a text value."""
+        doc = yield from self.handle_get(item)
+        if doc is None:
+            return None
+
+        return doc.value.c8_array.text or None
+
+    @asyncio.coroutine
+    def handle_int(self, item):
+        """Helper method for fetching a integer value."""
+        doc = yield from self.handle_get(item)
+        if doc is None:
+            return None
+
+        return int(doc.value.u8.text) or None
+
+    # returns an int, assuming the value does not exceed 8 bits
+    @asyncio.coroutine
+    def handle_long(self, item):
+        """Helper method for fetching a long value. Result is integer."""
+        doc = yield from self.handle_get(item)
+        if doc is None:
+            return None
+
+        return int(doc.value.u32.text) or None
+
+    @asyncio.coroutine
+    def handle_list(self, item):
+        """Helper method for fetching a list(map) value."""
+        doc = yield from self.call('LIST_GET_NEXT/'+item+'/-1', dict(
+            maxItems=100,
+        ))
+
+        if doc is None:
+            return []
+
+        if not doc.status == 'FS_OK':
+            return []
+
+        ret = list()
+        for index, item in enumerate(list(doc.iterchildren('item'))):
+            temp = dict(band=index)
+            for field in list(item.iterchildren()):
+                temp[field.get('name')] = list(field.iterchildren()).pop()
+            ret.append(temp)
+
+        return ret
+
+    @asyncio.coroutine
+    def collect_labels(self, items):
+        """Helper methods for extracting the labels from a list with maps."""
+        if items is None:
+            return []
+
+        return [str(item['label']) for item in items if item['label']]
+
+    # API implementation starts here
+
+    # sys
+    @asyncio.coroutine
+    def get_friendly_name(self):
+        """Get the friendly name of the device."""
+        return (yield from self.handle_text(self.API.get('friendly_name')))
+
+    @asyncio.coroutine
+    def set_friendly_name(self, value):
+        """Set the friendly name of the device."""
+        return (yield from self.handle_set(
+            self.API.get('friendly_name'), value))
+
+    @asyncio.coroutine
+    def get_power(self):
+        """Check if the device is on."""
+        power = (yield from self.handle_int(self.API.get('power')))
+        return bool(power)
+
+    @asyncio.coroutine
+    def set_power(self, value=False):
+        """Power on or off the device."""
+        power = (yield from self.handle_set(
+            self.API.get('power'), int(value)))
+        return bool(power)
+
+    @asyncio.coroutine
+    def get_modes(self):
+        """Get the modes supported by this device."""
+        if not self.__modes:
+            self.__modes = yield from self.handle_list(
+                self.API.get('valid_modes'))
+
+        return self.__modes
+
+    @asyncio.coroutine
+    def get_mode_list(self):
+        """Get the label list of the supported modes."""
+        self.__modes = yield from self.get_modes()
+        return (yield from self.collect_labels(self.__modes))
+
+    @asyncio.coroutine
+    def get_mode(self):
+        """Get the currently active mode on the device (DAB, FM, Spotify)."""
+        mode = None
+        int_mode = (yield from self.handle_long(self.API.get('mode')))
+        modes = yield from self.get_modes()
+        for temp_mode in modes:
+            if temp_mode['band'] == int_mode:
+                mode = temp_mode['label']
+
+        return str(mode)
+
+    @asyncio.coroutine
+    def set_mode(self, value):
+        """Set the currently active mode on the device (DAB, FM, Spotify)."""
+        mode = -1
+        modes = yield from self.get_modes()
+        for temp_mode in modes:
+            if temp_mode['label'] == value:
+                mode = temp_mode['band']
+
+        return (yield from self.handle_set(self.API.get('mode'), mode))
+
+    @asyncio.coroutine
+    def get_volume_steps(self):
+        """Read the maximum volume level of the device."""
+        if not self.__volume_steps:
+            self.__volume_steps = yield from self.handle_int(
+                self.API.get('volume_steps'))
+
+        return self.__volume_steps
+
+    # Volume
+    @asyncio.coroutine
+    def get_volume(self):
+        """Read the volume level of the device."""
+        return (yield from self.handle_int(self.API.get('volume')))
+
+    @asyncio.coroutine
+    def set_volume(self, value):
+        """Set the volume level of the device."""
+        return (yield from self.handle_set(self.API.get('volume'), value))
+
+    # Mute
+    @asyncio.coroutine
+    def get_mute(self):
+        "Check if the device is muted."
+        mute = (yield from self.handle_int(self.API.get('mute')))
+        return bool(mute)
+
+    @asyncio.coroutine
+    def set_mute(self, value=False):
+        """Mute or unmute the device."""
+        mute = (yield from self.handle_set(self.API.get('mute'), int(value)))
+        return bool(mute)
+
+    @asyncio.coroutine
+    def get_play_status(self):
+        """Get the play status of the device."""
+        status = yield from self.handle_int(self.API.get('status'))
+        return self.PLAY_STATES.get(status)
+
+    @asyncio.coroutine
+    def get_play_name(self):
+        """Get the name of the played item."""
+        return (yield from self.handle_text(self.API.get('name')))
+
+    @asyncio.coroutine
+    def get_play_text(self):
+        """Get the text associated with the played media."""
+        return (yield from self.handle_text(self.API.get('text')))
+
+    @asyncio.coroutine
+    def get_play_artist(self):
+        """Get the artists of the current media(song)."""
+        return (yield from self.handle_text(self.API.get('artist')))
+
+    @asyncio.coroutine
+    def get_play_album(self):
+        """Get the songs's album."""
+        return (yield from self.handle_text(self.API.get('album')))
+
+    @asyncio.coroutine
+    def get_play_graphic(self):
+        """Get the album art associated with the song/album/artist."""
+        return (yield from self.handle_text(self.API.get('graphic_uri')))
+
+    @asyncio.coroutine
+    def get_play_duration(self):
+        """Get the duration of the played media."""
+        return (yield from self.handle_long(self.API.get('duration')))
+
+    # play controls
+
+    @asyncio.coroutine
+    def play_control(self, value):
+        """
+        Controls the player of the device.
+        1=Play; 2=Pause; 3=Next; 4=Previous (song/station)
+        """
+        return (yield from self.handle_set(self.API.get('control'), value))
+
+    @asyncio.coroutine
+    def play(self):
+        """Play media."""
+        return (yield from self.play_control(1))
+
+    @asyncio.coroutine
+    def pause(self):
+        """Pause media."""
+        return (yield from self.play_control(2))
+
+    @asyncio.coroutine
+    def next(self):
+        """Next media."""
+        return (yield from self.play_control(3))
+
+    @asyncio.coroutine
+    def prev(self):
+        """Previous media."""
+        return (yield from self.play_control(4))
+
+    @asyncio.coroutine
+    def get_equalisers(self):
+        """Get the equaliser modes supported by this device."""
+        if not self.__equalisers:
+            self.__equalisers = yield from self.handle_list(
+                self.API.get('equalisers'))
+
+        return self.__equalisers
+
+    @asyncio.coroutine
+    def get_equaliser_list(self):
+        """Get the label list of the supported modes."""
+        self.__equalisers = yield from self.get_equalisers()
+        return (yield from self.collect_labels(self.__equalisers))
+
+    # Sleep
+    @asyncio.coroutine
+    def get_sleep(self):
+        "Check when and if the device is going to sleep."
+        return (yield from self.handle_long(self.API.get('sleep')))
+
+    @asyncio.coroutine
+    def set_sleep(self, value=False):
+        """Set device sleep timer."""
+        return (yield from self.handle_set(self.API.get('sleep'), int(value)))