将文件排序到文件夹中的 Python 代码

Question

Python 2.5.1

http://www.cgsecurity.org/wiki/After_Using_PhotoRec

我已经只需运行 PhotoRec，作为将文件类型排序到其自己的文件夹中的方法给出的代码就会返回此错误。 关于如何改变有什么建议吗？ 谢谢：

[编辑2：两点：

1. 这个问题被否决了，因为它是代码的“使用”，而不是编程问题。 它符合编码问题吗？ 我认为是的。

2. 我已经返回并编辑了代码所在的页面，以澄清对参数的需求，以便其他人受益。]

   gyaresu$ python recovery.py 回溯（最近一次调用最后一次）： 文件“recovery.py”，第 8 行，位于 来源 = sys.argv[1] IndexError: list index out of range

脚本：

#!/usr/bin/env python
import os
import os.path
import shutil
import string
import sys

source = sys.argv[1]
destination = sys.argv[2]

while os.path.exists(source) != True:
    source = raw_input('Enter a valid source directory\n')
while os.path.exists(destination) != True:
    destination = raw_input('Enter a valid destination directory\n')

for root, dirs, files in os.walk(source, topdown=False):
    for file in files:
        extension = string.upper(os.path.splitext(file)[1][1:])
        destinationPath = os.path.join(destination,extension)

        if os.path.exists(destinationPath) != True:
            os.mkdir(destinationPath)
        if os.path.exists(os.path.join(destinationPath,file)):
            print 'WARNING: this file was not copied :' + os.path.join(root,file)
        else:
            shutil.copy2(os.path.join(root,file), destinationPath)

Answer 1

它只是意味着程序需要两个命令行参数：源和目标。 如果您希望在另一个函数中使用相同的代码，请将 sys.argv[1] 和 [2] 替换为您自己的变量。

Answer 2

或者您可以修改原始脚本并

if len(sys.argv) != 3:
    print "Require 2 arguments: %s <source> <destination>" %(sys.argv[0])
    sys.exit(1)

在导入语句之后添加以进行正确的错误处理。

Answer 3

由于脚本将询问路径（如果路径不存在），因此您可以将程序参数设置为可选。

改成

source = sys.argv[1]
destination = sys.argv[2]

​

source = sys.argv[1] if len(sys.argv > 1) else ""
destination = sys.argv[2] if len(sys.argv > 2) else ""