f306366629
Use a small helper function from util, tidy up some parts of the output.
33 lines
927 B
Python
33 lines
927 B
Python
import shlex
|
|
import subprocess
|
|
try:
|
|
from exceptions import RuntimeError
|
|
except ImportError:
|
|
# Python3 doesn't require this anymore
|
|
pass
|
|
|
|
def bytefmt(num):
|
|
for unit in [ "", "Ki", "Mi", "Gi" ]:
|
|
if num < 1024.0:
|
|
return "{:.2f}{}B".format(num, unit)
|
|
num /= 1024.0
|
|
return "{:05.2f%}{}GiB".format(num)
|
|
|
|
def durationfmt(duration):
|
|
minutes, seconds = divmod(duration, 60)
|
|
hours, minutes = divmod(minutes, 60)
|
|
res = "{:02d}:{:02d}".format(minutes, seconds)
|
|
if hours > 0: res = "{:02d}:{}".format(hours, res)
|
|
|
|
return res
|
|
|
|
def execute(cmd, wait=True):
|
|
args = shlex.split(cmd)
|
|
p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
|
|
if wait:
|
|
out, err = p.communicate()
|
|
if p.returncode != 0:
|
|
raise RuntimeError("{} exited with {}".format(cmd, p.returncode))
|
|
return out.decode("utf-8")
|
|
return None
|