Menentukan Apakah Direktori Dapat Ditulisi

Apa cara terbaik dengan Python untuk menentukan apakah direktori dapat ditulisi untuk pengguna yang menjalankan skrip? Karena ini kemungkinan akan melibatkan penggunaan modul os, saya harus menyebutkan bahwa saya menjalankannya di bawah lingkungan * nix.

Meskipun yang disarankan Christophe adalah solusi yang lebih Pythonic, modul os memang memiliki fungsi os.access untuk memeriksa akses:

os.access('/path/to/folder', os.W_OK) # W_OK untuk menulis, R_OK untuk membaca, dll.

Mungkin tampak aneh untuk menyarankan ini, tetapi idiom Python yang umum adalah

Lebih mudah meminta maaf daripada izin

Mengikuti idiom itu, orang mungkin berkata:

Coba tulis ke direktori yang dimaksud, dan tangkap kesalahannya jika Anda tidak memiliki izin untuk melakukannya.

Solusi saya menggunakan tempfilemodul:

import tempfile
import errno

def isWritable(path):
    try:
        testfile = tempfile.TemporaryFile(dir = path)
        testfile.close()
    except OSError as e:
        if e.errno == errno.EACCES:  # 13
            return False
        e.filename = path
        raise
    return True

Pembaruan: Setelah menguji kode lagi pada Windows saya melihat bahwa memang ada masalah saat menggunakan tempfile di sana, lihat issue22107: kesalahan tafsir modul tempfile akses ditolak pada Windows . Dalam kasus direktori yang tidak dapat ditulis, kode macet selama beberapa detik dan akhirnya melempar IOError: [Errno 17] No usable temporary file name found. Mungkin ini yang diamati oleh user2171842? Sayangnya masalah tersebut belum teratasi untuk saat ini, jadi untuk menanganinya, kesalahan juga perlu diperhatikan:

    except (OSError, IOError) as e:
        if e.errno == errno.EACCES or e.errno == errno.EEXIST:  # 13, 17

Penundaan tentu saja masih ada dalam kasus-kasus ini.

Tersandung di utas ini mencari contoh untuk seseorang. Hasil pertama di Google, selamat!

Orang-orang berbicara tentang cara Pythonic melakukannya di utas ini, tetapi tidak ada contoh kode sederhana? Ini dia, untuk siapa pun yang tersandung:

import sys

filepath = 'C:\\path\\to\\your\\file.txt'

try:
    filehandle = open( filepath, 'w' )
except IOError:
    sys.exit( 'Unable to write to file ' + filepath )

filehandle.write("I am writing this text to the file\n")

Ini mencoba untuk membuka filehandle untuk menulis, dan keluar dengan kesalahan jika file yang ditentukan tidak dapat ditulis ke: Ini jauh lebih mudah dibaca, dan merupakan cara yang jauh lebih baik untuk melakukannya daripada melakukan prechecks pada jalur file atau direktori , karena menghindari kondisi balapan; kasus di mana file menjadi tidak dapat ditulis antara saat Anda menjalankan pemeriksaan awal, dan saat Anda benar-benar mencoba untuk menulis ke file.

Jika Anda hanya peduli tentang file perm, os.access(path, os.W_OK)harus melakukan apa yang Anda minta. Jika Anda justru ingin mengetahui apakah Anda dapat menulis ke direktori, open()file pengujian untuk ditulis (seharusnya tidak ada sebelumnya), tangkap dan periksa apa saja IOError, dan bersihkan file pengujian setelahnya.

Secara lebih umum, untuk menghindari serangan TOCTOU (hanya masalah jika skrip Anda berjalan dengan hak istimewa yang lebih tinggi - suid atau cgi atau lebih), Anda tidak boleh benar-benar mempercayai pengujian yang akan dilakukan sebelumnya ini, tetapi jatuhkan privasi, lakukan open(), dan harapkan itu IOError.

Periksa bit mode:

def isWritable(name):
  uid = os.geteuid()
  gid = os.getegid()
  s = os.stat(dirname)
  mode = s[stat.ST_MODE]
  return (
     ((s[stat.ST_UID] == uid) and (mode & stat.S_IWUSR)) or
     ((s[stat.ST_GID] == gid) and (mode & stat.S_IWGRP)) or
     (mode & stat.S_IWOTH)
     )

