102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
"""
|
|
Show progress for cp, mv, dd, ...
|
|
|
|
Parameters:
|
|
* progress.placeholder: Text to display while no process is running (defaults to 'n/a')
|
|
* progress.barwidth: Width of the progressbar if it is used (defaults to 8)
|
|
* progress.format: Format string (defaults to '{bar} {cmd} {arg}')
|
|
Available values are: {bar} {pid} {cmd} {arg} {percentage} {quantity} {speed} {time}
|
|
* progress.barfilledchar: Character used to draw the filled part of the bar (defaults to '#'), notice that it can be a string
|
|
* progress.baremptychar: Character used to draw the empty part of the bar (defaults to '-'), notice that it can be a string
|
|
|
|
Requires the following executable:
|
|
* progress
|
|
"""
|
|
|
|
import core.module
|
|
import core.widget
|
|
|
|
import util.cli
|
|
import util.format
|
|
|
|
import re
|
|
|
|
class Module(core.module.Module):
|
|
def __init__(self, config):
|
|
super().__init__(config, core.widget.Widget(self.get_progress_text))
|
|
self.__active = False
|
|
|
|
def get_progress_text(self, widget):
|
|
if self.update_progress_info(widget):
|
|
width = util.format.asint(self.parameter('barwidth', 8))
|
|
count = round((width * widget.get('per')) / 100)
|
|
filledchar = self.parameter('barfilledchar', '#')
|
|
emptychar = self.parameter('baremptychar', '-')
|
|
|
|
bar = '[{}{}]'.format(
|
|
filledchar * count,
|
|
emptychar * (width - count)
|
|
)
|
|
|
|
str_format = self.parameter('format', '{bar} {cmd} {arg}')
|
|
return str_format.format(
|
|
bar = bar,
|
|
pid = widget.get('pid'),
|
|
cmd = widget.get('cmd'),
|
|
arg = widget.get('arg'),
|
|
percentage = widget.get('per'),
|
|
quantity = widget.get('qty'),
|
|
speed = widget.get('spd'),
|
|
time = widget.get('tim')
|
|
)
|
|
else:
|
|
return self.parameter('placeholder', 'n/a')
|
|
|
|
def update_progress_info(self, widget):
|
|
"""Update widget's informations about the copy"""
|
|
if not self.__active: return
|
|
|
|
# These regex extracts following groups:
|
|
# 1. pid
|
|
# 2. command
|
|
# 3. arguments
|
|
# 4. progress (xx.x formated)
|
|
# 5. quantity (.. unit / .. unit formated)
|
|
# 6. speed
|
|
# 7. time remaining
|
|
extract_nospeed = re.compile('\[ *(\d*)\] ([a-zA-Z]*) (.*)\n\t(\d*\.*\d*)% \((.*)\)\n.*')
|
|
extract_wtspeed = re.compile('\[ *(\d*)\] ([a-zA-Z]*) (.*)\n\t(\d*\.*\d*)% \((.*)\) (\d*\.\d .*) remaining (\d*:\d*:\d*)\n.*')
|
|
|
|
try:
|
|
raw = util.cli.execute('progress -qW 0.1')
|
|
result = extract_wtspeed.match(raw)
|
|
|
|
if not result:
|
|
# Abord speed measures
|
|
raw = util.cli.execute('progress -q')
|
|
result = extract_nospeed.match(raw)
|
|
|
|
widget.set('spd', '???.? B/s')
|
|
widget.set('tim', '??:??:??')
|
|
else:
|
|
widget.set('spd', result.group(6))
|
|
widget.set('tim', result.group(7))
|
|
|
|
widget.set('pid', int(result.group(1)))
|
|
widget.set('cmd', result.group(2))
|
|
widget.set('arg', result.group(3))
|
|
widget.set('per', float(result.group(4)))
|
|
widget.set('qty', result.group(5))
|
|
return True
|
|
except Exception:
|
|
return False
|
|
|
|
def update(self):
|
|
self.__active = bool(util.cli.execute('progress -q'))
|
|
|
|
def state(self, widget):
|
|
if self.__active:
|
|
return 'copying'
|
|
return 'pending'
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|