#!/usr/bin/env python

# Converts a CSV dataset to XML using a map spreadsheet

import copy

import csv

import os.path

import re

import sys

from xml.dom.minidom import getDOMImplementation

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

import xpath

def main():

    prog_name = sys.argv[0]

    try: prog_name, map_path = sys.argv

    except ValueError:

        raise Exception('Usage: '+prog_name+' map_path <dataset >output')

    # Get map

    map = {}

    has_types = False # whether outer elements are type containiners

    stream = open(map_path, 'rb')

    reader = csv.reader(stream, delimiter=',', quotechar='"')

    src, dest = reader.next()[:2]

    for row in reader:

        name, path = row[:2]

        if name != '' and path != '':

            if path.startswith('/*s/'): has_types = True # *s used for type elem

            path = path.replace('<name>', name)

            map[name] = xpath.XpathParser(path).parse()

    stream.close()

    # Process dataset

    doc = getDOMImplementation().createDocument(None, dest, None)

    stream = sys.stdin

    reader = csv.reader(stream, delimiter=',', quotechar='"')

    fieldnames = reader.next()

    row_idx = 0

    for row in reader:

        row_id = str(row_idx)

        for idx, name in enumerate(fieldnames):

            value = row[idx]

            if value != '' and name in map:

                path = copy.deepcopy(map[name]) # don't modify main value!

                xpath.set_id(path, row_id, has_types)

                xpath.set_value(path, value)

                xpath.get(doc, path, True)

        row_idx += 1

    doc.writexml(sys.stdout, addindent='    ', newl='\n')

main()