mirror of
https://github.com/borgbackup/borg.git
synced 2024-12-27 18:28:42 +00:00
Add ProgressIndicatorMessage and abstract class ProgressIndicatorBase
This commit is contained in:
parent
34f529c7df
commit
e999f3ff51
1 changed files with 38 additions and 19 deletions
|
@ -1209,23 +1209,10 @@ def ellipsis_truncate(msg, space):
|
||||||
return msg + ' ' * (space - msg_width)
|
return msg + ' ' * (space - msg_width)
|
||||||
|
|
||||||
|
|
||||||
class ProgressIndicatorPercent:
|
class ProgressIndicatorBase:
|
||||||
LOGGER = 'borg.output.progress'
|
LOGGER = 'borg.output.progress'
|
||||||
|
|
||||||
def __init__(self, total=0, step=5, start=0, msg="%3.0f%%"):
|
def __init__(self):
|
||||||
"""
|
|
||||||
Percentage-based progress indicator
|
|
||||||
|
|
||||||
:param total: total amount of items
|
|
||||||
:param step: step size in percent
|
|
||||||
:param start: at which percent value to start
|
|
||||||
:param msg: output message, must contain one %f placeholder for the percentage
|
|
||||||
"""
|
|
||||||
self.counter = 0 # 0 .. (total-1)
|
|
||||||
self.total = total
|
|
||||||
self.trigger_at = start # output next percentage value when reaching (at least) this
|
|
||||||
self.step = step
|
|
||||||
self.msg = msg
|
|
||||||
self.handler = None
|
self.handler = None
|
||||||
self.logger = logging.getLogger(self.LOGGER)
|
self.logger = logging.getLogger(self.LOGGER)
|
||||||
|
|
||||||
|
@ -1247,6 +1234,41 @@ def __del__(self):
|
||||||
self.logger.removeHandler(self.handler)
|
self.logger.removeHandler(self.handler)
|
||||||
self.handler.close()
|
self.handler.close()
|
||||||
|
|
||||||
|
|
||||||
|
def justify_to_terminal_size(message):
|
||||||
|
terminal_space = get_terminal_size(fallback=(-1, -1))[0]
|
||||||
|
# justify only if we are outputting to a terminal
|
||||||
|
if terminal_space != -1:
|
||||||
|
return message.ljust(terminal_space)
|
||||||
|
return message
|
||||||
|
|
||||||
|
|
||||||
|
class ProgressIndicatorMessage(ProgressIndicatorBase):
|
||||||
|
def output(self, msg):
|
||||||
|
self.logger.info(justify_to_terminal_size(msg))
|
||||||
|
|
||||||
|
def finish(self):
|
||||||
|
self.output('')
|
||||||
|
|
||||||
|
|
||||||
|
class ProgressIndicatorPercent(ProgressIndicatorBase):
|
||||||
|
def __init__(self, total=0, step=5, start=0, msg="%3.0f%%"):
|
||||||
|
"""
|
||||||
|
Percentage-based progress indicator
|
||||||
|
|
||||||
|
:param total: total amount of items
|
||||||
|
:param step: step size in percent
|
||||||
|
:param start: at which percent value to start
|
||||||
|
:param msg: output message, must contain one %f placeholder for the percentage
|
||||||
|
"""
|
||||||
|
self.counter = 0 # 0 .. (total-1)
|
||||||
|
self.total = total
|
||||||
|
self.trigger_at = start # output next percentage value when reaching (at least) this
|
||||||
|
self.step = step
|
||||||
|
self.msg = msg
|
||||||
|
|
||||||
|
super().__init__()
|
||||||
|
|
||||||
def progress(self, current=None, increase=1):
|
def progress(self, current=None, increase=1):
|
||||||
if current is not None:
|
if current is not None:
|
||||||
self.counter = current
|
self.counter = current
|
||||||
|
@ -1279,10 +1301,7 @@ def show(self, current=None, increase=1, info=None):
|
||||||
|
|
||||||
def output(self, message, justify=True):
|
def output(self, message, justify=True):
|
||||||
if justify:
|
if justify:
|
||||||
terminal_space = get_terminal_size(fallback=(-1, -1))[0]
|
message = justify_to_terminal_size(message)
|
||||||
# no need to ljust if we're not outputing to a terminal
|
|
||||||
if terminal_space != -1:
|
|
||||||
message = message.ljust(terminal_space)
|
|
||||||
self.logger.info(message)
|
self.logger.info(message)
|
||||||
|
|
||||||
def finish(self):
|
def finish(self):
|
||||||
|
|
Loading…
Reference in a new issue