mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-09-05 22:15:13 +02:00
Make sure it only runs rtmpdump one time in test mode and return True if the download can be resumed
This commit is contained in:
parent
81d7f1928c
commit
9026dd3858
@ -270,6 +270,7 @@ def report_finish(self, data_len_str, tot_time):
|
||||
def _download_with_rtmpdump(self, filename, url, player_url, page_url, play_path, tc_url):
|
||||
self.report_destination(filename)
|
||||
tmpfilename = self.temp_name(filename)
|
||||
test = self.params.get('test', False)
|
||||
|
||||
# Check for rtmpdump first
|
||||
try:
|
||||
@ -291,7 +292,7 @@ def _download_with_rtmpdump(self, filename, url, player_url, page_url, play_path
|
||||
basic_args += ['--playpath', play_path]
|
||||
if tc_url is not None:
|
||||
basic_args += ['--tcUrl', url]
|
||||
if self.params.get('test', False):
|
||||
if test:
|
||||
basic_args += ['--stop', '1']
|
||||
args = basic_args + [[], ['--resume', '--skip', '1']][self.params.get('continuedl', False)]
|
||||
if self.params.get('verbose', False):
|
||||
@ -302,7 +303,7 @@ def _download_with_rtmpdump(self, filename, url, player_url, page_url, play_path
|
||||
shell_quote = repr
|
||||
self.to_screen(u'[debug] rtmpdump command line: ' + shell_quote(args))
|
||||
retval = subprocess.call(args)
|
||||
while retval == 2 or retval == 1:
|
||||
while (retval == 2 or retval == 1) and not test:
|
||||
prevsize = os.path.getsize(encodeFilename(tmpfilename))
|
||||
self.to_screen(u'\r[rtmpdump] %s bytes' % prevsize, skip_eol=True)
|
||||
time.sleep(5.0) # This seems to be needed
|
||||
@ -315,7 +316,7 @@ def _download_with_rtmpdump(self, filename, url, player_url, page_url, play_path
|
||||
self.to_screen(u'\r[rtmpdump] Could not download the whole video. This can happen for some advertisements.')
|
||||
retval = 0
|
||||
break
|
||||
if retval == 0:
|
||||
if retval == 0 or (test and retval == 2):
|
||||
fsize = os.path.getsize(encodeFilename(tmpfilename))
|
||||
self.to_screen(u'\r[rtmpdump] %s bytes' % fsize)
|
||||
self.try_rename(tmpfilename, filename)
|
||||
|
Loading…
Reference in New Issue
Block a user