Add plugin api to template.py, add news plugin

1 files changed, 81 insertions, 85 deletions

diff --git a/scripts/template.py b/scripts/template.py
index a4d4e9e..c5d348a 100755
--- a/scripts/template.py
+++ b/scripts/template.py
@@ -1,8 +1,15 @@
 #!/usr/bin/env python
 #Filename: newtemp.py
 #Author: stayawake@lavabit.com
-
-
+# ---
+# Copyright (C) 2011 Christian Franke <nobody@nowhere.ws>
+#
+# Permission is hereby granted to use and distribute this
+# file for any purpose, provided the above copyright notice
+# is kept in place.
+#
+
+import imp
 import os
 import os.path
 import sys
@@ -12,104 +19,93 @@ import datetime
 from string import Template
 import ConfigParser
 
-def readdir(directory):
-	return_value = {"content.html":"", "keywords":"", "desc_de":"", "desc_en":""}
-	
-	for c in return_value:
-		f = open(os.path.join(directory, c), "r")
-		return_value[c] = f.read()
-		f.close
-	return return_value
-
-def substitute(target, prefix, htmltemplate, datenow, topic, content):
-	
-	#create template-string
-	htmltext = Template(htmltemplate)
-	htmltext = htmltext.substitute(
-		template_date = datenow,
-		template_chattopic = topic,
-		template_keywords = content["keywords"],
-		template_content = content["content.html"],
-		template_desc_de = content["desc_de"],
-		template_desc_en = content["desc_en"]
-	)
-
-	#write it to the outputfile
-	outputfile = open(os.path.join(target, page + ".html"), "w")
-	outputfile.write(htmltext)
-	outputfile.close()
-
-
-def readarguments(argv):
-	
-	global configfile
-	global verbose
-
-	try:
-		opts, args = getopt.getopt(argv, "c:vh", ["config=", "verbose", "help"])
-	except getopt.GetoptError:
-		print "There is a problem with your argument(s)"
-		sys.exit(2)
-	for opt, arg in opts:
-		if opt in ("-c", "--config"):
-			configfile = arg
-		elif opt in ("-v", "--verbose"):
-			verbose = True
-		elif opt in ("-h", "--help"):
-			print "Please read the non-existing man page for further assistance"
-			sys.exit(0)
-
+import plugin
+
+class Page:
+	def __init__(self, name):
+		self.name = name
+		data = {"content.html":"", "keywords":"", "desc_de":"", "desc_en":""}
+
+		for c in data:
+			f = open(os.path.join(pagepath, name, c), "r")
+			data[c] = f.read()
+			f.close
+
+		self._data = plugin.plugin_manager.process_content(data)
+
+	@property
+	def html(self):
+		#create template-string
+		htmltext = Template(htmltemplate)
+		htmltext = htmltext.substitute(
+			template_date = datenow,
+			template_chattopic = topic,
+			template_keywords = self._data["keywords"],
+			template_content = self._data["content.html"],
+			template_desc_de = self._data["desc_de"],
+			template_desc_en = self._data["desc_en"]
+		)
+		return htmltext
+
+	def render_and_save(self):
+		#write it to the outputfile
+		outputfile = open(os.path.join(targetdir, self.name + '.html'), "w")
+		outputfile.write(self.html)
+		outputfile.close()
 
 configfile = "template.conf"
 verbose = False
+opts, args = getopt.getopt(sys.argv[1:], "c:vh", ["config=", "verbose", "help"])
+for opt, arg in opts:
+	if opt in ("-c", "--config"):
+		configfile = arg
+	elif opt in ("-v", "--verbose"):
+		verbose = True
+	elif opt in ("-h", "--help"):
+		print "Please read the non-existing man page for further assistance"
+		sys.exit(0)
+
 basedir = os.path.dirname(os.path.abspath(__file__))
 
-readarguments(sys.argv[1:])
+def makeabs(path):
+	if not os.path.isabs(path):
+		return os.path.join(basedir, path)
+	return path
 
+configfile = makeabs(configfile)
 config = ConfigParser.ConfigParser()
-if not os.path.isabs(configfile):
-	configfile = os.path.join(basedir, configfile)
 config.read(configfile)
-path = config.get("Default", "path")
-if not os.path.isabs(path):
-	path = os.path.join(basedir, path)
-
-target = config.get("Default", "target")
-if not os.path.isabs(target):
-	target = os.path.join(basedir, target)
 
+path = makeabs(config.get("Default", "path"))
+targetdir = makeabs(config.get("Default", "target"))
 pagepath = os.path.join(path, "pages")
 templatepath = os.path.join(path, "template")
 topicpath = os.path.join(path, "topic")
 
-if verbose:
-	print "Pagepath: %s" % pagepath
-	print "Templatepath: %s" % templatepath
-	print "topicpath: %s" % topicpath
-
 #set template
-f = open(os.path.join(templatepath, "template.html"), "r")
-htmltemplate = f.read()
-f.close
-if verbose:
-	print "template read"
-
-#set topic
-f = open(os.path.join(topicpath, "topic"), "r")
-topic = f.read()
-f.close
-if verbose:
-	print "topic read"
-
-#set date
+htmltemplate = open(os.path.join(templatepath, "template.html"), "r").read()
+topic = open(os.path.join(topicpath, "topic"), "r").read()
 datenow = time.strftime("%Y-%m-%dT%H:%M:%S +0100")
-if verbose:
-	print "Set date: " + datenow
 
-for page in os.listdir(pagepath):
-	prefix = os.path.join(pagepath, page)
-	page_data = readdir(prefix)
-	substitute(target, prefix, htmltemplate, datenow, topic, page_data)
+if __name__ == '__main__':
 	if verbose:
-		print "%s.html written" % prefix
+		print "Pagepath: %s" % pagepath
+		print "Templatepath: %s" % templatepath
+		print "topicpath: %s" % topicpath
+		print "datenow: %s" % datenow
+
+	pluginlist = config.items('Plugins')
+	for pluginname, pluginarg in pluginlist:
+		modinfo = imp.find_module(pluginname)
+		try:
+			imp.load_module(pluginname, *modinfo)
+		finally:
+			modinfo[0].close()
+
+if __name__ == '__main__':
+	for page in os.listdir(pagepath):
+		Page(page).render_and_save()
+		if verbose:
+			print "%s.html written" % prefix
+
 # vi: noexpandtab:tabstop=8:shiftwidth=8