cmd/groups_of.py

27 lines
762 B
Python

import argparse
import sys
from voussoirkit import gentools
from voussoirkit import pipeable
def groupsof_argparse(args):
lines = pipeable.input(args.source, read_files=True, strip=True, skip_blank=True)
chunks = gentools.chunk_generator(lines, args.chunk_size)
for chunk in chunks:
chunk = args.separator.join(chunk)
pipeable.stdout(chunk)
def main(argv):
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('source')
parser.add_argument('chunk_size', type=int)
parser.add_argument('--separator', default=',')
parser.set_defaults(func=groupsof_argparse)
args = parser.parse_args(argv)
return args.func(args)
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))