Question

我知道这种模式可以读取Python中的umask：

current_umask = os.umask(0)  # line1
os.umask(current_umask)      # line2
return current_umask         # line3

但这不是线程安全的。

在line1和line2之间执行的线程将具有不同的umask。

是否有线程安全的方法来读取Python中的umask？

相关：https://bugs.python.org/issue35275

Answer 1

如果您的系统在Umask中有/proc/[pid]/status字段，则可以从中读取：

import os

def getumask():
    pid = os.getpid()
    with open(f'/proc/{pid}/status') as f:
        for l in f:
            if l.startswith('Umask'):
                return int(l.split()[1], base=8)
        return None

在CentOS 7.5，Debian 9.6下进行了测试。

或者，您可以添加线程锁：）

Answer 2

umask由子进程继承。您可以创建一个管道，派生一个子进程，在其中获得umask并将结果写入管道，以便父级可以读取它。

非常昂贵，但没有任何特殊要求，例如pgadmin4虚拟文件系统。仅具有低级OS调用（均异步安全）且下面没有错误检查的示例：

/proc

Answer 3

可以通过创建一个临时文件并检查其权限来确定umask。这应该适用于所有* nix系统：

def get_umask():
    import os, os.path, random, tempfile
    while True:
        # Generate a random name
        name = 'test'
        for _ in range(8):
            name += chr(random.randint(ord('a'), ord('z')))
        path = os.path.join(tempfile.gettempdir(), name)
        # Attempt to create a file with full permissions
        try:
            fd = os.open(path, os.O_RDONLY|os.O_CREAT|os.O_EXCL, 0o777)
        except FileExistsError:
            # File exists, try again
            continue
        try:
            # Deduce umask from the file's permission bits
            return 0o777 & ~os.stat(fd).st_mode
        finally:
            os.close(fd)
            os.unlink(path)

Answer 4

我知道的唯一真正，明确的线程安全方法是调用新进程。

import subprocess
umask_cmd = ('python', '-c', 'import os; print(os.umask(0777))')
umask = int(subprocess.check_output(umask_cmd))

请注意，如果您有bash或其他shell，也可以调用它。由于它可能位于怪异的系统上，因此我选择在umask_cmd中使用python子进程，因为您必须具有python。如果您使用的是非怪异的* nix系统，则可以使用sh或bash代替。

读取umask（线程安全）

4 个答案: