Saverr/svtplay-dl
1
0
Fork 0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-24 12:15:40 +01:00
Code Issues Projects Releases Wiki Activity

__init__: support for yield

Browse Source
This commit is contained in:
Johan Andersson 2014-04-21 20:05:36 +02:00
parent e2051436d0
commit 8acc75999a
1 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lib/svtplay_dl/__init__.py
View File

@ -13,7 +13,8 @@ from svtplay_dl.error import UIException
from svtplay_dl.log import log from svtplay_dl.log import log
from svtplay_dl.utils import get_http_data, decode_html_entities, filenamify from svtplay_dl.utils import get_http_data, decode_html_entities, filenamify
from svtplay_dl.service import service_handler, Generic from svtplay_dl.service import service_handler, Generic
from svtplay_dl.fetcher import VideoRetriever
from svtplay_dl.subtitle import subtitle, subtitle_json, subtitle_sami, subtitle_smi, subtitle_tt, subtitle_wsrt
__version__ = "0.9.2014.04.27" __version__ = "0.9.2014.04.27"
@ -102,17 +103,31 @@ def get_one_media(stream, options):
if options.output.find("\\") > 0: if options.output.find("\\") > 0:
options.output = options.output.replace("\\", "/") options.output = options.output.replace("\\", "/")
videos = []
subs = []
streams = stream.get(options)
for i in streams:
if isinstance(i, VideoRetriever):
videos.append(i)
if isinstance(i, subtitle):
subs.append(i)
if options.subtitle and options.output != "-":
if subs:
subs[0].download(options)
bitrate = sorted(x.bitrate for x in videos)
for i in videos:
if i.bitrate == bitrate[0]:
stream = i
try: try:
stream.get(options) stream.download()
except UIException as e: except UIException as e:
if options.verbose: if options.verbose:
raise e raise e
log.error(e.message) log.error(e.message)
sys.exit(2) sys.exit(2)
if options.subtitle:
if options.output != "-":
stream.get_subtitle(options)
if options.thumbnail: if options.thumbnail:
if hasattr(stream, "get_thumbnail"): if hasattr(stream, "get_thumbnail"):
log.info("thumb requested") log.info("thumb requested")