mirror of
https://github.com/kennethreitz/clint.git
synced 2026-06-05 23:00:18 +00:00
robbles
68 lines
1.2 KiB
Python
68 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
clint.textui.progress
|
|
~~~~~~~~~~~~~~~~~
|
|
|
|
This module provides the progressbar functionality.
|
|
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import sys
|
|
|
|
STREAM = sys.stderr
|
|
|
|
BAR_TEMPLATE = '%s[%s%s] %i/%i\r'
|
|
|
|
DOTS_CHAR = '.'
|
|
BAR_FILLED_CHAR = '#'
|
|
BAR_EMPTY_CHAR = ' '
|
|
|
|
def bar(it, label='', width=32, hide=False, empty_char=BAR_EMPTY_CHAR, filled_char=BAR_FILLED_CHAR):
|
|
"""Progress iterator. Wrap your iterables with it."""
|
|
|
|
def _show(_i):
|
|
x = int(width*_i/count)
|
|
if not hide:
|
|
STREAM.write(BAR_TEMPLATE % (
|
|
label, filled_char*x, empty_char*(width-x), _i, count))
|
|
STREAM.flush()
|
|
|
|
count = len(it)
|
|
|
|
if count:
|
|
_show(0)
|
|
|
|
for i, item in enumerate(it):
|
|
|
|
yield item
|
|
_show(i+1)
|
|
|
|
if not hide:
|
|
STREAM.write('\n')
|
|
STREAM.flush()
|
|
|
|
|
|
def dots(it, label='', hide=False):
|
|
"""Progress iterator. Prints a dot for each item being iterated"""
|
|
|
|
count = 0
|
|
|
|
if not hide:
|
|
STREAM.write(label)
|
|
|
|
for item in it:
|
|
if not hide:
|
|
STREAM.write(DOTS_CHAR)
|
|
sys.stderr.flush()
|
|
|
|
count += 1
|
|
|
|
yield item
|
|
|
|
STREAM.write('\n')
|
|
STREAM.flush()
|
|
|