我的Python_野生自学笔记-02__USB 后台复制文件-CSDN博客

本文链接：https://blog.csdn.net/jingmo_see/article/details/103170820

USB 后台复制文件

请勿用于非法途径

笔者开发环境:

python 3.7.5
pycharm

上代码:

import os
import time
import shutil
from os.path import join, getsize

# U盘路径
USB = 'E:\\'

# 保存路径
SAVE = 'D:\\U盘资料'

# 获取U盘中所有文件夹，及文件大小
# 形参：理解函数可以实现某种功能的模块，方法。实参，申明实参时，要给一个确定的值
def get_dir_size(dir):
    # 初始化
    size = 0
    for root, dirs, files in os.walk(dir):
        # 获取U盘中资料大小，推导式
        size += sum(getsize(join(root, name)) for name in files)
    print(size)
    return size


# 专门做拷贝
def usb_copy():
    # 是shutil的内置函数，专做拷贝
    shutil.copytree(USB, SAVE)


def main():
    old_dirsize = 0
    # 一直运行，不退出，死循环
    while True:
        if os.path.exists(USB):
            print('检测到优盘.......')
            new_dirsize = get_dir_size(USB)
            # 做文件夹判断，如果文件夹大小不等于U盘大小
            # 这样就确定没有复制文件
            if old_dirsize != new_dirsize:
                usb_copy()
                old_dirsize = new_dirsize
            else:
                print('没有变化......')
        else:
            print('暂时没有检测到优盘......')

        print('开始休眠')
        time.sleep(5)
        print('休眠文件.....')


if __name__ == "__main__":
    main()