#!/usr/bin/env python

# Translates a data dictionary spreadsheet to Redmine formatting

import csv

import re

import sys

# Spreadsheet format

source_sep = ', '

##### URLs

url_comment_re = r'(?:\([^)]*\))'

def simplify_url(url): return re.sub(r'\(.*?\)', r'', url)

def url_comment_text(comment):

    if not comment: return comment

    match = re.match(r'^\(\d*:?(.*?)\)$', comment)

    assert match

    text, = match.groups()

    return text

def url_term(url):

    '''Prefixes any provider in the URL to the term name, to create a namespace.

    Each hierarchical component of the provider is stored in a URL comment.

    '''

    match = re.match(r'^('+url_comment_re+r'*).*?([\w:-]+)$', url)

    assert match

    provider_str, term = match.groups()

    provider = filter(bool, map(url_comment_text,

        re.findall(url_comment_re, provider_str)))

    return ':'.join(provider+[term])

##### Redmine

def redmine_bold(text): return '*'+text+'*'

def redmine_url(text, url):

    if url: return '"'+text+'":'+url

    else: return text

def source2redmine_url(url):

    if url: return redmine_url(url_term(url), simplify_url(url))

    else: return ''

class RedmineTableWriter:

    '''Formats rows as a Redmine table'''

    def __init__(self, stream):

        self.stream = stream

    def writerow(self, row):

        self.stream.write(('|'.join(['']+row+['']))+'\n')

class RedmineDictWriter:

    '''Formats rows as Redmine sections containing a table'''

    def __init__(self, term_col, def_col, header, stream):

        header.pop(term_col)

        header.pop(def_col)

        self.term_col = term_col

        self.def_col = def_col

        self.header = header

        self.stream = stream

    def writerow(self, row):

        term = row[self.term_col]

        def_ = row[self.def_col]

        self.stream.write('h3. '+term+'\n\n')

        if def_: self.stream.write(def_+'\n\n')

        table = RedmineTableWriter(self.stream)

        for i, col_name in enumerate(self.header):

            if i not in set([self.term_col, self.def_col]):

                value = row[i]

                if value: table.writerow([redmine_bold(col_name), value])

        self.stream.write('\n')

def main():

    try: _prog_name, term_col, def_col, sources_col = sys.argv

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

        +' <spreadsheet term_col# definition_col# sources_col# >redmine')

    term_col, def_col, sources_col = map(int, [term_col, def_col, sources_col])

    # Translate input

    reader = csv.reader(sys.stdin)

    header = reader.next()

    writer = RedmineDictWriter(term_col, def_col, header, sys.stdout)

    for row in reader:

        term = row[term_col]

        sources = row[sources_col].split(source_sep)

        row[term_col] = redmine_url(term, simplify_url(sources[0]))

        row[sources_col] = source_sep.join(map(source2redmine_url, sources))

        writer.writerow(row)

main()