Question

我有一个小脚本我正在研究哪些＆＃34;碎片＆＃34;单个文件（文件的字节为零），我试图使它可以根据它是否通过命令行接收-file或-directory标志/ arg来粉碎目录中的每个文件。它对单个文件工作正常，但我得到了：

IsADirectoryError: [Errno 21] 'Path/To/Directory' error when trying to use it on all files in a directory。我已经尝试了几种方法来安排循环，以便它们可以工作，但到目前为止还没有。我希望有人能帮助我纠正这个问题。

我已设置argparser，因此-F file, -D directory, -s srcpath, -i iterations以及与bar有关的任何内容都只是我的进度条，其中显示了脚本在运行过程中的进度。

这是我的代码：

parser= argparse.ArgumentParser()
parser.add_argument('-F', '--file', action=store_true)
parser.add_argument('-D', '--directory', action=store_true)
parser.add_argument('-s', '--source', required=True, type=str)
parser.add_argument('-i', '--iterations', required=True, type=int)
args = parser.parse_args()

bar = Bar ("Progress: ", max=args.iterations)
srcpath = "path/to/file/"

def shredfile(source, filebytes):
    with open(source, 'r+b') as f:
        byte = f.read(1)
        while byte:
            f.write(filebytes)
            byte = f.read(1)

zeros = ("0000000").encode('UTF-8')

if args.file:
    x=0
    for x in range (args.iterations):
        shredfile(srcpath, zeros)
        bar.next()
    bar.finish()
    print("Completed")

if args.directory:
    for d in os.listdir(srcpath):
        x=0
        for x in range (args.iterations):
            shredfile(srcpath, zeros)
            bar.next()
        bar.finish()
        print("Completed")

Answer 1

os.listdir返回目录

中的文件名

for f in os.listdir(srcpath):
  full_path = os.path.join(f, srcpath)
  ...
    shredfile(full_path, zeros)
  ...
  print("Completed")

Answer 2

我猜你不能破坏目录，只能破坏单个文件。 shredfile（os.path.join（srcpath，d），零）

Answer 3

这是一种正确的方法。你只需添加你的部分，如进度条和参数解析。您还需要添加一些try-except块来控制OSErrors上发生的事情，例如拒绝权限和类似的东西。



import sys
import os

def shredfile (source, filebytes=16*"\0"):
    f = open(source, "rb")
    f.seek(0, 2)
    size = f.tell()
    f.close()
    if len(filebytes)>size:
        chunk = filebytes[:size]
    else:
        chunk = filebytes
    f = open(source, "wb")
    l = len(chunk)
    n = 0
    while n+l<size:
        f.write(chunk)
        n += l
    # Ensure that file is overwritten to the end if size/len(filebytes)*len(filebytes) is not equal to size
    chunk = filebytes[:size-f.tell()]
    if chunk: f.write(chunk)
    f.close()

def shreddir (source, filebytes=16*"\0", recurse=0):
    for x in os.listdir(source):
        path = os.path.join(source, x)
        if os.path.isfile(path):
            shredfile(path, filebytes)
            continue
        if recurse:
            shreddir(path, filebytes, 1)

def shred (source, filebytes=16*"\0", recurse=0):
    if os.path.isdir(source):
        shreddir(source, filebytes, recurse)
        return
    shredfile(source, filebytes)

if len(sys.argv)>1:
    print "Are you sure you want to shred '%s'?" % sys.argv[-1],
    c = raw_input("(Yes/No?) ").lower()
    if c=="yes":
        shred(sys.argv[-1])
# Here you can iterate shred as many times as you want.
# Also you may choose to recurse into subdirectories which would be what you want if you need whole directory shredded

如何在python中对目录中的所有文件或基于args的单个文件执行函数

3 个答案: