diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /js/src/tests/lib/progressbar.py | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip |
Add m-esr52 at 52.6.0
Diffstat (limited to 'js/src/tests/lib/progressbar.py')
-rw-r--r-- | js/src/tests/lib/progressbar.py | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/js/src/tests/lib/progressbar.py b/js/src/tests/lib/progressbar.py new file mode 100644 index 000000000..29361660d --- /dev/null +++ b/js/src/tests/lib/progressbar.py @@ -0,0 +1,116 @@ +# Text progress bar library, like curl or scp. + +from datetime import datetime, timedelta +import math +import sys + +if sys.platform.startswith('win'): + from terminal_win import Terminal +else: + from terminal_unix import Terminal + +class NullProgressBar(object): + def update(self, current, data): pass + def poke(self): pass + def finish(self, complete=True): pass + def beginline(self): pass + def message(self, msg): sys.stdout.write(msg + '\n') + @staticmethod + def update_granularity(): return timedelta.max + +class ProgressBar(object): + def __init__(self, limit, fmt): + assert self.conservative_isatty() + + self.prior = None + self.atLineStart = True + self.counters_fmt = fmt # [{str:str}] Describtion of how to lay out each + # field in the counters map. + self.limit = limit # int: The value of 'current' equal to 100%. + self.limit_digits = int(math.ceil(math.log10(self.limit))) # int: max digits in limit + self.t0 = datetime.now() # datetime: The start time. + + # Compute the width of the counters and build the format string. + self.counters_width = 1 # [ + for layout in self.counters_fmt: + self.counters_width += self.limit_digits + self.counters_width += 1 # | (or ']' for the last one) + + self.barlen = 64 - self.counters_width + + @staticmethod + def update_granularity(): + return timedelta(seconds=0.1) + + def update(self, current, data): + # Record prior for poke. + self.prior = (current, data) + self.atLineStart = False + + # Build counters string. + sys.stdout.write('\r[') + for layout in self.counters_fmt: + Terminal.set_color(layout['color']) + sys.stdout.write(('{:' + str(self.limit_digits) + 'd}').format( + data[layout['value']])) + Terminal.reset_color() + if layout != self.counters_fmt[-1]: + sys.stdout.write('|') + else: + sys.stdout.write('] ') + + # Build the bar. + pct = int(100.0 * current / self.limit) + sys.stdout.write('{:3d}% '.format(pct)) + + barlen = int(1.0 * self.barlen * current / self.limit) - 1 + bar = '=' * barlen + '>' + ' ' * (self.barlen - barlen - 1) + sys.stdout.write(bar + '|') + + # Update the bar. + dt = datetime.now() - self.t0 + dt = dt.seconds + dt.microseconds * 1e-6 + sys.stdout.write('{:6.1f}s'.format(dt)) + Terminal.clear_right() + + # Force redisplay, since we didn't write a \n. + sys.stdout.flush() + + def poke(self): + if not self.prior: + return + self.update(*self.prior) + + def finish(self, complete=True): + if not self.prior: + sys.stdout.write('No test run... You can try adding' + ' --run-slow-tests or --run-skipped to run more tests\n') + return + final_count = self.limit if complete else self.prior[0] + self.update(final_count, self.prior[1]) + sys.stdout.write('\n') + + def beginline(self): + if not self.atLineStart: + sys.stdout.write('\n') + self.atLineStart = True + + def message(self, msg): + self.beginline() + sys.stdout.write(msg) + sys.stdout.write('\n') + + @staticmethod + def conservative_isatty(): + """ + Prefer erring on the side of caution and not using terminal commands if + the current output stream may be a file. We explicitly check for the + Android platform because terminal commands work poorly over ADB's + redirection. + """ + try: + import android + return False + except ImportError: + return sys.stdout.isatty() + return False |