Python获取指定目录下的所有文件路径、获取指定目录下所有文件名（但是不包含子目录中文件名）、获取指定目录下所有pdf文件名（但是不包含子目录中pdf文件名）

Python获取指定目录下的所有文件路径、获取指定目录下所有文件名（但是不包含子目录中文件名）、获取指定目录下所有pdf文件名（但是不包含子目录中pdf文件名）

# -*- coding: utf-8 -*-
import os

# 指定目录
strSpecifiedDirPath = os.getcwd()


# 获取指定目录下所有文件路径（包含子目录中文件路径）
def getDirAllFileNames(strDirPath):
    listFilePaths = []

    for root, dirs, files in os.walk(strDirPath):
        for file in files:
            listFilePaths.append(os.path.join(root, file))

    return listFilePaths


# 获取指定目录下所有文件名（但是不包含子目录中文件名）
def getDirFileNames(strDirPath):
    listFileNames = []

    for root, dirs, files in os.walk(strDirPath):
        for file in files:
            if os.path.exists(os.path.join(strDirPath, file)):
                # listFileNames.append(os.path.join(strDirPath, file))
                listFileNames.append(file)

    return listFileNames


# 获取指定目录下所有pdf文件名（但是不包含子目录中pdf文件名）
def getDirPdfFileNames(strDirPath):
    listPdfFileNames = []

    for root, dirs, files in os.walk(strDirPath):
        for file in files:
            if os.path.splitext(file)[1] == '.pdf':
                if os.path.exists(os.path.join(strDirPath, file)):
                    # listPdfFileNames.append(os.path.join(strDirPath, file))
                    listPdfFileNames.append(file)

    return listPdfFileNames


# 主函数
if __name__ == "__main__":
    # 获取指定目录下所有文件路径（包含子目录中文件路径）
    print(getDirAllFileNames(strSpecifiedDirPath))

    # 获取指定目录下所有pdf文件名（但是不包含子目录中pdf文件名）
    print(getDirFileNames(strSpecifiedDirPath))

    # 获取指定目录下所有pdf文件名（但是不包含子目录中pdf文件名）
    print(getDirPdfFileNames(strSpecifiedDirPath))