python - 我如何使用 Python 获取包含特定文件(即'__main__.py)的子目录列表

Question

嘿伙计们，我正在尝试获取包含 python 中特定文件的子目录列表？？

我使用了 subprocess.call()并传递 find 命令，

subprocess.call(['find','.','-name','__main__.py'])

但它在终端中打印路径("./directory/file")...

Answer 1

使用 subprocess.check_output代替 subprocess.call，您可以获得命令的输出:

try:
    output = subprocess.check_output(['find','.','-name','__main__.py'])
except subprocess.CalledProcessError:
    output = '' # fallback output

output 是一个字符串。您可能需要拆分行来处理每个文件。

filelist = output.splitlines()

要仅获取包含该文件的目录的名称，请使用 os.path.dirname :

import os
for filepath in filelist:
    dirpath = os.path.dirname(filepath)
    # Do something with `dirpath`