diff options
| author | Filippo Valsorda <filippo.valsorda@gmail.com> | 2013-06-06 15:07:05 +0200 |
|---|---|---|
| committer | Filippo Valsorda <filippo.valsorda@gmail.com> | 2013-06-06 15:07:05 +0200 |
| commit | 476203d025dd2619ea9f9e2f99ffce507dec6596 (patch) | |
| tree | ea9a98f23a3014ae97e14f9ddf3273b0d9255088 /test/test_download.py | |
| parent | 468e2e926b8d1f55d6ce67fee67e33a7fa6d8371 (diff) | |
| download | youtube-dl-476203d025dd2619ea9f9e2f99ffce507dec6596.tar.gz youtube-dl-476203d025dd2619ea9f9e2f99ffce507dec6596.tar.xz youtube-dl-476203d025dd2619ea9f9e2f99ffce507dec6596.zip | |
print WARNINGs during test + minor fix to NBAIE
Diffstat (limited to 'test/test_download.py')
| -rw-r--r-- | test/test_download.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/test/test_download.py b/test/test_download.py index 3eca333f2..3e6bdd44e 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -40,9 +40,18 @@ def _try_rm(filename): class FileDownloader(youtube_dl.FileDownloader): def __init__(self, *args, **kwargs): + self._to_stderr = self.to_stderr self.to_stderr = self.to_screen self.processed_info_dicts = [] return youtube_dl.FileDownloader.__init__(self, *args, **kwargs) + def report_warning(self, message): + # let warnings pass to output + if sys.stderr.isatty() and os.name != 'nt': + _msg_header=u'\033[0;33mWARNING:\033[0m' + else: + _msg_header=u'WARNING:' + warning_message=u'%s %s' % (_msg_header,message) + self._to_stderr(warning_message) def process_info(self, info_dict): self.processed_info_dicts.append(info_dict) return youtube_dl.FileDownloader.process_info(self, info_dict) |