linux中,文件的权限分为"所有者、组、其他用户"三个角色,每个角色由3个bit位表示它的权限,3bit从左到右分别为读写执行三个权限,3bit的值范围为0~7。所以如果直接在linux执行chmod 777 xxx.sh
代表,将xxx.sh文件赋予所有者、组、其他用户这三个角色对xxx.sh文件的读写执行权限。
python的os模块负责操作系统层面的操作。修改文件权限可以通过os的chmod方法来操作。
os.chmod(path, mode)
方法的path
指向要赋权限的文件或目录;
mode
为权限值,也就是文件权限的9位bit的10进制数,取值范围0~511;
比如我们要将xxx.sh文件赋予所有者读权限,其他角色没有任何的权限,则linux中对应的文件权限用ls -l查看为:-r---------
;第一个-
是文件类型,后面的9个位就代表了整个文件的权限,对应二进制为100000000,那么我们的mode
应该取值为256;
所以上面的权限修改代码为os.chmod("xxx.sh",256)
。
这么麻烦?
还好stat
模块提供了常量值可以直接对应使用。
常量 | 描述 |
---|---|
S_IRWXU | 所有者读写执行权限,代表值:448 |
S_IRUSR | 所有者读,代表值:256 |
S_IWUSR | 所有者写,代表值:128 |
S_IXUSR | 所有者执行,代表值:64 |
S_IRWXG | 用户组读写执行权限,代表值:56 |
S_IRGRP | 用户组读,代表值:32 |
S_IWGRP | 用户组写,代表值:16 |
S_IXGRP | 用户组执行,代表值:8 |
S_IRWXO | 其他用户读写执行,代表值:7 |
S_IROTH | 其他用户读,代表值:4 |
S_IWOTH | 其他用户写,代表值:2 |
S_IXOTH | 其他用户执行,代表值:1 |
如下的代码是将xxx.sh文件赋予所有者读写执行、用户组读执行、其他用户读的操作,即-rwxr-xr--
:
import os
import stat
os.chmod("xxx.sh", stat.S_IRWXU+stat.S_IRGRP+stat.S_IXGRP+stat.S_IROTH)
什么?感觉常量记不住?还是不方便?
写个类似chmod 777
的使用方式的util吧…
import re
import os
import stat
RD, WD, XD = 4, 2, 1
BNS = [RD, WD, XD]
MDS = [
[stat.S_IRUSR, stat.S_IRGRP, stat.S_IROTH],
[stat.S_IWUSR, stat.S_IWGRP, stat.S_IWOTH],
[stat.S_IXUSR, stat.S_IXGRP, stat.S_IXOTH]
]
def chmod(path, mode):
if isinstance(mode, int):
mode = str(mode)
if not re.match("^[0-7]{1,3}$", mode):
raise Exception("mode does not conform to ^[0-7]{1,3}$ pattern")
mode = "{0:0>3}".format(mode)
mode_num = 0
for midx, m in enumerate(mode):
for bnidx, bn in enumerate(BNS):
if (int(m) & bn) > 0:
mode_num += MDS[bnidx][midx]
os.chmod(path, mode_num)
if __name__ == '__main__':
chmod("xxx.sh", "744") #744代表所有者读写执行,组和其他的权限只有读
类似的你也可以让它支持类似chmod u+rwx,g+rw,o+rw xxx.sh
的使用方式。
注意:os.chmod貌似没有办法递归修改目录下所有文件的权限?那使用下面的subprocess直接调用命令吧…
subprocess模块实现了子进程运行。可以在子进程中调用操作系统命令或者执行shell,并获取stdout、stderr。
import subprocess
if __name__ == '__main__':
cmd = ['chmod', "-R", "777", "xxx.sh"]
res = subprocess.run(cmd, universal_newlines=True, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, shell=False)
print(res.returncode, res.stdout, res.stderr)