05-遍历三种方式(递归遍历,广度遍历,深度遍历)

#复制day08目录,并显示复制进度百分比(递归遍历,广度遍历,深度遍历)
"""
目录删除:remove_dir
文件复制:file_operation
目录复制:copy_dir调用了文件复制
目录总大小:dir_size
"""
#1.递归方式遍历
import os

#目录的删除,参数1:目录路径
def remove_dir(dir_path):
    global counter
    list1 = os.listdir(dir_path)
    for name in list1:
        abs_path1 = os.path.join(dir_path,name)
        if os.path.isdir(abs_path1):
            counter += 1
            remove_dir(abs_path1)
            counter -= 1
            os.rmdir(abs_path1)
        elif os.path.isfile(abs_path1):
            os.remove(abs_path1)
    if counter == 0:
        os.rmdir(dir_path)

#文件复制操作
def file_operation(old_dir,new_dir,names):
    dict_old = os.path.join(old_dir,names)
    dict_new = os.path.join(new_dir,names)
    file1 = open(dict_old,"rb")
    file2 = open(dict_new,"wb")
    while True:
        rf = file1.read(1024*1024)
        if len(rf) == 0:
            file1.close()
            file2.close()
            #读取该文件大小,getsize只会读取文件大小,不会读取文件夹
            res = os.path.getsize(dict_old)
            return res
        file2.write(rf)
        file2.flush()


#复制目录,并通过百分比显示已经复制了多少
def copy_dir(new_dir,old_dir,file_size,flag):
    #原目录不存在,退出
    if not os.path.exists(old_dir):
        return
    if not os.path.exists(new_dir):
        os.makedirs(new_dir)
    #传入总文件大小
    global sum_size
    list1 = os.listdir(old_dir)
    for name in list1:
        #判断是文件还是目录
        if os.path.isdir(os.path.join(old_dir,name)):   #index == -1 or index == 0
            # 是目录
            dir_dir_old = os.path.join(old_dir,name)
            dir_dir_new = os.path.join(new_dir,name)
            #在新文件夹,创建新目录
            os.mkdir(dir_dir_new)
            #递归,只有将file_size,flag与函数一起带入,才能进行完全的计算
            file_size,flag= copy_dir(dir_dir_new,dir_dir_old,file_size,flag)
        else:
            # 是文件
            file_size += file_operation(old_dir,new_dir,name)
            percentage = float("%2.2f"%((file_size / sum_size)*100))
            #相同的数据则不会进行重复显示
            if flag != percentage:
                 flag = percentage
                 print("已经完成进度:%2.2f" %(percentage)+"%")
    return file_size,flag


#获取目录的总大小   参数1:目录路径名, 参数2:传入0 ,返回值:目录的大小
def dir_size(old_dir,sum_size):
    if not os.path.exists(old_dir):
        return
    list1 = os.listdir(old_dir)
    for name in list1:
        abs_path = os.path.join(old_dir,name)
        if os.path.isdir(abs_path):
            sum_size = dir_size(abs_path,sum_size)
        elif os.path.isfile(abs_path):
            sum_size += os.path.getsize(abs_path)

    return sum_size

if __name__ == '__main__':
    old_dir = r"F:\千峰Python\python1704\day08-文件,目录,路径操作"
    new_dir = r"F:\千峰Python\python1704\day08-复制品"
    #用来删除目录使用
    counter = 0
    #求目录大小
    sum_size = dir_size(old_dir,0)
    print("目录大小为:%2.1f M"%(sum_size/(1024*1024)))
    #新目录若存在,则删除,再创建
    if os.path.exists(new_dir):
        #删除目录
        print("该目录已经存在,正在删除.......")
        remove_dir(new_dir)
        print("删除完成")
    #复制文件,并显示已经完成进度,百分比显示
    copy_dir(new_dir,old_dir,0,0.0)

# ---------------------------------------------------------------------------------------------------
"""
用多种方式独立完成目录的遍历,目录的复制
1.递归-----上面的问题就以递归方式解决
2.广度遍历
3.深度遍历
"""
#2.广度遍历(队列)

# 创建空队列
# 1.将根目录加载进对队列
# 2.取出队列元素
# 3.遍历目录下一层的文件和目录,若文件复制,若目录压入队列
# 4.直到队列无元素,关闭文件

import os
from collections import deque

old_dir = r"F:\千峰Python\python1704\day08-文件,目录,路径操作"
new_dir = r"F:\千峰Python\python1704\day08-复制品"


def file_operation(old_dir,new_dir):
    file1 = open(old_dir,"rb")
    file2 = open(new_dir,"wb")
    while True:
        rf = file1.read(1024*1024)
        if len(rf) == 0:
            file1.close()
            file2.close()
            break
        file2.write(rf)
        file2.flush()

#创建两个空队列
queue1 = deque()
queue2 = deque()
#入队根目录
queue1.append(old_dir)
queue2.append(new_dir)
#如果目录不存在,则创建
if not os.path.exists(new_dir):
    os.makedirs(new_dir)
while True:
    if len(queue1) == 0:
        break
    file_path1 = queue1.popleft()
    file_path2 = queue2.popleft()
    list1 = os.listdir(file_path1)
    for res in list1:
        abs_path1 = os.path.join(file_path1,res)
        abs_path2 = os.path.join(file_path2,res)
        if os.path.isfile(abs_path1):
            print("文件操作")
            file_operation(abs_path1,abs_path2)
        else:
            print("目录操作")
            queue1.append(abs_path1)
            queue2.append(abs_path2)
            os.makedirs(abs_path2)

--------------------------------------------------------------------------------------------------------

#3.深度遍历
import os
from collections import deque

old_dir = r"F:\千峰Python\python1704\day08-文件,目录,路径操作"
new_dir = r"F:\千峰Python\python1704\day08-复制品"

def file_operation(old_dir,new_dir):
    file1 = open(old_dir,"rb")
    file2 = open(new_dir,"wb")
    while True:
        rf = file1.read(1024*1024)
        if len(rf) == 0:
            file1.close()
            file2.close()
            break
        file2.write(rf)
        file2.flush()

#创建两个空栈
stack1 = []
stack2 = []
#入栈根目录
stack1.append(old_dir)
stack2.append(new_dir)
#如果目录不存在,则创建
if not os.path.exists(new_dir):
    print("1111")
    os.makedirs(new_dir)
while True:
    if len(stack1) == 0:
        break
    file_path1 = stack1.pop()
    file_path2 = stack2.pop()
    list1 = os.listdir(file_path1)
    for res in list1:
        abs_path1 = os.path.join(file_path1,res)
        abs_path2 = os.path.join(file_path2,res)
        if os.path.isfile(abs_path1):
            print("文件操作")
            file_operation(abs_path1,abs_path2)
        else:
            print("目录操作")
            stack1.append(abs_path1)
            stack2.append(abs_path2)
            os.makedirs(abs_path2)