Python3通过chmod修改目录或文件权限

简单的介绍下linux文件权限

linux中，文件的权限分为"所有者、组、其他用户"三个角色，每个角色由3个bit位表示它的权限，3bit从左到右分别为读写执行三个权限，3bit的值范围为0~7。所以如果直接在linux执行chmod 777 xxx.sh代表，将xxx.sh文件赋予所有者、组、其他用户这三个角色对xxx.sh文件的读写执行权限。

os的chmod

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直接调用chmod命令

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)