如何执行一个目录中的所有文件或基于args的单个文件的功能，在python

-1

我有一个小脚本我正在处理“碎化”单个文件（零文件的字节），我试图使它能够根据是否通过命令行接收到-file或-directory标志/ arg来将目录中的每个文件都碎片化。这对单个文件工作正常，但我得到一个：如何执行一个目录中的所有文件或基于args的单个文件的功能，在python

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

来源

2017-08-14 mypython

什么是'args'。 ['argparse.ArgumentParser（）.parse_args（）']的返回值（https://docs.python.org/3.6/library/argparse.html#argparse.ArgumentParser）？ –

“bar”在哪里定义？ – Kevin

你可以发布你得到的_full_追踪？ –

os.listdir返回目录中的文件名

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

来源

2017-08-14 14:56:46 Teo

感谢这工作完美，一旦实施，立即解决问题 – mypython

我猜你不能一丝一毫的目录，但只有个别的文件。

shredfile(os.path.join(srcpath, d), zeros)

来源

2017-08-14 14:55:22 dnozay

谢谢你的帮助，这个答案是正确的以及上面的问题，现在问题已经修复。非常感谢你的帮助@dnozay – mypython

这将是这样做有道。您只需添加进度条和参数解析等部件即可。您还需要添加一些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

来源

2017-08-14 16:13:03 Dalen

如何执行一个目录中的所有文件或基于args的单个文件的功能，在python

回答

相关问题