浏览代码

Use red color when printing error messages

Jaime Marquínez Ferrándiz 12 年之前
父节点
当前提交
4e1582f372
共有 1 个文件被更改,包括 12 次插入0 次删除
  1. 12 0
      youtube_dl/FileDownloader.py

+ 12 - 0
youtube_dl/FileDownloader.py

@@ -246,6 +246,18 @@ class FileDownloader(object):
         warning_message=u'%s %s' % (_msg_header,message)
         self.to_stderr(warning_message)
 
+    def report_error(self, message, tb=None):
+        '''
+        Do the same as trouble, but prefixes the message with 'ERROR:', colored
+        in red if stderr is a tty file.
+        '''
+        if sys.stderr.isatty():
+            _msg_header = u'\033[0;31mERROR:\033[0m'
+        else:
+            _msg_header = u'ERROR:'
+        error_message = u'%s %s' % (_msg_header, message)
+        self.trouble(error_message, tb)
+
     def slow_down(self, start_time, byte_counter):
         """Sleep if the download speed is over the rate limit."""
         rate_limit = self.params.get('ratelimit', None)