如何在Ubuntu中使用Python脚本高效复制U盘文件到本地目录

前言

在现代办公和生活中，U盘作为一种便捷的存储设备，经常被用于数据传输。然而，手动复制大量文件不仅费时费力，还容易出错。作为一名技术爱好者，我决定利用Python脚本来自动化这一过程，提高工作效率。本文将详细介绍如何在Ubuntu系统中使用Python脚本高效地复制U盘文件到本地目录。

一、准备工作

环境搭建
- 确保你的Ubuntu系统已安装Python。可以通过以下命令检查：
```
python3 --version
```
- 如果未安装，可以使用以下命令进行安装：
```
sudo apt update
sudo apt install python3
```
挂载U盘
- 将U盘插入电脑，通常Ubuntu会自动挂载U盘。可以通过以下命令查看挂载点：
```
df -h
```
- 找到U盘的挂载点，例如 /media/yourusername/USB_DRIVE_NAME。

二、编写Python脚本

导入所需库
```
import os
import shutil
```

source_dir = '/media/yourusername/USB_DRIVE_NAME'
target_dir = '/home/yourusername/local_directory'

创建目标目录（如果不存在）

if not os.path.exists(target_dir):
   os.makedirs(target_dir)

复制文件

def copy_files(src, dst):
   for item in os.listdir(src):
       s = os.path.join(src, item)
       d = os.path.join(dst, item)
       if os.path.isdir(s):
           shutil.copytree(s, d)
       else:
           shutil.copy2(s, d)

主函数

if __name__ == '__main__':
   copy_files(source_dir, target_dir)
   print("文件复制完成！")

三、运行脚本

保存脚本
- 将上述代码保存为 copy_usb_files.py。
赋予执行权限
```
chmod +x copy_usb_files.py
```
运行脚本
```
./copy_usb_files.py
```

四、优化与扩展

错误处理
- 在实际使用中，可能会遇到各种异常情况，如U盘未挂载、目标目录无写权限等。可以在脚本中添加异常处理机制，提高脚本的健壮性。
```
try:
   copy_files(source_dir, target_dir)
   print("文件复制完成！")
except Exception as e:
   print(f"发生错误：{e}")
```
日志记录
- 为了更好地跟踪脚本执行情况，可以添加日志记录功能。
”`python import logging

logging.basicConfig(filename=‘copy_usb_files.log’, level=logging.INFO)

def copy_files(src, dst):

   for item in os.listdir(src):
       s = os.path.join(src, item)
       d = os.path.join(dst, item)
       try:
           if os.path.isdir(s):
               shutil.copytree(s, d)
           else:
               shutil.copy2(s, d)
           logging.info(f"已复制：{s} -> {d}")
       except Exception as e:
           logging.error(f"复制失败：{s} -> {d}，原因：{e}")


3. **命令行参数**
   - 为了使脚本更灵活，可以添加命令行参数，允许用户指定源目录和目标目录。
   ```python
   import argparse

   parser = argparse.ArgumentParser(description='复制U盘文件到本地目录')
   parser.add_argument('source', help='U盘挂载点')
   parser.add_argument('target', help='本地目标目录')
   args = parser.parse_args()

   source_dir = args.source
   target_dir = args.target

五、总结

后记

Python作为一种强大的编程语言，其应用范围非常广泛。除了文件复制，你还可以利用Python进行数据分析、网络爬虫、自动化测试等多种任务。希望你能继续探索Python的无限可能，提升自己的技术水平。

Happy Coding! ?