aboutsummaryrefslogtreecommitdiff
path: root/test/test_download.py
diff options
context:
space:
mode:
authorFilippo Valsorda <filippo.valsorda@gmail.com>2013-06-06 15:07:05 +0200
committerFilippo Valsorda <filippo.valsorda@gmail.com>2013-06-06 15:07:05 +0200
commit476203d025dd2619ea9f9e2f99ffce507dec6596 (patch)
treeea9a98f23a3014ae97e14f9ddf3273b0d9255088 /test/test_download.py
parent468e2e926b8d1f55d6ce67fee67e33a7fa6d8371 (diff)
print WARNINGs during test + minor fix to NBAIE
Diffstat (limited to 'test/test_download.py')
-rw-r--r--test/test_download.py9
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)