view numerics/split_table.py @ 179:f63194f81f7d

stubbing table splitting
author Jeff Hammel <k0scist@gmail.com>
date Tue, 09 Aug 2016 14:03:33 -0700
parents
children 69543d62ae7a
line wrap: on
line source

#!/usr/bin/env python

"""
split a CSV file with a header into parts
"""

# imports
import chunk
import sys
import table

class SplitTableParser(table.TableParser):

    def add_arguments(self):
        self.add_argument('input', type=argparse.FileType('r'),
                          help="input CSV file")
        self.add_argument('-o', '--output', dest='output',
                          type=argparse.FileType('w'), default=sys.stdout,
                          help="output file to write to, or stdout by default")
        self.add_argument('-c', '--column', dest='columns', nargs='+',
                          help="column names to output")
        self.add_argument('-v', '--verbose', dest='verbose',
                          action='store_true', default=False,
                          help="be verbose")


def main(args=sys.argv[1:]):
    """CLI"""

    # parse command line
    parser = SplitTableParser(description=__doc__)
    options = parser.parse_args(args)

    # read CSV
    data = parser.read_table()

    print 'hi'

if __name__ == '__main__':
    main()