more error handling

[svtplaydump.git] / tv4playdump.py

#!/usr/bin/env python3
import requests
import svtplaydump
import os.path
def get_hls_playlist(url):
    ret = requests.get(url).json()
    #import pdb;pdb.set_trace()
    videourl = None
    if isinstance(ret['playback']['items']['item'], list):
        videourl = sorted(ret['playback']['items']['item'], key=lambda k: int(k['bitrate']))[-1]['url']
    else:
        videourl = ret['playback']['items']['item']['url']
    return videourl

res = requests.get("http://webapi.tv4play.se/video/programs/search.json?categoryids=pokemon&start=0&rows=1000").json()
videos = []
#import pdb;pdb.set_trace()
for vid in res['results']:
    video = {}
    video['title'] = vid['name']
    video['description'] = vid['lead']
    video['url'] = get_hls_playlist("http://premium.tv4play.se/api/web/asset/{}/play.json?protocol=hls&videoFormat=MP4+WVM+SMI".format(vid['href']))
    video['filename'] = "{} {}.ts".format(vid['ontime'],vid['name'])
    videos.append(video)
    basename = video['filename'].split('.ts')[0]
    if os.path.exists(basename+'.mkv'):
        print("Skipping {}".format(basename))
        continue
    svtplaydump.download_from_playlist(video)
    svtplaydump.remux(video)