python os模块调用小练习

1.使用os模块的内置方法，显示指定路径下所有指定的格式文件, 提示: html，md， filename.endswith('html')

import os
list1=[]
def func(path):
    # 判断路径是否正确
    if not os.path.exists(path):    
        return "path路径不存在"
    
    filename=os.listdir(path)
    for x in filename:
        # 拼接到原路径之后
        filepath = os.path.join(path, x)
        # 递归,如果拼接后的路径是文件夹，则回调原函数
        if os.path.isdir(filepath):
            func(filepath)
        else:
            # 如果拼接后的路径不是文件夹，则对路径后缀进行判断
            if filepath.rpartition('.')[-1] =='html':
                list1.append(filepath)
    return list1
# 传入路径
list2=func(r'C:\Users\continue\Desktop')
print(list2)

2. 统计文件夹大小 (os.path.getsize()：获取文件大小)

def dir_size(path):
    if not os.path.exists(path):
        return 'path路径不存在'
    size = 0        # 初始一个容器
    # 获取文件路径下的目录：文件夹、文件
    for file in os.listdir(path):
        # 拼接到路径
        abs_path = os.path.join(path, file)
        # 判断拼接后的路径是否是文件
        if os.path.isfile(abs_path):
            # 获取文件大小，并装入到容器中
            size += os.path.getsize(abs_path)
        # 判断拼接后的路径如果是文件夹，则回调原函数：递归
        elif os.path.isdir(abs_path):
            size += dir_size(abs_path)
    return size
size2 = dir_size(r'路径')   # 自定义路径
print('size:', size2)