在 python 制作文件列表

Question

I don't know where's the problem. Can you help me?

main.py

import os
from functions import *


if __name__ == "__main__":
    path = os.getenv('path')
    print(readFiles(path))
    

functions.py

import os

def readFiles(patha):
    res = []

    # Iterate directory
    currentpath = os.listdir(patha)
    
    for path in currentpath:
        # check if current path is a file
        if os.path.isfile(os.path.join(patha, path)):
            res.append(path)
        else:
            pathf = patha + path + '/'
            res.append(readFiles(pathf))


    return res

error

I wanted to make a file list in python and this error appeared. No idea what went wrong.

Answer 1

Check your os.getenv('path') variable to make sure it is as expected. Also, I updated your code to make it work smoothly.

from pathlib import Path


def readFiles(patha):
    res = []
    print(patha)

    # Iterate directory
    currentpath = os.listdir(patha)

    for path in currentpath:
        # check if current path is a file
        try:
            f = os.path.join(patha, path)
            if os.path.isfile(f):
                res.append(f)
            elif os.path.isdir(f):
                res.extend(readFiles(f))
        except FileNotFoundError as e:
            print("warning,", e)

    return res


if __name__ == "__main__":
    path = Path.cwd()
    print(readFiles(path))

I suggest you to take a look at this question as well: How do I list all files of a directory?