|
1
|
#!/usr/bin/env python
|
|
2
|
# Translates a data dictionary spreadsheet to Redmine formatting
|
|
3
|
|
|
4
|
import csv
|
|
5
|
import re
|
|
6
|
import sys
|
|
7
|
|
|
8
|
# Spreadsheet format
|
|
9
|
source_sep = ', '
|
|
10
|
|
|
11
|
##### URLs
|
|
12
|
|
|
13
|
url_comment_re = r'(?:\([^)]*\))'
|
|
14
|
|
|
15
|
def simplify_url(url): return re.sub(r'\(.*?\)', r'', url)
|
|
16
|
|
|
17
|
def url_comment_text(comment):
|
|
18
|
if not comment: return comment
|
|
19
|
|
|
20
|
match = re.match(r'^\(\d*:?(.*?)\)$', comment)
|
|
21
|
assert match
|
|
22
|
text, = match.groups()
|
|
23
|
return text
|
|
24
|
|
|
25
|
def url_term(url):
|
|
26
|
'''Prefixes any provider in the URL to the term name, to create a namespace.
|
|
27
|
Each hierarchical component of the provider is stored in a URL comment.
|
|
28
|
'''
|
|
29
|
match = re.match(r'^('+url_comment_re+r'*).*?([\w:-]+)$', url)
|
|
30
|
assert match
|
|
31
|
provider_str, term = match.groups()
|
|
32
|
provider = filter(bool, map(url_comment_text,
|
|
33
|
re.findall(url_comment_re, provider_str)))
|
|
34
|
|
|
35
|
return ':'.join(provider+[term])
|
|
36
|
|
|
37
|
##### Redmine
|
|
38
|
|
|
39
|
sp = ' ' # Unicode thin space
|
|
40
|
|
|
41
|
def redmine_bold(text): return '*'+text+'*'
|
|
42
|
|
|
43
|
def redmine_url(text, url):
|
|
44
|
if not url: return text
|
|
45
|
elif url.find('://') >= 0: return '"'+text+'":'+url # external link
|
|
46
|
else: return '[['+url+'|'+text+']]' # internal link
|
|
47
|
|
|
48
|
def source2redmine_url(url):
|
|
49
|
if url: return redmine_url(url_term(url), simplify_url(url))
|
|
50
|
else: return ''
|
|
51
|
|
|
52
|
def redmine_add_links(page_name, text):
|
|
53
|
# Link citations to entry in sources list
|
|
54
|
text = re.sub(r'(?<=\[)(.*?)(?=\])', sp+'[['+page_name+r'#\g<0>|\g<0>]]'+sp,
|
|
55
|
text) # sp prevents [[[, which Redmine would parse incorrectly
|
|
56
|
return text
|
|
57
|
|
|
58
|
class RedmineTableWriter:
|
|
59
|
'''Formats rows as a Redmine table'''
|
|
60
|
|
|
61
|
def __init__(self, stream):
|
|
62
|
self.stream = stream
|
|
63
|
|
|
64
|
def writerow(self, row):
|
|
65
|
self.stream.write(('|'.join(['']+row+['']))+'\n')
|
|
66
|
|
|
67
|
class RedmineDictWriter:
|
|
68
|
'''Formats rows as Redmine sections containing a table'''
|
|
69
|
|
|
70
|
def __init__(self, term_col, def_col, header, stream):
|
|
71
|
self.term_col = term_col
|
|
72
|
self.def_col = def_col
|
|
73
|
self.header = header
|
|
74
|
self.page_name = header[term_col]
|
|
75
|
self.stream = stream
|
|
76
|
|
|
77
|
def writerow(self, row):
|
|
78
|
row = map(lambda f: redmine_add_links(self.page_name, f), row)
|
|
79
|
|
|
80
|
term = row[self.term_col]
|
|
81
|
def_ = row[self.def_col]
|
|
82
|
self.stream.write('h3. '+term+'\n\n')
|
|
83
|
if def_: self.stream.write(def_+'\n\n')
|
|
84
|
|
|
85
|
table = RedmineTableWriter(self.stream)
|
|
86
|
for i, col_name in enumerate(self.header):
|
|
87
|
if not i in set([self.term_col, self.def_col]):
|
|
88
|
value = row[i]
|
|
89
|
if value: table.writerow([redmine_bold(col_name), value])
|
|
90
|
self.stream.write('\n')
|
|
91
|
|
|
92
|
|
|
93
|
def main():
|
|
94
|
try: _prog_name, term_col, def_col, sources_col = sys.argv
|
|
95
|
except ValueError: raise SystemExit('Usage: '+sys.argv[0]
|
|
96
|
+' <spreadsheet term_col# definition_col# sources_col# >redmine')
|
|
97
|
term_col, def_col, sources_col = map(int, [term_col, def_col, sources_col])
|
|
98
|
|
|
99
|
# Translate input
|
|
100
|
reader = csv.reader(sys.stdin)
|
|
101
|
header = reader.next()
|
|
102
|
writer = RedmineDictWriter(term_col, def_col, header, sys.stdout)
|
|
103
|
for row in reader:
|
|
104
|
term = row[term_col]
|
|
105
|
sources = row[sources_col].split(source_sep)
|
|
106
|
|
|
107
|
row[term_col] = redmine_url(term, simplify_url(sources[0]))
|
|
108
|
row[sources_col] = source_sep.join(map(source2redmine_url, sources))
|
|
109
|
|
|
110
|
writer.writerow(row)
|
|
111
|
|
|
112
|
main()
|
