使用python读取文件夹下文件名称

想读取某个文件夹下的文件名及类别；

测试代码如下

#读取某个一个文件夹下所有子文件的名称，占用内存，格式类型；一个文件存储一行，以txt文本输出
import os
import sys

def list_files_in_folder(folder_path):
    """
    通过argv参数导入文件夹路径，读取某个文件夹下所有子文件的名称，占用内存，格式类型；一个文件存储一行，以txt文本输出.

    :param folder_path: str, 文件夹的路径
    :return: None
    """
    # 检查是否存在文件夹路径
    if not os.path.exists(folder_path):
        print("文件夹路径不存在")
        return

    # 获取文件夹中的所有文件
    file_names = [file for file in os.listdir(folder_path) if os.path.isfile(os.path.join(folder_path, file))]

    # 将文件信息写入txt文件
    with open("file_info.txt", "w", encoding="utf-8") as file:
        for file_name in file_names:
            file_path = os.path.join(folder_path, file_name)
            file_size = os.path.getsize(file_path)
            file_type = os.path.splitext(file_name)[1]
            file.write(f"{file_name},{file_size},{file_type}\n")

# 从命令行参数获取文件夹路径
if __name__ == "__main__":
    folder_path = sys.argv[1]
    list_files_in_folder(folder_path)

运行方法：

python .\*.py  "file_path"