16387a3f50
This greatly improves loading times and takes some load off the APIs. * Add generic directory class to also hold item count * Radiobrowser: Get station count directly from API * Optionally show broken stations and their count * Remove minimum station bitrate to not filter away some listings * Improve code wording * Log API requests
116 lines
4.3 KiB
Python
116 lines
4.3 KiB
Python
import requests
|
|
import logging
|
|
|
|
import ycast.vtuner as vtuner
|
|
import ycast.generic as generic
|
|
|
|
MINIMUM_COUNT_GENRE = 5
|
|
MINIMUM_COUNT_COUNTRY = 5
|
|
DEFAULT_STATION_LIMIT = 200
|
|
SHOW_BROKEN_STATIONS = False
|
|
ID_PREFIX = "RB_"
|
|
|
|
|
|
def get_json_attr(json, attr):
|
|
try:
|
|
return json[attr]
|
|
except:
|
|
return None
|
|
|
|
|
|
class Station:
|
|
def __init__(self, station_json):
|
|
self.id = ID_PREFIX + get_json_attr(station_json, 'id')
|
|
self.name = get_json_attr(station_json, 'name')
|
|
self.url = get_json_attr(station_json, 'url')
|
|
self.icon = get_json_attr(station_json, 'favicon')
|
|
self.tags = get_json_attr(station_json, 'tags').split(',')
|
|
self.country = get_json_attr(station_json, 'country')
|
|
self.language = get_json_attr(station_json, 'language')
|
|
self.votes = get_json_attr(station_json, 'votes')
|
|
self.codec = get_json_attr(station_json, 'codec')
|
|
self.bitrate = get_json_attr(station_json, 'bitrate')
|
|
|
|
def to_vtuner(self):
|
|
return vtuner.Station(self.id, self.name, ', '.join(self.tags), self.url, self.icon,
|
|
self.tags[0], self.country, self.codec, self.bitrate, None)
|
|
|
|
|
|
def request(url):
|
|
logging.debug("Radiobrowser API request: %s", url)
|
|
headers = {'content-type': 'application/json', 'User-Agent': 'YCast'}
|
|
response = requests.get('http://www.radio-browser.info/webservice/json/' + url, headers=headers)
|
|
if response.status_code != 200:
|
|
logging.error("Could not fetch data from Radiobrowser (%s)", response.status_code)
|
|
return {}
|
|
return response.json()
|
|
|
|
|
|
def get_station_by_id(uid):
|
|
station_json = request('stations/byid/' + str(uid))
|
|
return Station(station_json[0])
|
|
|
|
|
|
def search(name, limit=DEFAULT_STATION_LIMIT):
|
|
stations = []
|
|
stations_json = request('stations/search?order=name&reverse=false&limit=' + str(limit) + '&name=' + str(name))
|
|
for station_json in stations_json:
|
|
if SHOW_BROKEN_STATIONS or get_json_attr(station_json, 'lastcheckok') == '1':
|
|
stations.append(Station(station_json))
|
|
return stations
|
|
|
|
|
|
def get_country_directories():
|
|
country_directories = []
|
|
apicall = 'countries'
|
|
if not SHOW_BROKEN_STATIONS:
|
|
apicall += '?hidebroken=true'
|
|
countries_raw = request(apicall)
|
|
for country_raw in countries_raw:
|
|
if get_json_attr(country_raw, 'name') and get_json_attr(country_raw, 'stationcount') and \
|
|
int(get_json_attr(country_raw, 'stationcount')) > MINIMUM_COUNT_COUNTRY:
|
|
country_directories.append(generic.Directory(get_json_attr(country_raw, 'name'),
|
|
get_json_attr(country_raw, 'stationcount')))
|
|
return country_directories
|
|
|
|
|
|
def get_genre_directories():
|
|
genre_directories = []
|
|
apicall = 'tags'
|
|
if not SHOW_BROKEN_STATIONS:
|
|
apicall += '?hidebroken=true'
|
|
genres_raw = request(apicall)
|
|
for genre_raw in genres_raw:
|
|
if get_json_attr(genre_raw, 'name') and get_json_attr(genre_raw, 'stationcount') and \
|
|
int(get_json_attr(genre_raw, 'stationcount')) > MINIMUM_COUNT_GENRE:
|
|
genre_directories.append(generic.Directory(get_json_attr(genre_raw, 'name'),
|
|
get_json_attr(genre_raw, 'stationcount')))
|
|
return genre_directories
|
|
|
|
|
|
def get_stations_by_country(country):
|
|
stations = []
|
|
stations_json = request('stations/search?order=name&reverse=false&countryExact=true&country=' + str(country))
|
|
for station_json in stations_json:
|
|
if SHOW_BROKEN_STATIONS or get_json_attr(station_json, 'lastcheckok') == '1':
|
|
stations.append(Station(station_json))
|
|
return stations
|
|
|
|
|
|
def get_stations_by_genre(genre):
|
|
stations = []
|
|
stations_json = request('stations/search?order=name&reverse=false&tagExact=true&tag=' + str(genre))
|
|
for station_json in stations_json:
|
|
if SHOW_BROKEN_STATIONS or get_json_attr(station_json, 'lastcheckok') == '1':
|
|
stations.append(Station(station_json))
|
|
return stations
|
|
|
|
|
|
def get_stations_by_votes(limit=DEFAULT_STATION_LIMIT):
|
|
stations = []
|
|
stations_json = request('stations?order=votes&reverse=true&limit=' + str(limit))
|
|
for station_json in stations_json:
|
|
if SHOW_BROKEN_STATIONS or get_json_attr(station_json, 'lastcheckok') == '1':
|
|
stations.append(Station(station_json))
|
|
return stations
|