utils.py

importmath
importos.path
importsys
importtextwrap


defformat_duration(seconds):
ms=math.ceil(seconds*1e3)
seconds, ms=divmod(ms, 1000)
minutes, seconds=divmod(seconds, 60)
hours, minutes=divmod(minutes, 60)

parts= []
ifhours:
parts.append('%s hour'%hours)
ifminutes:
parts.append('%s min'%minutes)
ifseconds:
ifparts:
# 2 min 1 sec
parts.append('%s sec'%seconds)
else:
# 1.0 sec
parts.append('%.1f sec'% (seconds+ms/1000))
ifnotparts:
return'%s ms'%ms

parts=parts[:2]
return' '.join(parts)


defremovepy(names):
ifnotnames:
return
foridx, nameinenumerate(names):
basename, ext=os.path.splitext(name)
ifext=='.py':
names[idx] =basename


defcount(n, word):
ifn==1:
return"%d %s"% (n, word)
else:
return"%d %ss"% (n, word)


defprintlist(x, width=70, indent=4, file=None):
"""Print the elements of iterable x to stdout.

 Optional arg width (default 70) is the maximum line length.
 Optional arg indent (default 4) is the number of blanks with which to
 begin each line.
 """

blanks=' '*indent
# Print the sorted list: 'x' may be a '--random' list or a set()
print(textwrap.fill(' '.join(str(elt) foreltinsorted(x)), width,
initial_indent=blanks, subsequent_indent=blanks),
file=file)


defprint_warning(msg):
print(f"Warning -- {msg}", file=sys.stderr, flush=True)