Python脚本实现USB自动复制文件

USB驱动器作为常见的数据存储设备,经常用于数据传输和备份。

然而,我们在手动处理文件复制可能效率低下且容易出错。

因此,我们可以利用Python编写脚本来自动化这一过程,提高效率和数据安全性。

准备工作

首先,我们需要安装Python以及几个必要的库:os、shutil、time和psutil。这些库将帮助我们实现文件操作、定时和系统监测功能。

获取硬盘驱动器列表

复制代码
这段代码利用psutil库获取系统中可移动的磁盘驱动器列表,例如USB驱动器。
c 复制代码
import psutil

def get_disk_drives():
    disk_partitions = psutil.disk_partitions(all=False)
    drives = [partition.device.upper() for partition in disk_partitions if partition.fstype != "" and "removable" in partition.opts]
    return drives

文件拷贝功能

复制代码
这段代码定义了一个函数,用于将指定文件类型(.jpg、.png、.txt)从USB驱动器拷贝到指定目标文件夹。
它确保在复制过程中限制了数据传输的速度,避免过载系统。
c 复制代码
import os
import shutil

def copy_ppt_files(source_folder, destination_folder, speed_limit_kb):
    for root, dirs, files in os.walk(source_folder):
        for file in files:
            if file.endswith((".jpg", ".png", ".txt")):
                src_file = os.path.join(root, file)
                dst_file = os.path.join(destination_folder, os.path.relpath(src_file, source_folder))
                os.makedirs(os.path.dirname(dst_file), exist_ok=True)
                with open(src_file, 'rb') as fsrc:
                    with open(dst_file, 'wb') as fdst:
                        shutil.copyfileobj(fsrc, fdst, length=speed_limit_kb * 1024)

检查新插入的USB驱动器

复制代码
此函数定期检查新插入的USB驱动器,并调用文件拷贝函数将特定文件类型复制到预定目标文件夹。
它通过在列表中记录已知驱动器来避免重复操作。
c 复制代码
import time

def check_for_new_drive(speed_limit_kb=10240):
    drives = get_disk_drives()
    new_drives = [drive for drive in drives if drive not in known_drives]
    for new_drive in new_drives:
        known_drives.append(new_drive)
        print(f"New drive detected: {new_drive}")
        time.sleep(3)  # 等待3秒后再开始拷贝
        copy_ppt_files(new_drive, destination_drive, speed_limit_kb)

主程序

复制代码
主程序初始化了已知驱动器列表和目标路径,并通过调用check_for_new_drive函数来持续检查新插入的USB驱动器。
c 复制代码
if __name__ == "__main__":
    known_drives = []
    excluded_drives = [drive + ':' for drive in "ABCDEFGHIJKLMNOPQRSTUVWXYZ"]
    destination_drive = "H://u盘"  # 目标路径
    if not os.path.exists(destination_drive):
        os.makedirs(destination_drive)
    while True:
        check_for_new_drive()
        time.sleep(60)  # 每隔60秒检查一次
相关推荐
风铃喵游2 分钟前
构建引擎: 打造小程序编译器
前端·小程序·架构
_WndProc4 分钟前
【Python】Flask网页
开发语言·python·flask
互联网搬砖老肖5 分钟前
Python 中如何使用 Conda 管理版本和创建 Django 项目
python·django·conda
sunbyte7 分钟前
50天50个小项目 (Vue3 + Tailwindcss V4) ✨ | ThemeClock(主题时钟)
前端·javascript·css·vue.js·前端框架·tailwindcss
深栈解码7 分钟前
JMM深度解析(三) volatile实现机制详解
java·后端
小飞悟16 分钟前
🎯 什么是模块化?CommonJS 和 ES6 Modules 到底有什么区别?小白也能看懂
前端·javascript·设计
浏览器API调用工程师_Taylor16 分钟前
AOP魔法:一招实现登录弹窗的全局拦截与动态处理
前端·javascript·vue.js
测试者家园16 分钟前
基于DeepSeek和crewAI构建测试用例脚本生成器
人工智能·python·测试用例·智能体·智能化测试·crewai
FogLetter17 分钟前
初识图片懒加载:让网页像"懒人"一样聪明加载
前端·javascript
微客鸟窝19 分钟前
一文搞懂NVM管理Node.js:从安装到实战全攻略
前端