python - 在命令行中为 python os.walk 脚本定义参数

Question

我正在编写一个更大的脚本，它遍历目录结构以对每个文件执行操作。

我无法弄清楚如何在命令行中将目标目录作为参数传递(作为 os.walk 的起点)。

谁能告诉我做错了什么？

import os
import sys

def main():
    for root, dirs, files in os.walk(sys.argv[1])
      for item in files:
        print(items)

if __name__ == '__main__':
    main(sys.argv[1])

Answer 1

是的，您对 main 的原始调用没有接受任何参数:

import os
import sys

def main(): # this has 0 arguments, instead you included it inside of the func.
    for root, dirs, files in os.walk(sys.argv[1]):
      for item in files:
        print(items)

if __name__ == '__main__':
    main(sys.argv[1]) # likely the source of your error since main() has the argument inside the func.

如果你想让 main 有参数:

def main(args): # any generic argument to be called
    for root, dirs, files, in os.walk(args):
        for item in files:
            print(items)
if __name__ = '__main__':
    main(sys.argv[1]) # specific argument called by function