kitty/kittens/tui/progress.py

44 lines
1.1 KiB
Python

#!/usr/bin/env python
# vim:fileencoding=utf-8
# License: GPLv3 Copyright: 2021, Kovid Goyal <kovid at kovidgoyal.net>
from .operations import styled, repeat
def render_progress_bar(frac: float, width: int = 80) -> str:
if frac >= 1:
return styled('🬋' * width, fg='green')
if frac <= 0:
return styled('🬋' * width, dim=True)
w = frac * width
fl = int(w)
overhang = w - fl
filled = repeat('🬋', fl)
if overhang < 0.2:
needs_break = True
elif overhang < 0.8:
filled += '🬃'
fl += 1
needs_break = False
else:
if fl < width - 1:
filled += '🬋'
fl += 1
needs_break = True
else:
filled += '🬃'
fl += 1
needs_break = False
ans = styled(filled, fg='blue')
unfilled = ''
if width > fl:
if needs_break:
unfilled += '🬇'
filler = width - fl - len(unfilled)
if filler > 0:
unfilled += repeat('🬋', filler)
if unfilled:
ans += styled(unfilled, dim=True)
return ans