Feststellen, ob ein Verzeichnis beschreibbar ist

Wie kann in Python am besten festgestellt werden, ob ein Verzeichnis für den Benutzer, der das Skript ausführt, beschreibbar ist? Da dies wahrscheinlich die Verwendung des OS-Moduls beinhaltet, sollte ich erwähnen, dass ich es unter einer * nix-Umgebung ausführe.

Obwohl Christophe eine pythonischere Lösung vorgeschlagen hat, verfügt das OS- Modul über die Funktion os.access , um den Zugriff zu überprüfen:

os.access('/path/to/folder', os.W_OK) # W_OK ist zum Schreiben, R_OK zum Lesen usw.

Es mag seltsam erscheinen, dies vorzuschlagen, aber eine verbreitete Python-Sprache ist

Es ist einfacher, um Vergebung zu bitten als um Erlaubnis

Nach dieser Redewendung könnte man sagen:

Versuchen Sie, in das betreffende Verzeichnis zu schreiben, und fangen Sie den Fehler ab, wenn Sie nicht die Berechtigung dazu haben.

Meine Lösung mit dem 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

Update: Nach dem erneuten Testen des Codes unter Windows stelle ich fest, dass bei der Verwendung von tempfile tatsächlich ein Problem vorliegt. Siehe Problem 22107: Das tempfile-Modul interpretiert den Fehler "Zugriff verweigert" unter Windows falsch . Bei einem nicht beschreibbaren Verzeichnis bleibt der Code einige Sekunden lang hängen und löst schließlich ein aus IOError: [Errno 17] No usable temporary file name found. Vielleicht hat user2171842 dies beobachtet? Leider ist das Problem vorerst nicht behoben. Um dies zu beheben, muss auch der Fehler abgefangen werden:

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

Die Verzögerung ist dann natürlich in diesen Fällen noch vorhanden.

Stolperte über diesen Thread und suchte nach Beispielen für jemanden. Erstes Ergebnis bei Google, herzlichen Glückwunsch!

In diesem Thread wird über die pythonische Vorgehensweise gesprochen, aber keine einfachen Codebeispiele? Auf geht's für alle anderen, die stolpern:

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")

Dies versucht, ein Dateihandle zum Schreiben zu öffnen, und wird mit einem Fehler beendet, wenn die angegebene Datei nicht beschrieben werden kann: Dies ist viel einfacher zu lesen und eine viel bessere Methode, als Vorprüfungen des Dateipfads oder des Verzeichnisses durchzuführen , da es Rennbedingungen vermeidet; Fälle, in denen die Datei zwischen dem Ausführen der Vorprüfung und dem tatsächlichen Versuch, in die Datei zu schreiben, nicht mehr beschreibbar ist.

Wenn Sie sich nur für die Dauerwellen interessieren, os.access(path, os.W_OK)sollten Sie das tun, wonach Sie fragen. Wenn Sie stattdessen wissen möchten, ob Sie in das Verzeichnis schreiben können , open()eine Testdatei zum Schreiben (sie sollte vorher nicht vorhanden sein), eine abfangen und untersuchen IOErrorund anschließend die Testdatei bereinigen.

Um TOCTOU- Angriffe zu vermeiden (nur ein Problem, wenn Ihr Skript mit erhöhten Berechtigungen ausgeführt wird - suid oder cgi oder so), sollten Sie diesen vorzeitigen Tests nicht wirklich vertrauen, sondern Privilegien löschen, das tun open()und erwarten die IOError.

Überprüfen Sie die Modusbits:

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)
     )

Folgendes habe ich basierend auf der Antwort von ChristopheD erstellt:

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")

Weitere Informationen zum Zugang finden Sie hier

Ich bin auf dasselbe Bedürfnis gestoßen, als ich über argparse ein Argument hinzugefügt habe. Das eingebaute type=FileType('w')würde für mich nicht funktionieren, da ich nach einem Verzeichnis suchte. Am Ende schrieb ich meine eigene Methode, um mein Problem zu lösen. Hier ist das Ergebnis mit Argparse-Snippet.

#! /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()

Das ergibt folgendes:

$ 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 ~

Ich ging zurück und fügte am Ende print opts.dir hinzu , und alles scheint wie gewünscht zu funktionieren.

Wenn Sie die Berechtigung eines anderen Benutzers überprüfen müssen (ja, mir ist klar, dass dies der Frage widerspricht, aber für jemanden nützlich sein kann), können Sie dies über das pwdModul und die Modusbits des Verzeichnisses tun .

Haftungsausschluss - funktioniert nicht unter Windows, da das POSIX-Berechtigungsmodell nicht verwendet wird (und das pwdModul dort nicht verfügbar ist), z. B. - Lösung nur für * nix-Systeme.

Beachten Sie, dass in einem Verzeichnis alle 3 Bits gesetzt sein müssen - Lesen, Schreiben und eXecute.
Ok, R ist kein absolutes Muss, aber ohne es können Sie die Einträge im Verzeichnis nicht auflisten (also müssen Sie ihre Namen kennen). Die Ausführung ist andererseits unbedingt erforderlich - ohne dass der Benutzer die Inodes der Datei nicht lesen kann; Selbst mit W können ohne X-Dateien keine Dateien erstellt oder geändert werden. Detaillierte Erklärung unter diesem Link.

Schließlich sind die Modi im statModul verfügbar , ihre Beschreibungen sind in inode (7) man .

Beispielcode zur Überprüfung:

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