Power management and runlevels

configs/main.conf

@@ -9,6 +9,9 @@ enabled = yes
 [system_services]
 enabled = yes
 
+[machine]
+enabled = yes
+
 [common_info]
 enabled = yes
 

functions/fmod_machine.py

+# -*- coding: utf-8 -*-
+
+
+import subprocess
+
+from settingsd import config
+from settingsd import service
+from settingsd import shared
+from settingsd import logger
+from settingsd import validators
+
+
+##### Private constants #####
+SERVICE_NAME = "machine"
+
+POWER_METHODS_NAMESPACE = "power"
+RUNLEVELS_METHODS_NAMESPACE = "runlevels"
+
+RUNLEVELS = "0123456"
+
+SHUTDOWN_OPTION_HALT = "-h"
+SHUTDOWN_OPTION_REBOOT = "-r"
+
+
+##### Exceptions #####
+class SubprocessFailure(Exception) :
+	pass
+
+
+##### Private classes #####
+class Machine(service.FunctionObject) :
+
+	### DBus methods ###
+
+	@service.functionMethod(POWER_METHODS_NAMESPACE, out_signature="i")
+	def shutdown(self) :
+		proc_args = "%s %s now" % (config.value(SERVICE_NAME, "shutdown_prog_path"), SHUTDOWN_OPTION_HALT)
+		return self.execProcess(proc_args)[2]
+
+	@service.functionMethod(POWER_METHODS_NAMESPACE, out_signature="i")
+	def reboot(self) :
+		proc_args = "%s %s now" % (config.value(SERVICE_NAME, "shutdown_prog_path"), SHUTDOWN_OPTION_REBOOT)
+		return self.execProcess(proc_args)[2]
+
+	@service.functionMethod(POWER_METHODS_NAMESPACE, out_signature="i")
+	def suspend(self) :
+		return self.execProcess(config.value(SERVICE_NAME, "pm_suspend_prog_path"))[2]
+
+	@service.functionMethod(POWER_METHODS_NAMESPACE, out_signature="i")
+	def hibernate(self) :
+		return self.execProcess(config.value(SERVICE_NAME, "pm_hibernate_prog_path"))[2]
+
+	###
+
+	@service.functionMethod(RUNLEVELS_METHODS_NAMESPACE, in_signature="i", out_signature="i")
+	def switchTo(self, level) :
+		proc_args = "%s %s" % (config.value(SERVICE_NAME, "telinit_prog_path"), validators.validRange(str(level), RUNLEVELS))
+		return self.execProcess(proc_args)[2]
+
+	@service.functionMethod(RUNLEVELS_METHODS_NAMESPACE, out_signature="i")
+	def currentLevel(self) :
+		proc_args = config.value(SERVICE_NAME, "runlevel_prog_path")
+		(proc_stdout, proc_stderr, proc_returncode) = self.execProcess(proc_args)
+
+		level_pairs_list = proc_stdout.strip().split(" ")
+		if len(level_pairs_list) != 2 or not level_pairs_list[1] in RUNLEVELS :
+			raise SubprocessFailure("Error while execute \"%s\"\nStdout: %s\nStderr: %s\nReturn code: %d" % (
+				proc_args, proc_stdout.strip(), proc_stderr.strip(), proc_returncode ))
+
+		return int(level_pairs_list[1])
+
+	@service.functionMethod(RUNLEVELS_METHODS_NAMESPACE, out_signature="i")
+	def previousLevel(self) :
+		proc_args = config.value(SERVICE_NAME, "runlevel_prog_path")
+		(proc_stdout, proc_stderr, proc_returncode) = self.execProcess(proc_args)
+
+		level_pairs_list = proc_stdout.strip().split(" ")
+		if len(level_pairs_list) != 2 or not level_pairs_list[1] in RUNLEVELS+"N" :
+			raise SubprocessFailure("Error while execute \"%s\"\nStdout: %s\nStderr: %s\nReturn code: %d" % (
+				proc_args, proc_stdout.strip(), proc_stderr.strip(), proc_returncode ))
+
+		return ( int(level_pairs_list[0]) if level_pairs_list[0] in RUNLEVELS else -1 )
+
+
+	### Private ###
+
+	def execProcess(self, proc_args) :
+		logger.debug("{mod}: Executing child process \"%s\"" % (proc_args))
+		proc = subprocess.Popen(proc_args, shell=True, bufsize=1024, close_fds=True,
+			stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+			env={ "LC_ALL" : "C" })
+		(proc_stdout, proc_stderr) = proc.communicate()
+		logger.debug("{mod}: Child process \"%s\" finished, return_code=%d" % (proc_args, proc.returncode))
+
+		return (proc_stdout, proc_stderr, proc.returncode)
+
+
+##### Public classes #####
+class Service(service.Service) :
+
+	### Public ###
+
+	def initService(self) :
+		shared.Functions.addSharedObject(SERVICE_NAME, Machine(SERVICE_NAME, self))
+
+	###
+
+	@classmethod
+	def serviceName(self) :
+		return SERVICE_NAME
+
+	@classmethod
+	def options(self) :
+		return [
+			(SERVICE_NAME, "shutdown_prog_path", "/sbin/shutdown", str),
+			(SERVICE_NAME, "pm_suspend_prog_path", "/usr/sbin/pm-suspend", str),
+			(SERVICE_NAME, "pm_hibernate_prog_path", "/usr/sbin/pm-hibernate", str),
+			(SERVICE_NAME, "telinit_prog_path", "/sbin/telinit", str),
+			(SERVICE_NAME, "runlevel_prog_path", "/sbin/runlevel", str)
+		]
+