aboutsummaryrefslogtreecommitdiffstats
path: root/templatefile.py
diff options
context:
space:
mode:
authorPeter Seebach <peter.seebach@windriver.com>2010-11-17 17:11:58 -0600
committerPeter Seebach <peter.seebach@windriver.com>2010-11-18 19:05:05 -0600
commitca54fdb785b17fb251c28b28f51d0e666b1f2e22 (patch)
tree9c85cae7e2d2b14518dee0721c52ad9b0431df25 /templatefile.py
parent105b1756b959b5e63e98ed0d86385e98c93cfd28 (diff)
downloadpseudo-ca54fdb785b17fb251c28b28f51d0e666b1f2e22.tar.gz
pseudo-ca54fdb785b17fb251c28b28f51d0e666b1f2e22.tar.bz2
pseudo-ca54fdb785b17fb251c28b28f51d0e666b1f2e22.zip
Messing with makewrappers: Move templating code out (for planned use
in an incoming "maketables".)
Diffstat (limited to 'templatefile.py')
-rw-r--r--templatefile.py99
1 files changed, 99 insertions, 0 deletions
diff --git a/templatefile.py b/templatefile.py
new file mode 100644
index 0000000..e08bc64
--- /dev/null
+++ b/templatefile.py
@@ -0,0 +1,99 @@
+from string import Template
+import os
+
+class TemplateFile:
+ """A template for creating a source file"""
+
+ def __init__(self, path):
+ # default values...
+ # no name or file yet
+ self.name = ''
+ self.sections = {}
+ self.file = None
+ self.path = None
+ # open a new file for each item
+ self.file_per_item = False
+
+ # empty footer if none specified:
+ self.sections['footer'] = []
+
+ # lines appended to body by default
+ self.sections['body'] = []
+ current = self.sections['body']
+
+ self.template = open(path)
+ for line in self.template:
+ line = line.rstrip()
+ if line.startswith('@'):
+ if ' ' in line:
+ leading, trailing = line.split(' ', 1)
+ else:
+ leading, trailing = line, None
+
+ if leading == '@name':
+ if not trailing:
+ raise Exception("@name requires a file name.")
+ self.path = trailing
+ if '$' in self.path:
+ self.file_per_item = True
+ else:
+ section = leading[1:]
+ if section not in self.sections:
+ self.sections[section] = []
+ current = self.sections[section]
+ else:
+ current.append(line)
+ self.template.close()
+ for section, data in self.sections.items():
+ self.sections[section] = Template("\n".join(data))
+
+ # You need a file if this isn't a file-per-item
+ if not self.file_per_item:
+ self.file = open(self.path, 'w')
+
+ def close(self):
+ """Close the associated file."""
+ if self.file:
+ self.file.close()
+ self.file = None
+
+ def __repr__(self):
+ strings = []
+ if self.file_per_item:
+ strings.append("path: %s (per item)" % self.path)
+ else:
+ strings.append("path: %s" % self.path)
+ for name, data in self.sections.items():
+ strings.append("%s:" % name)
+ strings.append(data.safe_substitute({}))
+ return "\n".join(strings)
+
+ def emit(self, template, item=None):
+ """Emit a template, with optional interpolation of an item."""
+ if self.file_per_item:
+ if not item:
+ return
+ path = Template(self.path).safe_substitute(item)
+ if os.path.exists(path):
+ # print "We don't overwrite existing files."
+ return
+ self.file = open(path, 'w')
+ if not self.file:
+ print "Couldn't open '%s' (expanded from %s), " \
+ "not emitting '%s'." % \
+ (path, self.path, template)
+ return
+
+ if template == "copyright":
+ # hey, at least it's not a global variable, amirite?
+ self.file.write(TemplateFile.copyright)
+ elif template in self.sections:
+ templ = self.sections[template]
+ self.file.write(templ.safe_substitute(item))
+ self.file.write("\n")
+ else:
+ print "Warning: Unknown template '%s'." % template
+
+ if self.file_per_item:
+ self.file.close()
+ self.file = None