#!/usr/bin/env python

# Translates a spreadsheet column using a thesaurus.

# The column header is also translated. CSVs without a header are supported.

# Unrecognized names are left untouched, permitting successive runs on different

# thesauruses.

# Case- and punctuation-sensitive. (Use canon first for case-insensitivity.)

import csv

import sys

def main():

    try: _prog_name, col_num, dict_path = sys.argv

    except ValueError: raise SystemExit('Usage: '+sys.argv[0]

        +' <in col# thesaurus [| '+sys.argv[0]+' col# thesaurus_2]... >out')

    col_num = int(col_num)

    # Get thesaurus

    dict_ = {}

    stream = open(dict_path, 'rb')

    reader = csv.reader(stream)

    for row in reader: dict_[row[0]] = row[1]

    stream.close()

    # Translate input

    reader = csv.reader(sys.stdin)

    writer = csv.writer(sys.stdout)

    for row in reader:

        term = row[col_num]

        try: row[col_num] = dict_[term]

        except KeyError: pass

        writer.writerow(row)

main()