551c4ae9423dc143a77b28378162c843ae7b110d
[youtube-dl] / youtube_dl / downloader / mplayer.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 MplayerFD(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         args = []
20         # Check for mplayer first
21         if check_executable('mplayer', ['-h']):
22             args = [
23                 'mplayer', '-really-quiet', '-vo', 'null', '-vc', 'dummy',
24                 '-dumpstream', '-dumpfile', tmpfilename, url]
25                         
26         # Check for mpv
27         elif check_executable('mpv', ['-h']):
28             args = [
29                 'mpv', '-really-quiet', '--vo=null', '--stream-dump=' + tmpfilename, url]
30         else:
31             self.report_error('MMS or RTSP download detected but neither "mplayer" nor "mpv" could be run')
32             return False
33
34         # Download using mplayer.
35         retval = subprocess.call(args)
36         if retval == 0:
37             fsize = os.path.getsize(encodeFilename(tmpfilename))
38             self.to_screen('\r[%s] %s bytes' % (args[0], fsize))
39             self.try_rename(tmpfilename, filename)
40             self._hook_progress({
41                 'downloaded_bytes': fsize,
42                 'total_bytes': fsize,
43                 'filename': filename,
44                 'status': 'finished',
45             })
46             return True
47         else:
48             self.to_stderr('\n')
49             self.report_error('mplayer exited with code %d' % retval)
50             return False