Ini adalah sesuatu yang saya buat berdasarkan jawaban ChristopheD:

import os

def isWritable(directory):
    try:
        tmp_prefix = "write_tester";
        count = 0
        filename = os.path.join(directory, tmp_prefix)
        while(os.path.exists(filename)):
            filename = "{}.{}".format(os.path.join(directory, tmp_prefix),count)
            count = count + 1
        f = open(filename,"w")
        f.close()
        os.remove(filename)
        return True
    except Exception as e:
        #print "{}".format(e)
        return False

directory = "c:\\"
if (isWritable(directory)):
    print "directory is writable"
else:
    print "directory is not writable"

 if os.access(path_to_folder, os.W_OK) is not True:
            print("Folder not writable")
 else :
            print("Folder writable")

info lebih lanjut tentang akses dapat ditemukan di sini

Saya mengalami kebutuhan yang sama ini sambil menambahkan argumen melalui argparse. Bawaan type=FileType('w')tidak akan berfungsi untuk saya karena saya sedang mencari direktori. Saya akhirnya menulis metode saya sendiri untuk memecahkan masalah saya. Inilah hasil dengan potongan argparse.

#! /usr/bin/env python
import os
import argparse

def writable_dir(dir):
    if os.access(dir, os.W_OK) and os.path.isdir(dir):
        return os.path.abspath(dir)
    else:
        raise argparse.ArgumentTypeError(dir + " is not writable or does not exist.")

parser = argparse.ArgumentParser()
parser.add_argument("-d","--dir", type=writable_dir(), default='/tmp/',
    help="Directory to use. Default: /tmp")
opts = parser.parse_args()

Hasilnya sebagai berikut:

$ python dir-test.py -h
usage: dir-test.py [-h] [-d DIR]

optional arguments:
  -h, --help         show this help message and exit
  -d DIR, --dir DIR  Directory to use. Default: /tmp

$ python dir-test.py -d /not/real
usage: dir-test.py [-h] [-d DIR]
dir-test.py: error: argument -d/--dir: /not/real is not writable or does not exist.

$ python dir-test.py -d ~

Saya kembali dan menambahkan print opts.dir sampai akhir, dan semuanya tampak berfungsi seperti yang diinginkan.

Jika Anda perlu memeriksa izin pengguna lain (ya, saya menyadari ini bertentangan dengan pertanyaan, tetapi mungkin berguna untuk seseorang), Anda dapat melakukannya melalui pwdmodul, dan bit mode direktori.

Penafian - tidak berfungsi di Windows, karena tidak menggunakan model izin POSIX (dan pwdmodul tidak tersedia di sana), misalnya - solusi hanya untuk sistem * nix.

Perhatikan bahwa direktori harus memiliki semua 3 bit set - Baca, Tulis dan eXecute.
Ok, R bukanlah suatu keharusan mutlak, tetapi tanpa itu Anda tidak dapat mencantumkan entri dalam direktori (jadi Anda harus mengetahui namanya). Eksekusi di sisi lain mutlak diperlukan - tanpa pengguna tidak dapat membaca inode file; jadi meskipun memiliki W, tanpa X file tidak dapat dibuat atau dimodifikasi. Penjelasan lebih detail ada di link ini.

Akhirnya, mode tersedia di statmodul, deskripsinya ada di inode (7) man .

Kode contoh cara memeriksa:

import pwd
import stat
import os

def check_user_dir(user, directory):
    dir_stat = os.stat(directory)

    user_id, group_id = pwd.getpwnam(user).pw_uid, pwd.getpwnam(user).pw_gid
    directory_mode = dir_stat[stat.ST_MODE]

    # use directory_mode as mask 
    if user_id == dir_stat[stat.ST_UID] and stat.S_IRWXU & directory_mode == stat.S_IRWXU:     # owner and has RWX
        return True
    elif group_id == dir_stat[stat.ST_GID] and stat.S_IRWXG & directory_mode == stat.S_IRWXG:  # in group & it has RWX
        return True
    elif stat.S_IRWXO & directory_mode == stat.S_IRWXO:                                        # everyone has RWX
        return True

    # no permissions
    return False