11d6312daa
Mopidy-Radionet will crash (and make the Mopidy search system crash with it) if the query has no key "any". This happens when searching specific fields. Since searching for a radio name only makes sense with "any", my patch will make mopidy-radionet skip the search entirely if "any" isn't present.
121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
import re
|
|
|
|
from mopidy import backend
|
|
from mopidy.models import Album, Artist, Ref, SearchResult, Track
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RadioNetLibraryProvider(backend.LibraryProvider):
|
|
root_directory = Ref.directory(uri='radionet:root', name='Radio.net')
|
|
|
|
def __init__(self, backend):
|
|
super().__init__(backend)
|
|
|
|
def lookup(self, uri):
|
|
|
|
if not uri.startswith('radionet:'):
|
|
return None
|
|
|
|
variant, identifier = self.parse_uri(uri)
|
|
|
|
if variant == 'station':
|
|
identifier = int(identifier)
|
|
radio_data = self.backend.radionet.get_station_by_id(identifier)
|
|
|
|
artist = Artist(name=radio_data.name)
|
|
|
|
album = Album(
|
|
artists=[artist],
|
|
name=radio_data.description + ' / ' + radio_data.continent +
|
|
' / ' + radio_data.country + ' - ' + radio_data.city,
|
|
uri='radionet:station:%s' % (identifier))
|
|
|
|
track = Track(
|
|
artists=[artist],
|
|
album=album,
|
|
name=radio_data.name,
|
|
genre=radio_data.genres,
|
|
comment=radio_data.description,
|
|
uri=radio_data.stream_url)
|
|
return [track]
|
|
|
|
return []
|
|
|
|
def browse(self, uri):
|
|
self.backend.refresh()
|
|
|
|
directories = []
|
|
tracks = []
|
|
variant, identifier = self.parse_uri(uri)
|
|
if variant == 'root':
|
|
if self.backend.radionet.local_stations:
|
|
directories.append(
|
|
self.ref_directory(
|
|
"radionet:category:localstations", "Local stations")
|
|
)
|
|
if self.backend.radionet.top_stations:
|
|
directories.append(
|
|
self.ref_directory(
|
|
"radionet:category:top100", "Top 100")
|
|
)
|
|
if self.backend.radionet.favorite_stations:
|
|
directories.append(
|
|
self.ref_directory(
|
|
"radionet:category:favorites", "Favorites")
|
|
)
|
|
return directories
|
|
elif variant == 'category' and identifier:
|
|
if identifier == "localstations":
|
|
for station in self.backend.radionet.local_stations:
|
|
tracks.append(self.station_to_ref(station))
|
|
if identifier == "top100":
|
|
for station in self.backend.radionet.top_stations:
|
|
tracks.append(self.station_to_ref(station))
|
|
if identifier == "favorites":
|
|
for station in self.backend.radionet.favorite_stations:
|
|
tracks.append(self.station_to_ref(station))
|
|
tracks.sort(key=lambda ref: ref.name)
|
|
return tracks
|
|
else:
|
|
logger.debug('Unknown URI: %s', uri)
|
|
return []
|
|
|
|
def search(self, query=None, uris=None, exact=False):
|
|
if 'any' not in query:
|
|
return None
|
|
|
|
result = []
|
|
|
|
self.backend.radionet.do_search(' '.join(query['any']))
|
|
|
|
for station in self.backend.radionet.search_results:
|
|
result.append(self.station_to_track(station))
|
|
|
|
return SearchResult(
|
|
tracks=result
|
|
)
|
|
|
|
def station_to_ref(self, station):
|
|
return Ref.track(
|
|
uri='radionet:station:%s' % (station.id),
|
|
name=station.name,
|
|
)
|
|
|
|
def station_to_track(self, station):
|
|
ref = self.station_to_ref(station)
|
|
return Track(uri=ref.uri, name=ref.name, album=Album(uri=ref.uri, name=ref.name),
|
|
artists=[Artist(uri=ref.uri, name=ref.name)])
|
|
|
|
def ref_directory(self, uri, name):
|
|
return Ref.directory(uri=uri, name=name)
|
|
|
|
def parse_uri(self, uri):
|
|
result = re.findall(r'^radionet:([a-z]+):?([a-z0-9]+|\d+)?$', uri)
|
|
if result:
|
|
return result[0]
|
|
return None, None
|