Project

General

Profile

1
#!/usr/bin/env python
2
# Selects columns from a spreadsheet
3

    
4
import csv
5
import os.path
6
import sys
7

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

    
10
import util
11

    
12
def main():
13
    col_nums = sys.argv[1:]
14
    if col_nums == []:
15
        raise SystemExit('Usage: '+sys.argv[0]+' col_num... <map >columns_map')
16
    if col_nums[0] == '*': col_nums = None # list_subset() value for all columns
17
    else: col_nums = map(int, col_nums) # 0-based
18
    
19
    # Process map
20
    reader = csv.reader(sys.stdin)
21
    writer = csv.writer(sys.stdout)
22
    def subset(row): return util.list_subset(row, col_nums)
23
    writer.writerow(subset(reader.next()))
24
    for row in reader: writer.writerow(subset(row))
25

    
26
main()
(5-5/43)