[Dev OpenGP] [77] Draft for "Message of the day" plugin (fix the bug before use...).

[subversion@xxxxxxxxxxxxx]

Revision: 77
Author:   alband85
Date:     2009-03-24 10:17:27 +0100 (Tue, 24 Mar 2009)

Log Message:
-----------
Draft for "Message of the day" plugin (fix the bug before use...).

Modified Paths:
--------------
    trunk/src/lib/ogp/plugins/plugin.py

Added Paths:
-----------
    trunk/src/lib/ogp/plugins/motd/
    trunk/src/lib/ogp/plugins/motd/__init__.py
    trunk/src/lib/ogp/plugins/motd/motd.py


Added: trunk/src/lib/ogp/plugins/motd/__init__.py
===================================================================
--- trunk/src/lib/ogp/plugins/motd/__init__.py	                        (rev 0)
+++ trunk/src/lib/ogp/plugins/motd/__init__.py	2009-03-24 09:17:27 UTC (rev 77)
@@ -0,0 +1,7 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*
+
+from ogp.plugins.plugin import *
+from motd import *
+
+Plugin.registerPlugin(Motd)


Property changes on: trunk/src/lib/ogp/plugins/motd/__init__.py
___________________________________________________________________
Name: svn:executable
   + *

Added: trunk/src/lib/ogp/plugins/motd/motd.py
===================================================================
--- trunk/src/lib/ogp/plugins/motd/motd.py	                        (rev 0)
+++ trunk/src/lib/ogp/plugins/motd/motd.py	2009-03-24 09:17:27 UTC (rev 77)
@@ -0,0 +1,95 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*
+
+from ogp.plugins.plugin import *
+from os import spawnl, P_WAIT
+
+class Motd(Plugin):
+	name = "motd"
+	files = ['motd', 'motd.tail']
+	__motd_xpath = "/" + OgpXmlConsts.TAG_PLUGIN + "/" + OgpXmlConsts.TAG_FILES + "/" + \
+			OgpXmlConsts.TAG_FILE + "[@" + OgpXmlConsts.ATTR_FILE_NAME + "='motd']" #+ \
+			#"/" + OgpXmlConsts.TAG_CONF
+	__conf_xpath = "/" + OgpXmlConsts.TAG_PLUGIN + "/" + OgpXmlConsts.TAG_CONF 
+	__distro_xpath = "/" + OgpXmlConsts.TAG_PLUGIN + "/" + OgpXmlConsts.TAG_CONF + "/distro" 
+
+	def pushFile(self, file, content, blocking=False):
+		if file == 'motd.tail':
+			print "motd.tail should not be edited! Editing motd..."
+			file = 'motd'
+
+		if file == 'motd':
+			print xpath_arg
+			file_e = self.currentConf.xpath(self.__motd_xpath)[0]
+			file_e.text = content
+			file_e.blocking = blocking
+
+	def pullFile(self, file, fullTree=False):
+		if fullTree:
+			parentConf = self.__core.pullPluginConf(self.parentDn, self.name, fullTree=True)
+			if parentConf is None:
+				parentConf = OgpElement.makePlugin(self.name, self.files)
+			parentConf.merge(self.currentConf)
+			return parentConf.xpath(self.__motd_xpath)[0].text
+		else:
+			# Not working here. Got an XML root issue. 
+			# Got to fix it quickly!
+			print self.currentConf.toString()
+			print "---------------------------------------"
+			xp_test = "//plugin/files/file[@name='motd']"
+			print self.currentConf.xpath(xp_test)
+			tmp = fromstring(self.currentConf.toString())
+			tmp2 = tmp.xpath(self.__motd_xpath)
+			print tmp2
+			print tmp2[0]
+			#print (self.__motd_xpath)
+			#print self.currentConf.xpath("//plugin")
+			#return self.currentConf.xpath(self.__motd_xpath)[0].text
+			return
+
+	def help(self, cmdName=None):
+		if cmdName is None:
+			return {'setdistro': "Sets distribution name."}
+		elif cmdName == 'setdistro':
+			return {'distro': "(string) If set to 'debian', will install to /etc/motd.tail instead of /etc/motd.", 
+					'blocking': "(bool, optionnal) Do not allow inheritance."}
+		else:
+			raise OgpPluginError('help: unknown command (' + cmdName + ')')
+
+	def runCommand(self, cmdName, argv):
+		if cmdName != 'setdistro':
+			raise OgpPluginError('runCommand: unknown command (' + cmdName + ')')
+		else:
+			distro = argv['distro']
+			try:
+				blocking = argv['blocking']
+			except:
+				blocking = False
+			dist_e = self.currentConf.xpath(self.__motd_xpath)
+			if len(dist_e) == 0:
+				dist_e = Element('distro')
+				conf_e = self.currentConf.xpath(self.__conf_xpath)[0]
+				conf_e.append(dist_e)
+			else:
+				dist_e = dist_e[0]
+			dist_e.text = distro
+			dist_e.blocking = blocking
+
+	def installConf(self):
+		motd = self.pullFile('motd', True)
+		dist_e = self.currentConf.xpath(self.__motd_xpath)
+		if len(dist_e) != 0:
+			distro = dist_e[0].text
+		else:
+			distro = None
+
+		if distro == 'debian':
+			f = open('/etc/motd.tail','w')
+			f.write(motd)
+			f.close()
+			spawnl(P_WAIT, '/etc/init.d/bootmisc.sh', 'start')
+		else:
+			f = open('/etc/motd', 'w')
+			f.write(motd)
+			f.close()
+


Property changes on: trunk/src/lib/ogp/plugins/motd/motd.py
___________________________________________________________________
Name: svn:executable
   + *

Modified: trunk/src/lib/ogp/plugins/plugin.py
===================================================================
--- trunk/src/lib/ogp/plugins/plugin.py	2009-03-23 15:47:57 UTC (rev 76)
+++ trunk/src/lib/ogp/plugins/plugin.py	2009-03-24 09:17:27 UTC (rev 77)
@@ -33,6 +33,7 @@
 	files = []
 	__metaclass__ = M_Plugin
 	parentDn = None
+	currentConf = None
 	dn = None
 	__core = None
 	__registeredPlugins = dict()