Question

在我的代码中，我的功能如下：

def myfunc():
    # Don't do anything if there's an instance already
    if get_var('running') == 'true':
         return

    set_var('running', 'true')
    # In case things go wrong
    atexit.register(set_var, 'running', 'false')

    do_something()
    do_something_else()

    set_var('running', 'false')
    # Unregister handler because nothing bad happened
    atexit.unregister(set_var)

set_var设置数据库中包含的变量。

所有set_var的目的是防止多个实例同时运行。

当程序被 Ctrl-C 中断时，

atexit处理程序正常工作，但是当它被系统或类似的东西杀死时，它不会正常工作。

我知道signal，但它不允许取消处理程序。

我该怎么做？或者如何改变结构以实现相同的功能？

Answer 1

我想我已经明白了。

# Used to check if myfunc is running in current program
running_here = False

# Set 'running' variable inside database to 'false' if myfunc was running
# inside current program at the time of exiting or don't do anything otherwise
atexit.register(lambda: set_var('running', 'false') if running_here else None)
# Call atexit handler when SIGTERM is recieved by calling sys.exit
signal.signal(signal.SIGTERM, lambda x, frame: sys.exit(0))

def myfunc():
    global running_here

    # Don't do anything if there's an instance already
    if get_var('running') == 'true':
         return

    # Don't let multiple instances to run at the same time
    set_var('running', 'true')
    running_here = True

    do_something()
    do_something_else()

    # Allow other instances to run
    set_var('running', 'false')
    running_here = False

我需要做的就是制作一个不需要一遍又一遍取消的处理程序。我通过添加全局变量running_here来实现这一点。

当程序被终止时，处理程序只检查running_here是否在当前程序中运行，如果它是True，那么处理程序只是在数据库中设置变量running < / strong>到'false'，因此其他实例无法启动。如果running_here为False，则表示myfunc未运行且无需重置running变量，因此只需退出。

在退出时执行代码

1 个答案: