[Python基础 ] Day_11_作业_参考答案

这期的作业有点复杂,如果不会一定要好好看笔记,自己多尝试几次,还是不会的可以积极私信问一下我~

''''''
import os

# 1.显示指定路径下所有视频格式文件, 提示: 视频格式mp4，avi，rmvb
#   filename.endswith('mp4')
def search_dir(path):
    if not os.path.exists(path):
        return

    file_list = os.listdir(path)
    for file in file_list:
        file_path = os.path.join(path, file)
        if os.path.isfile(file_path):
            # print('文件:', file)
            if file.endswith('.mp4') or file.endswith('.avi') \
                    or file.endswith('.rmvb') or file.endswith('.wmv'):
                print(file)
        else:
            # print("目录:", file)

            search_dir(file_path)


search_dir(r'C:\Users\JohnLuo\Desktop\Python2004')


# 2.自定义模块:
#  a.建立一个包
#  b.在包的下创建一个排序的模块
#     模块下的功能
#        1. 使用冒泡排序对列表进行降序排序
#        def fn(l):
#
#        2. 使用选择排序对列表进行升序排序
#
#        3. 查找列表的元素
#           找到返回下标, 找不到返回-1
#           def fn(l, n):
#
#        4.使用顺序查询，获取列表中所有与指定元素重复的元素下标
#                 def fn(l, n):
#
#  在另外一个文件中导入上述包中的模块，完成模块中功能的调用
from user import sort
print(sort.fn1([2,3,4,5,6]))


# 3. 统计文件夹大小 (os.path.getSize()：获取文件大小)
# 提示: 遍历目录


def get_path_size(path):
    if not os.path.exists(path):
        return

    size = 0

    file_list = os.listdir(path)
    for file in file_list:
        file_path = os.path.join(path, file)
        # 如果是文件
        if os.path.isfile(file_path):
            size += os.path.getsize(file_path)

        else:
            size += get_path_size(file_path)

    return size


s = get_path_size(r'C:\Users\JohnLuo\Desktop\Python2004')
print(s)