cmd/groups_of.py

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)
    return 0

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:]))