确定目录是否可写

Question

Python中确定目录对于执行脚本的用户是否可写的最佳方法是什么？由于这可能涉及使用os模块,我应该提到我在*nix环境下运行它.

Answer 1

虽然Christophe建议的是更Pythonic解决方案,但os模块确实具有os.access函数来检查访问:

os.access('/path/to/folder', os.W_OK) #W_OK用于写入,R_OK用于阅读等.

Answer 2

建议这可能看起来很奇怪,但一个常见的Python习语是

要求宽恕比获得许可更容易

在这个成语之后,有人会说:

尝试写入相关目录,如果您没有权限,请捕获错误.

Answer 3

我使用该tempfile模块的解决方案:

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

Answer 4

偶然发现这个线程正在寻找某人的例子.对谷歌的第一个结果,恭喜!

人们在这个帖子中谈论Pythonic的做法,但没有简单的代码示例？在这里,对于其他任何偶然发现的人:

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

这会尝试打开文件句柄进行写入,如果指定的文件无法写入,则会以错误退出:这更容易阅读,并且是一种更好的方法,而不是在文件路径或目录上进行预先检查,因为它避免了竞争条件; 在运行预检和实际尝试写入文件之间文件变为不可写的情况.

Answer 5

如果您只关心文件烫发,os.access(path, os.W_OK)应该按照您的要求进行操作.如果您想要知道是否可以写入目录,则需要open()一个用于写入的测试文件(它不应该预先存在),捕获并检查任何内容IOError,然后清理测试文件.

更一般地说,为了避免TOCTOU攻击(如果你的脚本以提升的权限运行,只有一个问题 - suid或cgi左右),你不应该真正相信这些提前测试,而是放弃私有,做open(),并期待这个IOError.

Answer 6

检查模式位:

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

Answer 7

这是我根据 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"