From 1f19b5b61c9a8c229d736853fe84ffe088b080ca Mon Sep 17 00:00:00 2001
From: Georgios Verigakis <verigak@gmail.com>
Date: Tue, 10 Jul 2018 13:23:43 +0300
Subject: [PATCH] Add flag to override tty check

---
 progress/helpers.py | 22 +++++++++++++++-------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/progress/helpers.py b/progress/helpers.py
index 0cde44e..5cf66f1 100644
--- a/progress/helpers.py
+++ b/progress/helpers.py
@@ -20,6 +20,7 @@ SHOW_CURSOR = '\x1b[?25h'
 
 
 class WriteMixin(object):
+    check_tty = True
     hide_cursor = False
 
     def __init__(self, message=None, **kwargs):
@@ -28,14 +29,14 @@ class WriteMixin(object):
         if message:
             self.message = message
 
-        if self.file and self.file.isatty():
+        if self.file and self.is_tty():
             if self.hide_cursor:
                 print(HIDE_CURSOR, end='', file=self.file)
             print(self.message, end='', file=self.file)
             self.file.flush()
 
     def write(self, s):
-        if self.file and self.file.isatty():
+        if self.file and self.is_tty():
             b = '\b' * self._width
             c = s.ljust(self._width)
             print(b + c, end='', file=self.file)
@@ -43,11 +44,15 @@ class WriteMixin(object):
             self.file.flush()
 
     def finish(self):
-        if self.file and self.file.isatty() and self.hide_cursor:
+        if self.file and self.is_tty() and self.hide_cursor:
             print(SHOW_CURSOR, end='', file=self.file)
 
+    def is_tty(self):
+        return self.file.isatty() if self.check_tty else True
+
 
 class WritelnMixin(object):
+    check_tty = True
     hide_cursor = False
 
     def __init__(self, message=None, **kwargs):
@@ -55,25 +60,28 @@ class WritelnMixin(object):
         if message:
             self.message = message
 
-        if self.file and self.file.isatty() and self.hide_cursor:
+        if self.file and self.is_tty() and self.hide_cursor:
             print(HIDE_CURSOR, end='', file=self.file)
 
     def clearln(self):
-        if self.file and self.file.isatty():
+        if self.file and self.is_tty():
             print('\r\x1b[K', end='', file=self.file)
 
     def writeln(self, line):
-        if self.file and self.file.isatty():
+        if self.file and self.is_tty():
             self.clearln()
             print(line, end='', file=self.file)
             self.file.flush()
 
     def finish(self):
-        if self.file and self.file.isatty():
+        if self.file and self.is_tty():
             print(file=self.file)
             if self.hide_cursor:
                 print(SHOW_CURSOR, end='', file=self.file)
 
+    def is_tty(self):
+        return self.file.isatty() if self.check_tty else True
+
 
 from signal import signal, SIGINT
 from sys import exit