Làm thế nào để bạn thực hiện một “chmod + x” đơn giản từ bên trong python?

Tôi muốn tạo một tệp từ bên trong một tập lệnh python có thể thực thi được.

import os
import stat
os.chmod('somefile', stat.S_IEXEC)

nó dường os.chmodnhư không 'thêm' quyền như cách unix chmodlàm. Với dòng cuối cùng được chú thích, tệp có mã tệp -rw-r--r--, với nó không được chú thích, chế độ tệp là ---x------. Làm cách nào để tôi vừa thêm u+xcờ vừa giữ nguyên các chế độ còn lại?

Sử dụng os.stat()để nhận các quyền hiện tại, sử dụng |hoặc các bit với nhau và sử dụng os.chmod()để đặt các quyền đã cập nhật.

Thí dụ:

import os
import stat

st = os.stat('somefile')
os.chmod('somefile', st.st_mode | stat.S_IEXEC)

Đối với các công cụ tạo tệp thực thi (ví dụ: tập lệnh), đoạn mã sau có thể hữu ích:

def make_executable(path):
    mode = os.stat(path).st_mode
    mode |= (mode & 0o444) >> 2    # copy R bits to X
    os.chmod(path, mode)

Điều này làm cho nó (ít nhiều) tôn trọng umaskhiệu lực khi tệp được tạo: Executable chỉ được đặt cho những tệp có thể đọc.

Sử dụng:

path = 'foo.sh'
with open(path, 'w') as f:           # umask in effect when file is created
    f.write('#!/bin/sh\n')
    f.write('echo "hello world"\n')

make_executable(path)

Nếu bạn biết các quyền bạn muốn thì ví dụ sau có thể là cách để giữ cho nó đơn giản.

Python 2:

os.chmod("/somedir/somefile", 0775)

Python 3:

os.chmod("/somedir/somefile", 0o775)

Tương thích với một trong hai (chuyển đổi bát phân):

os.chmod("/somedir/somefile", 509)

tham chiếu các ví dụ về quyền

Bạn cũng có thể làm điều này

>>> import os
>>> st = os.stat("hello.txt")

Danh sách hiện tại của tệp

$ ls -l hello.txt
-rw-r--r--  1 morrison  staff  17 Jan 13  2014 hello.txt

Bây giờ làm điều này.

>>> os.chmod("hello.txt", st.st_mode | 0o111)

và bạn sẽ thấy điều này trong thiết bị đầu cuối.

ls -l hello.txt    
-rwxr-xr-x  1 morrison  staff  17 Jan 13  2014 hello.txt

Bạn có thể bitwise hoặc với 0o111 để làm cho tất cả có thể thực thi được, 0o222 để làm cho tất cả có thể ghi và 0o444 để làm cho tất cả có thể đọc được.

Tôn trọng umasknhưchmod +x

man chmodnói rằng nếu augokhông được đưa ra như trong:

chmod +x mypath

sau đó ađược sử dụng nhưng với umask:

Tổ hợp các chữ cái ugoa kiểm soát quyền truy cập của người dùng vào tệp sẽ bị thay đổi: người dùng sở hữu nó (u), những người dùng khác trong nhóm của tệp (g), những người dùng khác không thuộc nhóm của tệp (o) hoặc tất cả người dùng (a). Nếu không có cái nào trong số này được đưa ra, hiệu ứng giống như (a) được đưa ra, nhưng các bit được đặt trong umask không bị ảnh hưởng.

Đây là phiên bản mô phỏng chính xác hành vi đó:

#!/usr/bin/env python3

import os
import stat

def get_umask():
    umask = os.umask(0)
    os.umask(umask)
    return umask

def chmod_plus_x(path):
    os.chmod(
        path,
        os.stat(path).st_mode |
        (
            (
                stat.S_IXUSR |
                stat.S_IXGRP |
                stat.S_IXOTH
            )
            & ~get_umask()
        )
    )

chmod_plus_x('.gitignore')

Xem thêm: Làm cách nào để nhận quyền truy cập tệp mặc định trong Python?

Đã thử nghiệm trong Ubuntu 16.04, Python 3.5.2.

Trong python3:

import os
os.chmod("somefile", 0o664)

Hãy nhớ thêm 0otiền tố vì quyền được đặt dưới dạng số nguyên bát phân và Python tự động xử lý bất kỳ số nguyên nào có số 0 đứng đầu là số bát phân. Nếu không, bạn đang os.chmod("somefile", 1230)thực sự vượt qua , là bát phân của 664.

Nếu bạn đang sử dụng Python 3.4 trở lên, bạn có thể sử dụng thuận tiện các thư viện chuẩn của pathlib .

Lớp Path của nó có các phương thức chmod và stat được tích hợp sẵn .

from pathlib import Path


f = Path("/path/to/file.txt")
f.chmod(f.stat().st_mode | stat.S_IEXEC)