1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-28 14:14:15 +01:00
svtplay-dl/lib/svtplay_dl/service/vimeo.py
Anders Waldenborg baa8d76551 Add get_urldata() method to service
self.get_urldata() is eqivalent to get_http_data(self.url), but
also caches the data, so no additional requests are made if it
is called multiple times (e.g when grabbing title or downloading
thumbnail).

Generic().get(url) still causes it to be fetched an extra time.
2014-02-18 19:00:20 +01:00

53 lines
1.8 KiB
Python

# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
from __future__ import absolute_import
import sys
import json
import re
from svtplay_dl.service import Service, OpenGraphThumbMixin
from svtplay_dl.utils import get_http_data
from svtplay_dl.fetcher.http import download_http
from svtplay_dl.log import log
class Vimeo(Service, OpenGraphThumbMixin):
supported_domains = ['vimeo.com']
def get(self, options):
match = re.search("data-config-url=\"(.*)\" data-fallback-url", self.get_urldata())
if not match:
log.error("Can't find data")
sys.exit(4)
player_url = match.group(1).replace("&", "&")
player_data = get_http_data(player_url)
if player_data:
jsondata = json.loads(player_data)
avail_quality = jsondata["request"]["files"]["h264"]
if options.quality:
try:
selected = avail_quality[options.quality]
except KeyError:
log.error("Can't find that quality. (Try one of: %s)",
", ".join([str(elm) for elm in avail_quality]))
sys.exit(4)
else:
try:
selected = self.select_highest_quality(avail_quality)
except KeyError:
log.error("Can't find any streams.")
sys.exit(4)
url = selected['url']
download_http(options, url)
else:
log.error("Can't find any streams.")
sys.exit(2)
def select_highest_quality(self, available):
if 'hd' in available:
return available['hd']
elif 'sd' in available:
return available['sd']
else:
raise KeyError()