2
0
Эх сурвалжийг харах

Use red color when printing error messages

Jaime Marquínez Ferrándiz 12 жил өмнө
parent
commit
4e1582f372

+ 12 - 0
youtube_dl/FileDownloader.py

@@ -246,6 +246,18 @@ class FileDownloader(object):
         warning_message=u'%s %s' % (_msg_header,message)
         warning_message=u'%s %s' % (_msg_header,message)
         self.to_stderr(warning_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):
     def slow_down(self, start_time, byte_counter):
         """Sleep if the download speed is over the rate limit."""
         """Sleep if the download speed is over the rate limit."""
         rate_limit = self.params.get('ratelimit', None)
         rate_limit = self.params.get('ratelimit', None)