]> gitweb @ CieloNegro.org - youtube-dl.git/blob - youtube_dl/downloader/rtsp.py
[generic] Update some _TESTS
[youtube-dl.git] / youtube_dl / downloader / rtsp.py
1 from __future__ import unicode_literals
2
3 import os
4 import subprocess
5
6 from .common import FileDownloader
7 from ..utils import (
8     check_executable,
9     encodeFilename,
10 )
11
12
13 class RtspFD(FileDownloader):
14     def real_download(self, filename, info_dict):
15         url = info_dict['url']
16         self.report_destination(filename)
17         tmpfilename = self.temp_name(filename)
18
19         if check_executable('mplayer', ['-h']):
20             args = [
21                 'mplayer', '-really-quiet', '-vo', 'null', '-vc', 'dummy',
22                 '-dumpstream', '-dumpfile', tmpfilename, url]
23         elif check_executable('mpv', ['-h']):
24             args = [
25                 'mpv', '-really-quiet', '--vo=null', '--stream-dump=' + tmpfilename, url]
26         else:
27             self.report_error('MMS or RTSP download detected but neither "mplayer" nor "mpv" could be run. Please install any.')
28             return False
29
30         self._debug_cmd(args)
31
32         retval = subprocess.call(args)
33         if retval == 0:
34             fsize = os.path.getsize(encodeFilename(tmpfilename))
35             self.to_screen('\r[%s] %s bytes' % (args[0], fsize))
36             self.try_rename(tmpfilename, filename)
37             self._hook_progress({
38                 'downloaded_bytes': fsize,
39                 'total_bytes': fsize,
40                 'filename': filename,
41                 'status': 'finished',
42             })
43             return True
44         else:
45             self.to_stderr('\n')
46             self.report_error('%s exited with code %d' % (args[0], retval))
47             return False