From bb94f84cde1d1250bd14c44ddf4e5887e3c5f913 Mon Sep 17 00:00:00 2001
From: ant31 <2t.antoine@gmail.com>
Date: Thu, 24 Nov 2016 18:24:36 +0100
Subject: [PATCH] Add ascii/no-color progressbar

---
 pytest_sugar.py | 88 +++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 81 insertions(+), 7 deletions(-)

diff --git a/pytest_sugar.py b/pytest_sugar.py
index 1764fe9..f65a7b7 100644
--- a/pytest_sugar.py
+++ b/pytest_sugar.py
@@ -28,13 +28,14 @@
 from _pytest.terminal import TerminalReporter
 
 
-__version__ = '0.7.1'
+__version__ = '0.7.2'
 
 LEN_RIGHT_MARGIN = 0
 LEN_PROGRESS_PERCENTAGE = 5
 LEN_PROGRESS_BAR_SETTING = '10'
 LEN_PROGRESS_BAR = None
-THEME = {
+
+DEFAULT_THEME = {
     'header': 'magenta',
     'skipped': 'blue',
     'success': 'green',
@@ -45,6 +46,7 @@
     'progressbar': 'green',
     'progressbar_fail': 'red',
     'progressbar_background': 'grey',
+    'progressbar_delimiters': '',
     'path': 'cyan',
     'name': None,
     'symbol_passed': '✓',
@@ -57,10 +59,52 @@
     'unknown': 'blue',
     'symbol_rerun': 'R',
     'rerun': 'blue',
+    'progress_bar': 'block'
 }
-PROGRESS_BAR_BLOCKS = [
-    ' ', '▏', '▎', '▎', '▍', '▍', '▌', '▌', '▋', '▋', '▊', '▊', '▉', '▉', '█',
-]
+
+NO_COLOR_THEME = {
+    'header': None,
+    'skipped': None,
+    'success': None,
+    'warning': None,
+    'fail': None,
+    'xfailed': None,
+    'xpassed': None,
+    'progressbar': None,
+    'progressbar_fail': None,
+    'progressbar_background': None,
+    'progressbar_delimiters': "[]",
+    'path': None,
+    'name': None,
+    'symbol_passed': ".",
+    'symbol_skipped': "s",
+    'symbol_failed': 'x',
+    'symbol_failed_not_call': 'x',
+    'symbol_xfailed_skipped': "Xs",
+    'symbol_xfailed_failed': "Xf",
+    'symbol_unknown': '?',
+    'unknown': None,
+    'symbol_rerun': 'R',
+    'rerun': None,
+    'progress_bar': 'equal'
+}
+
+THEMES = {
+    'no-color': NO_COLOR_THEME,
+    'default': DEFAULT_THEME
+    }
+
+THEME = THEMES['default']
+
+PROGRESS_BAR_BLOCKS_THEMES = {
+    'block': [' ', '▏', '▎', '▎', '▍', '▍', '▌', '▌', '▋', '▋', '▊', '▊', '▉', '▉', '█'],
+    'progress': ['-', '\\', '|', '/', '-', '\\', '|', '/', '-', '\\', '|', '/', '-', '\\', '='],
+    'equal': ['='] * 15
+}
+
+PROGRESS_BAR_BLOCKS = PROGRESS_BAR_BLOCKS_THEMES[THEME['progress_bar']]
+
+
 
 
 def flatten(l):
@@ -109,10 +153,24 @@ def pytest_addoption(parser):
             "Show tests that failed instead of one-line tracebacks"
         )
     )
+    group._addoption(
+        '--show-count', action="store_true",
+        dest="show_count", default=False,
+        help=(
+            "Show tests taken / tests count"
+        )
+    )
+    group._addoption(
+        '--progressbar-len', action="store",
+        dest="progressbar_len", default="10",
+        help=(
+            "customize progress-bar size"
+        )
+    )
 
 
 def pytest_sessionstart(session):
-    global LEN_PROGRESS_BAR_SETTING
+    global LEN_PROGRESS_BAR_SETTING, PROGRESS_BAR_BLOCKS
     config = ConfigParser()
     config.read(['pytest-sugar.conf', os.path.expanduser('~/.pytest-sugar.conf')])
 
@@ -130,6 +188,8 @@ def pytest_sessionstart(session):
     if config.has_option('sugar', 'progressbar_length'):
         LEN_PROGRESS_BAR_SETTING = config.get('sugar', 'progressbar_length')
 
+    PROGRESS_BAR_BLOCKS = PROGRESS_BAR_BLOCKS_THEMES[THEME['progress_bar']]
+
 
 def strip_colors(text):
     ansi_escape = re.compile(r'\x1b[^m]*m')
@@ -181,6 +241,7 @@ def pytest_report_teststatus(report):
 
 class SugarTerminalReporter(TerminalReporter):
     def __init__(self, reporter):
+        global THEME, LEN_PROGRESS_BAR_SETTING
         TerminalReporter.__init__(self, reporter.config)
         self.writer = self._tw
         self.paths_left = []
@@ -192,6 +253,10 @@ def __init__(self, reporter):
         self.unreported_errors = []
         self.progress_blocks = []
 
+        if self.config.option.color == "no":
+            THEME = THEMES['no-color']
+        LEN_PROGRESS_BAR_SETTING = self.config.option.progressbar_len
+
     def report_collect(self, final=False):
         pass
 
@@ -245,8 +310,10 @@ def get_progress_bar():
             bar = PROGRESS_BAR_BLOCKS[-1] * floored
             if rem > 0:
                 bar += PROGRESS_BAR_BLOCKS[rem]
-            bar += ' ' * (LEN_PROGRESS_BAR - len(bar))
 
+            bar += ' ' * (LEN_PROGRESS_BAR - len(bar))
+            if THEME['progressbar_delimiters'] and len(THEME['progressbar_delimiters']) == 2:
+                bar = THEME['progressbar_delimiters'][0] + bar + THEME['progressbar_delimiters'][1]
             last = 0
             last_theme = None
 
@@ -281,6 +348,9 @@ def get_progress_bar():
             return progressbar
 
         append_string = get_progress_bar()
+        if self.config.option.show_count:
+            count = str(self.tests_count)
+            append_string += "[%s/%s]" % (str(self.tests_taken).zfill(len(count)), count)
         return self.append_string(append_string)
 
     def append_string(self, append_string=''):
@@ -295,11 +365,15 @@ def overwrite(self, line):
         self.writer.write("\r" + line)
 
     def get_max_column_for_test_status(self):
+        show_count_len = 0
+        if self.config.option.show_count:
+            show_count_len = len(str(self.tests_count)) * 2 + 5
         return (
             self._tw.fullwidth
             - LEN_PROGRESS_PERCENTAGE
             - LEN_PROGRESS_BAR
             - LEN_RIGHT_MARGIN
+            - show_count_len
         )
 
     def begin_new_line(self, report, print_filename):