diff options
| author | Philipp Hagemeister <phihag@phihag.de> | 2013-11-17 11:39:52 +0100 | 
|---|---|---|
| committer | Philipp Hagemeister <phihag@phihag.de> | 2013-11-17 11:39:52 +0100 | 
| commit | 1e5b9a95fd2049e024b3ee2f13b4da5c308d2e9c (patch) | |
| tree | 01853abf43d4507d14a68b019b328d13fc097f51 | |
| parent | 1d699755e0978f2ec0932ef8d6562394a4799871 (diff) | |
| download | hypervideo-pre-1e5b9a95fd2049e024b3ee2f13b4da5c308d2e9c.tar.lz hypervideo-pre-1e5b9a95fd2049e024b3ee2f13b4da5c308d2e9c.tar.xz hypervideo-pre-1e5b9a95fd2049e024b3ee2f13b4da5c308d2e9c.zip | |
Move console_title to YoutubeDL
| -rw-r--r-- | youtube_dl/FileDownloader.py | 17 | ||||
| -rw-r--r-- | youtube_dl/YoutubeDL.py | 13 | 
2 files changed, 16 insertions, 14 deletions
| diff --git a/youtube_dl/FileDownloader.py b/youtube_dl/FileDownloader.py index 088f59586..e5a542ed5 100644 --- a/youtube_dl/FileDownloader.py +++ b/youtube_dl/FileDownloader.py @@ -5,9 +5,6 @@ import subprocess  import sys  import time -if os.name == 'nt': -    import ctypes -  from .utils import (      compat_urllib_error,      compat_urllib_request, @@ -151,16 +148,8 @@ class FileDownloader(object):      def to_stderr(self, message):          self.ydl.to_screen(message) -    def to_cons_title(self, message): -        """Set console/terminal window title to message.""" -        if not self.params.get('consoletitle', False): -            return -        if os.name == 'nt' and ctypes.windll.kernel32.GetConsoleWindow(): -            # c_wchar_p() might not be necessary if `message` is -            # already of type unicode() -            ctypes.windll.kernel32.SetConsoleTitleW(ctypes.c_wchar_p(message)) -        elif 'TERM' in os.environ: -            self.to_screen('\033]0;%s\007' % message, skip_eol=True) +    def to_console_title(self, message): +        self.ydl.to_console_title(message)      def trouble(self, *args, **kargs):          self.ydl.trouble(*args, **kargs) @@ -249,7 +238,7 @@ class FileDownloader(object):          else:              self.to_screen(u'\r%s[download] %s of %s at %s ETA %s' %                  (clear_line, percent_str, data_len_str, speed_str, eta_str), skip_eol=True) -        self.to_cons_title(u'youtube-dl - %s of %s at %s ETA %s' % +        self.to_console_title(u'youtube-dl - %s of %s at %s ETA %s' %                  (percent_str.strip(), data_len_str.strip(), speed_str.strip(), eta_str.strip()))      def report_resuming_byte(self, resume_len): diff --git a/youtube_dl/YoutubeDL.py b/youtube_dl/YoutubeDL.py index d29e8bec5..6ea865bd9 100644 --- a/youtube_dl/YoutubeDL.py +++ b/youtube_dl/YoutubeDL.py @@ -13,6 +13,9 @@ import sys  import time  import traceback +if os.name == 'nt': +    import ctypes +  from .utils import *  from .extractor import get_info_extractor, gen_extractors  from .FileDownloader import FileDownloader @@ -176,6 +179,16 @@ class YoutubeDL(object):              output = output.encode(preferredencoding())          sys.stderr.write(output) +    def to_console_title(self, message): +        if not self.params.get('consoletitle', False): +            return +        if os.name == 'nt' and ctypes.windll.kernel32.GetConsoleWindow(): +            # c_wchar_p() might not be necessary if `message` is +            # already of type unicode() +            ctypes.windll.kernel32.SetConsoleTitleW(ctypes.c_wchar_p(message)) +        elif 'TERM' in os.environ: +            self.to_screen('\033]0;%s\007' % message, skip_eol=True) +      def fixed_template(self):          """Checks if the output template is fixed."""          return (re.search(u'(?u)%\\(.+?\\)s', self.params['outtmpl']) is None) | 
