【Python】多线程实现批量文件移动（带进度条）

引言

编写Python脚本实现将一个文件夹中的所有文件按照修改日期（xxxx年xx月xx日）分类移动。

依赖

pip install alive_progress 

代码

import os
import time
import shutil
import concurrent.futures as cf
from alive_progress import alive_bar  # pip install alive_progress


def main():
    name_list = os.listdir()
    name_list.remove(os.path.basename(__file__))
    with alive_bar(len(name_list)) as bar: # 进度条
        with cf.ThreadPoolExecutor() as p: # 线程池
            for name in name_list:
                time_stamp = os.path.getmtime(name) # 获取文件修改日期时间戳
                time_local = time.localtime(time_stamp) # 将时间戳转换为本地时间
                time_style = time.strftime(r'%Y%m%d', time_local) # 将本地时间转换为指定格式的时间
                os.makedirs(time_style, exist_ok=True)
                p.submit(shutil.move, name, time_style).add_done_callback(lambda func: bar())


if __name__ == '__main__':
    main()

提示

这里进度条打印用的是alive_progress库，用tqdm库也可以，代码改一下：

import os
import time
import shutil
from tqdm import tqdm # pip install tqdm
import concurrent.futures as cf


def main():
    name_list = os.listdir()
    name_list.remove(os.path.basename(__file__))
    with tqdm(total=len(name_list)) as bar: # 进度条
        with cf.ThreadPoolExecutor() as p: # 线程池
            for name in name_list:
                time_stamp = os.path.getmtime(name) # 获取文件修改日期时间戳
                time_local = time.localtime(time_stamp) # 将时间戳转换为本地时间
                time_style = time.strftime(r'%Y%m%d', time_local) # 将本地时间转换为指定格式的时间
                os.makedirs(time_style, exist_ok=True)
                p.submit(shutil.move, name, time_style).add_done_callback(lambda func: bar.update())


if __name__ == '__main__':
    main()

个人觉得tqdm进度条不如alive_progress好看和详细，对比如下：

• tqdm
  
• alive_progress
  

参考

https://docs.python.org/zh-cn/3.11/library/concurrent.futures.html