python - 检查当前进程/线程是否是主进程/线程的正确方法是什么？

Question

生成多个线程时，我可以执行以下操作来检查该线程是否是主线程:

isinstance(threading.currentThread(), threading._MainThread)

但是这个方法依赖于threading包的一个“protected”成员，并且在多处理库中没有类似的类型可以导入，所以检查当前进程是否是主进程的正确方法是什么(并做和线程一样？)

我现在是这样做的:

multiprocessing.current_process().name == 'MainProcess'

但是这些名字可靠吗？

Answer 1

这个看起来比较靠谱:

assert threading.current_thread() == threading.main_thread()

因为它不依赖于静态文本。

对于进程，执行 process.name == 'MainProcess' 是不可靠的。你应该这样做:

import re

def is_main_process(process_name):
    return not bool(re.match(r'Process-\d+', process_name))

原因是子进程保证有一个这样格式的名字:Process-N。