#!/usr/bin/env python

# Subtracts map spreadsheet A1->C from A0->B to A->B

import csv

import os.path

import sys

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

import opts

import util

def main():

    ignore = opts.env_flag('ignore')

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

    except ValueError:

        raise SystemExit('Usage: env [ignore=1] '+sys.argv[0]+' <from_map '

            'subtract_map [compare_col_num...] [| '+sys.argv[0]

            +' subtract_map_2]... >difference_map')

    compare_col_nums = map(int, sys.argv[2:]) # 0-based

    if compare_col_nums == []:

        compare_col_nums = None # list_subset() value for all columns

    def compare_on(row): return tuple(util.list_subset(row, compare_col_nums))

    # Get map 1

    compare_cols = set()

    stream = open(map_1_path, 'rb')

    reader = csv.reader(stream)

    map_1_cols = reader.next()

    for row in reader:

        if row[0] != '': compare_cols.add(compare_on(row))

    stream.close()

    # Open map 0

    reader = csv.reader(sys.stdin)

    map_0_cols = reader.next()

    if not ignore and map_0_cols[0] != map_1_cols[0]: raise SystemExit(

        'Map error: Map 1 column 0 name doesn\'t match map 0 column 0 name')

    # Subtract map 1 from map 0

    writer = csv.writer(sys.stdout)

    writer.writerow(map_0_cols)

    for row in reader:

        if compare_on(row) not in compare_cols: writer.writerow(row)

main()