在Linux下使用Python/GTK检测用户注销/关机 - 无法接收SIGTERM/HUP信号

Question

好的，这可能是一个比较难的问题。我有一个使用pyGTK编写的应用程序，由于无法捕获/控制的X Window错误，导致随机崩溃。

因此，我创建了一个包装器，在检测到崩溃时立即重新启动该应用程序。现在出现了问题，当用户注销或关闭系统时，该应用程序会以状态1退出。但在某些X错误上也会这样做。

因此，我尝试了各种方法来捕获关机/注销事件，但都没有成功。以下是我尝试过的方法：

import pygtk
import gtk
import sys


class Test(gtk.Window):
    def delete_event(self, widget, event, data=None):
        open("delete_event", "wb")

    def destroy_event(self, widget, data=None):
        open("destroy_event", "wb")

    def destroy_event2(self, widget, event, data=None):
        open("destroy_event2", "wb")

    def __init__(self):
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
        self.show()
        self.connect("delete_event", self.delete_event)
        self.connect("destroy", self.destroy_event)
        self.connect("destroy-event", self.destroy_event2)      

def foo():
    open("add_event", "wb")

def ex():
    open("sys_event", "wb")


from signal import *
def clean(sig):
    f = open("sig_event", "wb")
    f.write(str(sig))
    f.close()
    exit(0)

for sig in (SIGABRT, SIGILL, SIGINT, SIGSEGV, SIGTERM):
    signal(sig, lambda *args: clean(sig))


def at():
    open("at_event", "wb")

import atexit
atexit.register(at)

f = Test()
sys.exitfunc = ex
gtk.quit_add(gtk.main_level(), foo)

gtk.main()
open("exit_event", "wb")

这些方法都不行，有没有低级别的方法来检测系统关闭？谷歌上没有相关信息。

我想肯定有一种方法，对吗？ :/

编辑: 好的，更多内容。

我创建了这个shell脚本：

#!/bin/bash


trap test_term TERM
trap test_hup HUP


test_term(){
    echo "teeeeeeeeeerm" >~/Desktop/term.info
    exit 0
}

test_hup(){
    echo "huuuuuuuuuuup" >~/Desktop/hup.info
    exit 1
}

while [ true ]
do
    echo "idle..."
    sleep 2
done

同时创建了一个 .desktop 文件来运行它：

[Desktop Entry]
Name=Kittens
GenericName=Kittens
Comment=Kitten Script
Exec=kittens
StartupNotify=true
Terminal=false
Encoding=UTF-8
Type=Application
Categories=Network;GTK;
Name[de_DE]=Kittens 

通常情况下，当使用 & 启动时，该命令应在注销时创建术语文件并在启动时创建hup文件。但是在我的系统上却没有这样的情况。 GDM根本不关心脚本，在我重新登录时，它仍在运行。

我还尝试使用 shopt -s huponexit，但没有成功。

编辑2：
此外，以下是关于实际代码的更多信息，整个代码如下：

Wrapper Script, that catches errors and restarts the programm
    -> Main Programm with GTK Mainloop
        -> Background Updater Thread

流程如下所示:

Start Wrapper
-> enter restart loop
    while restarts < max:
        -> start program
            -> check return code
                -> write error to file or exit the wrapper on 0

现在在关闭时，start program返回1。这意味着它可能已经挂起或父进程终止了，主要问题是弄清楚刚刚发生的是哪一个。
X错误也会导致返回1。在shell脚本中捕获无效。
如果您想查看实际代码，请在GitHub上查看：
http://github.com/BonsaiDen/Atarashii

- Ivo Wetzel

Answer 1

好的，我终于找到了解决方案 :)

在这种情况下，你不能仅依靠信号。你必须连接到桌面会话，以便收到注销通知。

import gnome.ui

gnome.program_init('Program', self.version) # This is going to trigger a warning that program name has been set twice, you can ignore this, it seems to be a problem with a recent version of glib, the warning is all over the place out there
client = gnome.ui.master_client() # connect us to gnome session manager, we need to init the program before this
client.connect('save-yourself', self.on_logout) # This gets called when the user confirms the logout/shutdown
client.connect('shutdown-cancelled', self.on_logout_cancel) # This gets called when the logout/shutdown is canceled
client.connect('die', self.on_logout) # Don't know when this gets called it never got in my tests

def on_logout(self, *args):
    # save settings an create a file that tells the wrapper that we have exited correctly!
    # we'll still return with status code 1, but that's just gtk crashing somehow

def on_logout_cancel(self, *args):
    # simply delete the logout file if it exists

这里有一个重要的提示：不要试图在on_logout中退出你的程序，如果这样做，GNOME将无法识别你的程序已经退出，并会弹出对话框提示仍有一些程序正在运行。

- Ivo Wetzel

Answer 2

你忘记关闭gtk的事件循环。

当你关闭窗口时，这段代码将以代码0退出：

import gtk

class Test(gtk.Window):
    def destroy_event(self, widget, data=None):
        gtk.main_quit()

    def __init__(self):
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
        self.connect("destroy", self.destroy_event)
        self.show()

f = Test()
gtk.main()

---

编辑：这里是捕获SIGTERM信号的代码：

import signal

def handler(signum, frame):
    print 'Signal handler called with signal', signum
    print 'Finalizing main loop'
    gtk.main_quit()

signal.signal(signal.SIGTERM, handler)

其余的代码与上面完全相同，没有任何更改。当我向Python进程发送SIGTERM时，它在这里工作：gtk主循环结束并以退出码0退出程序。

- nosklo

---

这并不能解决问题，当用户退出或关闭系统时，destroy_event甚至没有被调用，上述文件中没有一个被创建。另一个问题源于包装脚本，它首先被“杀死”，因此应用程序将以1退出，因为其父进程已经消失。 - Ivo Wetzel

进一步的研究表明，实际上该进程应该接收到一个SIGTERM信号。但是它没有收到，似乎只是被强制终止了... - Ivo Wetzel

根据GDM的源代码，它首先发送SIGTERM信号，但似乎Python并不关心:/ - Ivo Wetzel

仍然无法工作，似乎GnomeDesktopManager出了问题，我连接了每个信号，但没有一个被接收到，我猜它真的只是做了一个-kill。我还创建了一个捕获所有信号的shell脚本，但仍然没有运气。 - Ivo Wetzel

@Ivo Wetzel：我无法复制此问题。这里运行非常顺畅。我已经让它在 SIGTERM 上创建一个文件，并且当我注销时可以正常地创建该文件。我在 Ubuntu 9.10 上使用 GDM。你的代码中一定存在其他问题（你是否在使用线程？） - nosklo

我也在使用Ubuntu，我已更新了问题，并添加了一个不起作用的Shell脚本。 - Ivo Wetzel