48 lines
1.3 KiB
Python

# coding: utf-8
from __future__ import unicode_literals
from .common import InfoExtractor
class PonyFMIE(InfoExtractor):
"""
InfoExtractor for PonyFM
This extractor is for tracks. Playlists and Albums may be defined at a later
point using a separate class that will be in this file.
"""
_VALID_URL = r'https?://pony\.fm/tracks/(?P<id>\d+)-.+'
_TESTS = []
def _real_extract(self, url):
track_id = self._match_id(url)
# Extract required fields from JSON API
apiurl = "https://pony.fm/api/web/tracks/%s" % track_id
json = self._download_json(apiurl, track_id)['track']
title = json['title']
formats = []
for f in json['formats']:
formats.append({
'format': f['name'],
'url': f['url'],
})
extracted = {
'id': track_id,
'title': title,
'formats': formats,
}
# Extract optional metadata (author, album art) from JSON API
author = json.get('user', {}).get('name')
artwork = json.get('covers', {}).get('original')
if artwork:
extracted['thumbnail'] = artwork
if author:
extracted['uploader'] = author
return extracted