#!/usr/bin/env python

# Selects columns from a spreadsheet

import csv

import os.path

import sys

sys.path.append(os.path.dirname(__file__)+"/../lib")

import format

import util

def main():

    col_nums = sys.argv[1:]

    if col_nums == []:

        raise SystemExit('Usage: '+sys.argv[0]+' col_num... <map >columns_map')

    col_nums = map(format.str2int_passthru, col_nums) # 0-based

    # Process map

    reader = csv.reader(sys.stdin)

    writer = csv.writer(sys.stdout)

    def subset(row): return util.list_subset(row, col_nums)

    writer.writerow(subset(reader.next()))

    for row in reader: writer.writerow(subset(row))

main()