Saverr/svtplay-dl
1
0
Fork 0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-27 21:54:17 +01:00
Code Issues Projects Releases Wiki Activity

get_media: split all_episodes to its own function

Browse Source
This commit is contained in:
Johan Andersson 2016-03-26 13:08:45 +01:00
parent 504b3ccf22
commit f7488af095
1 changed files with 27 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/svtplay_dl/__init__.py
View File

@ -152,6 +152,12 @@ def get_media(url, options):
sys.exit(2) sys.exit(2)
if options.all_episodes: if options.all_episodes:
get_all_episodes(stream, options, url)
else:
get_one_media(stream, options)
def get_all_episodes(stream, options, url):
if options.output and os.path.isfile(options.output): if options.output and os.path.isfile(options.output):
log.error("Output must be a directory if used with --all-episodes") log.error("Output must be a directory if used with --all-episodes")
sys.exit(2) sys.exit(2)
@ -175,8 +181,6 @@ def get_media(url, options):
# get_one_media overwrites options.output... # get_one_media overwrites options.output...
get_one_media(substream, copy.copy(options)) get_one_media(substream, copy.copy(options))
else:
get_one_media(stream, options)
def get_one_media(stream, options): def get_one_media(stream, options):