Question

在python 2.7中，我将数据保存到路径，例如A/B/C/data.txt：

import os

file_path = 'A/B/C/data.txt'

# create directory A/B/C
dir_name = os.path.dirname(file_path)
if not os.path.exists(dir_name):
    os.makedirs(dirp_name)

# save data to file
with open(file_path, 'w') as f:
    json.dump(data, f)

# change file permission mode to be 0x666
os.chmod(file_path, 0666)

文件data.txt的权限模式已更改。但是，此代码不会沿路径更改目录A/B/C的权限模式。我也想设置目录的权限模式。

os.chmod('A', 0666)
os.chmod('A/B', 0666)
os.chmod('A/B/C', 0666)

这样做有一种优雅的方法吗？

谢谢！

Answer 1

os.mkdir(path[, mode])允许在创建目录时设置权限模式。默认模式为0777（八进制）。如果该目录已存在，则引发OSError。

解决方案1：

# RW only permission mode is 0666 in python 2 and 0o666 in python 3
RW_only = 0666

# create directory A/B/C
dir_name = os.path.dirname(file_path)

if not os.path.exists(dir_name):
    os.makedirs(dirp_name, RW_only)

# save data to file
with open(file_path, 'w') as f:
    json.dump(data, f)

# change file permission mode to be 0x666
os.chmod(file_path, RW_only)

但是，从python 3.7开始，mode参数不再影响新创建的中间级目录的文件权限位，因此解决方案1仅适用于旧版本。

这就是我最终的结果：使用流程umask

解决方案2 [更好]：

# mkdirs has mode 0777 by default, we get 0666 by masking out 0111
old_umask = os.umask(0111)

# create directory A/B/C
dir_name = os.path.dirname(file_path)

if not os.path.exists(dir_name):
    os.makedirs(dirp_name)

# save data to file (file mode is 0666 by default)
with open(file_path, 'w') as f:
    json.dump(data, f)

# restore old_umask
os.umask(old_umask)

更改新目录和文件的权限模式

1 个答案: