#!/usr/bin/env python

# Modifies a map spreadsheet A->B or any file using a replacements spreadsheet

# A->C or B->C

import csv

import os.path

import re

import sys

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

import maps

import opts

def main():

    env_names = []

    def usage_err():

        raise SystemExit('Usage: '+opts.env_usage(env_names, True)+' '

            +sys.argv[0]+' <map repl [col_num] [| '+sys.argv[0]

            +' repl_1 [col_num_1]]... >new_map')

    text = opts.env_flag('text', False, env_names) # all patterns are plain text

    try: _prog_name, repl_path = sys.argv[:2]

    except ValueError: usage_err()

    col_num = None

    try: col_num = sys.argv[2]

    except IndexError: pass

    if col_num != None: col_num = int(col_num) # 0-based

    # Get replacements

    repls = []

    stream = open(repl_path, 'rb')

    reader = csv.reader(stream)

    repl_in, repl_out = reader.next()[:2]

    for row in reader:

        in_, out = row[:2]

        if in_ != '':

            if text or re.match(r'^\w+$', in_):

                in_ = r'(?<![^\W_])'+re.escape(in_)+r'(?![^\W_])' # match word

            repls.append((r'(?m)'+in_, out))

    stream.close()

    def repl_all(str_):

        for repl, with_ in repls: str_ = re.sub(repl, with_, str_)

        return str_

    # Modify map or file

    if col_num != None:

        reader = csv.reader(sys.stdin)

        writer = csv.writer(sys.stdout)

        cols = reader.next()

        label, root = maps.col_info(cols[col_num])[:2]

        if label != repl_in: raise SystemExit('Map error: Map column '+

            str(col_num)+' label "'+label+'" doesn\'t match replacements input '

            'column label "'+repl_in+'"')

        cols[col_num] = repl_out+sep+repl_all(root)

        writer.writerow(cols)

        for row in reader:

            row[col_num] = repl_all(row[col_num])

            writer.writerow(row)

    else: sys.stdout.write(repl_all(sys.stdin.read()))

main()