python - Windows 目录名称无效

Question

这是我正在尝试的代码，但它不断重复同样的错误。如果我做错了什么，请告诉我。

import os
import sys

for x in sys.argv:
    for ff in os.listdir(x):
        path = os.path.join(x, ff)
        if os.path.isdir(path):
            print('\n--' + path)
        else:
            print('\t------' + path)

这个错误一直重复，一次又一次，请有人帮助我。

Traceback (most recent call last):
  File "E:/projects/Intern/file&folders/cc.py", line 5, in <module>
    for ff in os.listdir(x):
NotADirectoryError: [WinError 267] The directory name is invalid: 'E:/projects/Intern/file&folders/cc.py'

Answer 1

使用以下代码:-

import os
import sys

for x in sys.argv[1:]:
    for ff in os.listdir(x):
        path = os.path.join(x, ff)
        if os.path.isdir(path):
            print('\n--' + path)
        else:
            print('\t------' + path)

您将整个 sys.argv 列表传递给 os.listdir() 函数。第一个元素始终是脚本本身，它不是目录。因此，我们按第一个元素对参数列表进行切片。