cryptonas-branches/pythonrewrite/plugins/shutdown/shutdown.py

import CryptoBoxPlugin

REDIRECT_DELAY = 180

class shutdown(CryptoBoxPlugin.CryptoBoxPlugin):

	pluginCapabilities = [ "system", "menu" ]
	requestAuth = False
	rank = 70

	def doAction(self, type=None):
		if not type:
			return "form_shutdown"
		elif type == "shutdown":
			if self.__doShutdown("shutdown"):
				self.hdf["Data.Success"] = "Plugins.shutdown.Shutdown"
				return None
			else:
				self.hdf["Data.Warning"] = "Plugins.shutdown.ShutdownFailed"
				return "form_shutdown"
		elif type == "reboot":
			if self.__doShutdown("reboot"):
				self.hdf["Data.Success"] = "Plugins.shutdown.Reboot"
				self.hdf["Data.Redirect.URL"] = ""
				self.hdf["Data.Redirect.Delay"] = REDIRECT_DELAY
				return None
			else:
				self.hdf["Data.Warning"] = "Plugins.shutdown.RebootFailed"
				return "form_shutdown"
		else:
			return "form_shutdown"


	def getStatus(self):
		return "the box is up'n'running"


	def __doShutdown(self, action):
		import subprocess
		import os
		proc = subprocess.Popen(
			shell = False,
			args = [
				self.cbox.prefs["Programs"]["super"],
				self.cbox.prefs["Programs"]["CryptoBoxRootActions"],
				"plugin",
				os.path.join(self.pluginDir, "root_action.py"),
				action])
		proc.communicate()
		return proc.returncode == 0