# Combines two map spreadsheets A0->B and A1->C to A->B, with B overwriting C

# Multi-safe (supports an input appearing multiple times). Note that if there is

# *any* non-empty mapping for an input in map_0, all mappings for that input in

import os.path

import sys

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

import opts

import util

def non_empty(row): return row[0] != '' and row[1] != ''

def main():

    ignore = opts.env_flag('ignore')

    header_num = util.cast(int, opts.get_env_var('header_num'))

        # selects which map's header to use as the output header

    all_ = opts.env_flag('all') # doesn't merge mappings that are in both maps

    try: _prog_name, map_1_path = sys.argv

        raise SystemExit('Usage: env [ignore=1] [header_num={0|1}] [all=1] '

    headers = [None]*2

    # Open map 0

    stream = open(map_1_path, 'rb')

    map_1_reader = csv.reader(stream)

    # Check col labels

    combinable = maps.combinable(*headers)

        raise SystemExit('Map error: '

        'Map 0 column 0 label doesn\'t contain map 1 column 0 label')

    # Pass through map 0, storing which mappings it defines

    writer = csv.writer(sys.stdout)

    writer.writerow(maps.merge_headers(*headers, **dict(prefer=header_num)))

    mappings = set()

    for row in map_0_reader:

        if all_ or non_empty(row):

            if not all_: mappings.add(row[0])

            writer.writerow(row)

        # Add mappings in map 1 that weren't already defined or mapped to

        for row in map_1_reader:

            if non_empty(row) and row[0] not in mappings: writer.writerow(row)