diff options
author | Julio Reyes <nemesiscodex@gmail.com> | 2013-07-05 14:13:11 (GMT) |
---|---|---|
committer | Julio Reyes <nemesiscodex@gmail.com> | 2013-07-05 14:13:11 (GMT) |
commit | b6db5b8af0af7ebfdd44cb53b09d63b819989338 (patch) | |
tree | ca439018bfa1c34c97bdaed4ec0f20bc36bd678f /src/jarabe/model/session.py |
Diffstat (limited to 'src/jarabe/model/session.py')
-rw-r--r-- | src/jarabe/model/session.py | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/src/jarabe/model/session.py b/src/jarabe/model/session.py new file mode 100644 index 0000000..4e66bdc --- /dev/null +++ b/src/jarabe/model/session.py @@ -0,0 +1,113 @@ +# Copyright (C) 2008, Red Hat, Inc. +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + +import gtk +import dbus +import os +import signal +import sys +import logging + +from sugar import session +from sugar import env + + +_session_manager = None + + +def have_systemd(): + return os.access("/sys/fs/cgroup/systemd", 0) >= 0 + + +class SessionManager(session.SessionManager): + MODE_LOGOUT = 0 + MODE_SHUTDOWN = 1 + MODE_REBOOT = 2 + + def __init__(self): + session.SessionManager.__init__(self) + self._logout_mode = None + + def logout(self): + self._logout_mode = self.MODE_LOGOUT + self.initiate_shutdown() + + def shutdown(self): + self._logout_mode = self.MODE_SHUTDOWN + self.initiate_shutdown() + + def reboot(self): + self._logout_mode = self.MODE_REBOOT + self.initiate_shutdown() + + def shutdown_completed(self): + if env.is_emulator(): + self._close_emulator() + elif self._logout_mode != self.MODE_LOGOUT: + bus = dbus.SystemBus() + if have_systemd(): + try: + proxy = bus.get_object('org.freedesktop.login1', + '/org/freedesktop/login1') + pm = dbus.Interface(proxy, + 'org.freedesktop.login1.Manager') + + if self._logout_mode == self.MODE_SHUTDOWN: + pm.PowerOff(False) + elif self._logout_mode == self.MODE_REBOOT: + pm.Reboot(True) + except: + logging.exception('Can not stop sugar') + self.session.cancel_shutdown() + return + else: + CONSOLEKIT_DBUS_PATH = '/org/freedesktop/ConsoleKit/Manager' + try: + proxy = bus.get_object('org.freedesktop.ConsoleKit', + CONSOLEKIT_DBUS_PATH) + pm = dbus.Interface(proxy, + 'org.freedesktop.ConsoleKit.Manager') + + if self._logout_mode == self.MODE_SHUTDOWN: + pm.Stop() + elif self._logout_mode == self.MODE_REBOOT: + pm.Restart() + except: + logging.exception('Can not stop sugar') + self.session.cancel_shutdown() + return + + session.SessionManager.shutdown_completed(self) + gtk.main_quit() + + def _close_emulator(self): + gtk.main_quit() + + if 'SUGAR_EMULATOR_PID' in os.environ: + pid = int(os.environ['SUGAR_EMULATOR_PID']) + os.kill(pid, signal.SIGTERM) + + # Need to call this ASAP so the atexit handlers get called before we + # get killed by the X (dis)connection + sys.exit() + + +def get_session_manager(): + global _session_manager + + if _session_manager == None: + _session_manager = SessionManager() + return _session_manager |