110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
import os
|
|
import os.path
|
|
import re
|
|
import contextlib
|
|
import tempfile
|
|
|
|
__ALL__ = [ 'DocTemplateError', 'make_doc_from_template' ]
|
|
|
|
VARIABLE_RE = re.compile(r'@[A-Z]{2,3}[0-9]{0,2}')
|
|
|
|
class DocTemplateError(RuntimeError):
|
|
def __str__(self):
|
|
return ' '.join(map(str, self.args))
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def delete_on_error(f):
|
|
'''Context manager which deletes a file if an exception occur'''
|
|
try:
|
|
yield
|
|
except:
|
|
os.remove(f.name)
|
|
raise
|
|
|
|
|
|
def replace_variables(template, variables, ignore_unused=True):
|
|
'''Lookup all substring looking like @XXX9 in the string template, and
|
|
replace them by the value of the key XXX9 in the dictionary variables.
|
|
|
|
Returns a new :class:str
|
|
|
|
:param template: string containing the variables references
|
|
:param variables: dictionary containing the variables values
|
|
|
|
'''
|
|
needed_vars = set([v[1:] for v in re.findall(VARIABLE_RE, template)])
|
|
given_vars = set(variables.keys())
|
|
if given_vars != needed_vars:
|
|
missing = needed_vars - given_vars
|
|
if missing or not ignore_unused:
|
|
unused = given_vars - needed_vars
|
|
raise DocTemplateError(
|
|
'Mismatch between given and needed variables: missing={0} unused={1}'
|
|
.format(map(repr, missing), map(repr, unused)))
|
|
def variable_replacement(match_obj):
|
|
return variables[match_obj.group(0)[1:]]
|
|
return re.sub(VARIABLE_RE, variable_replacement, template)
|
|
|
|
|
|
def char_to_rtf(c):
|
|
if ord(c) < 128:
|
|
return c
|
|
else:
|
|
return '\u%d?' % ord(c)
|
|
|
|
|
|
def utf8_to_rtf(s):
|
|
s = ''.join([ char_to_rtf(c) for c in s])
|
|
return '{\uc1{%s}}' % s
|
|
|
|
|
|
def unicode_to_rtf(value):
|
|
try:
|
|
value = unicode(value)
|
|
except Exception, e:
|
|
raise DocTemplateError('Unable to get a unicode value', e)
|
|
return utf8_to_rtf(value)
|
|
|
|
|
|
def variables_to_rtf(variables):
|
|
return dict(((k, unicode_to_rtf(v)) for k,v in variables.iteritems()))
|
|
|
|
|
|
def make_doc_from_template(from_path, to_path, variables, persistent):
|
|
'''Use file from_path as a template to combine with the variables
|
|
dictionary and place the result in the file to_path.
|
|
Encode value of variable into encoding of UTF-8 for the RTF file format.
|
|
|
|
:param from_path: the template file path
|
|
:param to_path: the newly created file containing the result of the templating
|
|
:param variables: the dictionary of variables to replace
|
|
'''
|
|
|
|
if not os.path.exists(from_path):
|
|
raise DocTemplateError('Template file does not exist', repr(from_path))
|
|
if os.path.exists(to_path):
|
|
raise DocTemplateError('Destination file already exists', repr(to_path))
|
|
variables = variables_to_rtf(variables)
|
|
if persistent:
|
|
with open(to_path, 'w') as to_file:
|
|
with open(from_path) as from_file:
|
|
with delete_on_error(to_file):
|
|
to_file.write(replace_variables(from_file.read(), variables))
|
|
else:
|
|
with tempfile.NamedTemporaryFile(prefix=to_path,
|
|
delete=False) as to_file:
|
|
with open(from_path) as from_file:
|
|
with delete_on_error(to_file):
|
|
to_file.write(replace_variables(from_file.read(), variables))
|
|
return to_file.name
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
try:
|
|
variables = dict((map(lambda x: unicode(x, 'utf-8'), arg.split('=')) for arg in sys.argv[3:]))
|
|
make_doc_from_template(sys.argv[1], sys.argv[2], variables)
|
|
except Exception, e:
|
|
raise
|
|
print 'Usage: python doc_template.py <template.rtf> <output.rtf> [KEY1=value1 KEY2=value2 ...]'
|