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)