Ethan Dalool
4a9051e617
With pathclass.glob_many, we can clean up and feel more confident about many programs that use pipeable to take glob patterns. Added return 0 to all programs that didn't have it, so we have consistent and explicit command line return values. Other linting and whitespace.
21 lines
471 B
Python
21 lines
471 B
Python
import sys
|
|
|
|
from voussoirkit import pathclass
|
|
from voussoirkit import pipeable
|
|
from voussoirkit import spinal
|
|
|
|
@pipeable.ctrlc_return1
|
|
def main(argv):
|
|
total = 0
|
|
for path in pipeable.go():
|
|
path = pathclass.Path(path)
|
|
if path.is_file:
|
|
total += path.size
|
|
elif path.is_dir:
|
|
total += spinal.get_dir_size(path)
|
|
|
|
pipeable.stdout(total)
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main(sys.argv[1:]))
|