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模塊提供了常量值可以直接對應使用。
 

如下的代碼是將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)