因此,我正在编写一个cli工具,该工具获取一个文件并计算该文件的校验和(CRC32)。我希望我的工具有递归作为一个选项。所以我想要一个目录中所有文件的列表,这样我就可以将它作为参数传递给我的checksum方法。
注意:'Test Dir/‘是与脚本位于同一文件夹中的目录,通过终端输入,而不是设置变量。
到目前为止,我有:
rootDir = ['Test Dir/']
files = []
for i in rootDir:
if not os.path.isdir(i):
files.append(i)
else:
for dir_, _, files in os.walk(i):
for fileName in files:
relDir = os.path.relpath(dir_, i)
relFile = os.path.join(relDir, fileName)
files.append(relFile)
这将返回:
['./root.py', './root.txt', 'Lv1/lv1.py', 'Lv1/lv1.pyc', 'Lv1/lv1.txt', 'Lv1/Lv2/lv2.pyc', 'Lv1/Lv2/lv2.py', 'Lv1/Lv2/lv2.txt', 'Lv1/Lv2/Lv3/lv3.txt', 'Lv1/Lv2/Lv3/lv3.py', 'Lv1/Lv2/Lv3/lv3.pyc']
但是如果我输入rootDir作为'Test Dir/*',我会得到:
['./lv1.py', './lv1.pyc', './lv1.txt', 'Lv2/lv2.pyc', 'Lv2/lv2.py', 'Lv2/lv2.txt', 'Lv2/Lv3/lv3.txt', 'Lv2/Lv3/lv3.py', 'Lv2/Lv3/lv3.pyc', 'Test Dir/root.py', 'Test Dir/root.txt']
文件相同,但列表略有不同。
我想要的是:['Test Dir/root.py', 'Test Dir/root.txt', 'Test Dir/Lv1/lv1.py', 'Test Dir/Lv1/lv1.pyc', 'Test Dir/Lv1/lv1.txt', 'Test Dir/Lv1/Lv2/lv2.pyc', 'Test Dir/Lv1/Lv2/lv2.py', 'Test Dir/Lv1/Lv2/lv2.txt', 'Test Dir/Lv1/Lv2/Lv3/lv3.txt', 'Test Dir/Lv1/Lv2/Lv3/lv3.py', 'Test Dir/Lv1/Lv2/Lv3/lv3.pyc']
现在我知道我可以使用字符串操作来做到这一点,但这看起来并不是很干净…真希望有人能帮我解决这个问题。
发布于 2018-06-29 08:40:45
os.path.relpath
的工作方式是,它将只包含给定基本路径下的路径名,因此,如果您希望最终列表都包含基本路径名,那么实际上一开始就不需要os.path.relpath
。
此外,用于for
循环的files
变量会隐藏您在开始时初始化并在for
循环中使用的files
列表。您的代码不会真正完成,因为您在将列表用作迭代器时一直在向列表追加内容。你应该给它起个别的名字。
您的代码应更正为类似以下内容:
rootDir = ['Test Dir/']
paths = []
for i in rootDir:
if not os.path.isdir(i):
paths.append(i)
else:
for dir_, _, files in os.walk(i):
for fileName in files:
paths.append(os.path.join(i, fileName))
https://stackoverflow.com/questions/51092023
复制相似问题