04-目录的操作

os.mkdir("目录路径")------------------------>只会删除空目录

os.path.getsize("文件路径")------------------>只能获得文件大小,不能获得目大小

#目录的删除,参数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)

#获取目录的总大小   参数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

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

#文件复制操作
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*8)
        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