› 首页 ›编程教程›正文

python实现U盘备份脚本

piaodoo 编程教程 2020-02-22 22:09:12 1004 0 python教程

本文来源吾爱破解论坛

本帖最后由 fenglinger 于 2019-11-6 20:07 编辑
U盘备份脚本——python 想法的形成：

前几天我不小心把U盘连带钥匙一起丢了，U盘里面全是我从的各种资料，觉得好可惜，于是脑海里就有了这个想法。

一、 主要原理 主要原理就是通过python提供的os库进行操作系统的文件拷贝命令 通过多线程技术加快文件拷贝速度 二、特点 开机自动运行，每次运行生成Log文件，以备出错时检查
[md]

Snipaste_2019-10-07_17-37-50.png (63.61 KB, 下载次数: 0)

下载附件  保存到相册

log文件

2019-10-9 18:51 上传
[md] U盘插入首先判断是否为需要复制的U盘，否则不进行复制 U盘内存放相应的配置文件，实现U盘内文件的版本更新（待实现） 限制多线程的数量（19.11.06添加需求待实现） 添加配置文件实现备份特定目录（19.11.06添加需求待实现）

楼主大学生准备出国所以忙于雅思，更新可能会很慢

三、 使用方法

因为项目尚未完工，所以还没有进行打包，项目详情请访问github

首先将文件拷贝至本地文件夹，确保电脑存在D盘盘符（因为U盘文件默认储存在'D:\CopyFileRoot'目录中） 然后右键生成USB_bak.pyw文件的快捷方式，将快捷方式放入win系统的启动文件夹下。 双击USB_bak.pyw文件即可运行，因为是后台进程所以不会有界面，打开进程管理器可以看到python虚拟机在运行。 如要关闭程序只需打开进程管理器杀掉python虚拟机进程。
四、 具体代码

首先声明以下全局变量以共使用

logRoot = "..\\Log"     # 本地log文件储存路径
logName = "Log_"+ time.strftime("%Y%m%d",time.localtime(time.time()))   # Log文件名字，以时间命名
targetRoot = 'D:\CopyFileRoot'  # 目标目录
oldDiskName = []  # 旧的磁盘列表
number = 0  # 磁盘数，判断是否为第一次运行
bakpath = ""    # u盘上的标记文件路径

程序初始化

程序每次运行，都会判断程序是否是第一次在次电脑上运行,初次运行就创建相应的文件及文件目录

    '''
    初始化函数
    '''
    def chackFirst():  # 判断是否是第一次运行
        if not os.path.exists(logRoot):  # 判断本地log文件夹是否存在
            os.mkdir(logRoot)
        if not os.path.exists(targetRoot):  # 判断U盘本地备份路径是否存在
            os.mkdir(targetRoot)

文件复制函数

通过递归的方式进行文件的拷贝


    '''
    从sourcepath复制文件和目录到targetPath
    '''
    def copyfile(sourcePath, targetPath, threadName):
        for f in os.listdir(sourcePath):
            if (f == 'System Volume Information'):  # 过滤系统文件夹
                continue

            f1 = os.path.join(sourcePath, f)  # 连接源文件（目录）名
            f2 = os.path.join(targetPath, f)  # 连接目标文件（目录）名

            if os.path.isfile(f1):  # 如果为文件，则进行复制操作
                if not os.path.exists(f2):
                    file1 = open(f1, 'rb')
                    file2 = open(f2, 'wb')
                    printLog(threadName + '-%s文件正在复制！' % (f1))
                    file2.write(file1.read())
                    printLog(threadName + '-%s文件复制成功！' % (f1))
                else:
                    printLog(threadName + '文件已存在！' )
            else:  # 如果为目录，创建新一级的目标目录，并递归操作
                printLog(threadName + '-%s目录正在复制！' % (f2))
                if not os.path.exists(f2):
                    os.mkdir(f2)
                    printLog(threadName + '-%s目标目录创建成功！' % (f2))
                else:
                    printLog('复制失败，原因：目录已存在！')
                copyfile(f1, f2, threadName)
                printLog(threadName + '-%s目录复制成功！' % (f1))

     '''
        复制盘符name中的文件到目标目录中
    '''
    def copy(name, threadName):
        f = open(bakpath, 'a+')     #以追加模式写入。此处是为了之后实现对于U盘文件的版本更新
        timeNow = str(time.strftime('%Y%m%d%H%M%S', time.localtime(time.time())))  # 获取当前时间字串
        targetPath = os.path.join(targetRoot, name[:1])  # 创建一个新目录，使用目标目录+盘符作为名字
        if not os.path.exists(targetPath):
            os.mkdir(targetPath)  # 创建新的目录
        copyfile(name, targetPath, threadName)  # 复制文件
        f.close()
        printLog(threadName + '-新磁盘：%s盘 复制完毕！' % (name[:1]))

处理盘符信息：

获取当前系统内的盘符列表并判断是否有U盘插入

    '''
    获取磁盘信息，并与上次获取的信息进行比较，判断是否有新的磁盘添加进来
    '''

    def getDiskMessage():
        global oldDiskName  # 声明全局变量
        global number

        if number == 0:  # 第一次操作，先获取一遍磁盘数据，然后返回
            for disk in psutil.disk_partitions():
                number = number + 1
                oldDiskName.append(disk.device[:2])  # 获取盘符信息
            return

        newDiskName = []  # 保存新获取的磁盘信息
        for disk in psutil.disk_partitions():
            newDiskName.append(disk.device[:2])  # 获取新的磁盘信息

        newDiskList = arrayCompare(oldDiskName, newDiskName)  # 获取新增盘符列表

        oldDiskName.clear()  # 清除旧盘符列表
        oldDiskName = newDiskName[:]  # 复制新盘符列表给旧盘符列表
        return newDiskList

    '''
    比较两个磁盘盘符列表，并返回新盘符列表中旧盘符列表没有的盘符名列表
    '''

    def arrayCompare(oldDiskName, newDiskName):
        newDiskList = []
        for name in newDiskName:
            if name not in oldDiskName:  # 旧盘符中没有，则添加这个到新增盘符列表中
                newDiskList.append(name)
        return newDiskList

整体代码：

```python
import os
import time
import psutil
from threading import Thread

logRoot = "..\\Log"
logName = "Log_"+ time.strftime("%Y%m%d",time.localtime(time.time()))
targetRoot = 'D:\CopyFileRoot'  # 目标目录
oldDiskName = []  # 旧的磁盘列表
number = 0  # 磁盘数，判断是否为第一次运行
bakpath = ""    # u盘上的标记文件路径

'''
    初始化函数
'''
def chackFirst():  # 判断是否是第一次运行
    if not os.path.exists(logRoot):  # 判断本地log文件夹是否存在
        os.mkdir(logRoot)
    if not os.path.exists(targetRoot):  # 判断本地备份文件是否存在
        os.mkdir(targetRoot)

'''
    写Log函数
'''
def printLog(content):
    f = open(os.path.join(logRoot,logName),'a+')
    f.write(time.strftime("%H:%M:%S    ",time.localtime(time.time())))
    f.write(content+'\n')
    f.close()

'''
从sourcepath复制文件和目录到targetPath
'''

def copyfile(sourcePath, targetPath, threadName):
    for f in os.listdir(sourcePath):
        if (f == 'System Volume Information'):  # 过滤系统文件夹
            continue

        f1 = os.path.join(sourcePath, f)  # 连接源文件（目录）名
        f2 = os.path.join(targetPath, f)  # 连接目标文件（目录）名

        if os.path.isfile(f1):  # 如果为文件，则进行复制操作
            if not os.path.exists(f2):
                file1 = open(f1, 'rb')
                file2 = open(f2, 'wb')
                printLog(threadName + '-%s文件正在复制！' % (f1))
                file2.write(file1.read())
                printLog(threadName + '-%s文件复制成功！' % (f1))
            else:
                printLog(threadName + '文件已存在！' )
        else:  # 如果为目录，创建新一级的目标目录，并递归操作
            printLog(threadName + '-%s目录正在复制！' % (f2))
            if not os.path.exists(f2):
                os.mkdir(f2)
                printLog(threadName + '-%s目标目录创建成功！' % (f2))
            else:
                printLog('复制失败，原因：目录已存在！')
            copyfile(f1, f2, threadName)
            printLog(threadName + '-%s目录复制成功！' % (f1))

'''
获取磁盘信息，并与上次获取的信息进行比较，判断是否有新的磁盘添加进来
'''

def getDiskMessage():
    global oldDiskName  # 声明全局变量
    global number

    if number == 0:  # 第一次操作，先获取一遍磁盘数据，然后返回
        for disk in psutil.disk_partitions():
            number = number + 1
            oldDiskName.append(disk.device[:2])  # 获取盘符信息
        return

    newDiskName = []  # 保存新获取的磁盘信息
    for disk in psutil.disk_partitions():
        newDiskName.append(disk.device[:2])  # 获取新的磁盘信息

    newDiskList = arrayCompare(oldDiskName, newDiskName)  # 获取新增盘符列表

    oldDiskName.clear()  # 清除旧盘符列表
    oldDiskName = newDiskName[:]  # 复制新盘符列表给旧盘符列表
    return newDiskList

'''
比较两个磁盘盘符列表，并返回新盘符列表中旧盘符列表没有的盘符名列表
'''

def arrayCompare(oldDiskName, newDiskName):
    newDiskList = []
    for name in newDiskName:
        if name not in oldDiskName:  # 旧盘符中没有，则添加这个到新增盘符列表中
            newDiskList.append(name)
    return newDiskList

'''
复制盘符name中的文件到目标目录中
'''

def copy(name, threadName):
        f = open(bakpath, 'a+')     #以追加模式写入
        timeNow = str(time.strftime('%Y%m%d%H%M%S', time.localtime(time.time())))  # 获取当前时间字串
        targetPath = os.path.join(targetRoot, name[:1])  # 创建一个新目录，使用目标目录+盘符作为名字
        if not os.path.exists(targetPath):
            os.mkdir(targetPath)  # 创建新的目录
        copyfile(name, targetPath, threadName)  # 复制文件
        f.close()
        printLog(threadName + '-新磁盘：%s盘 复制完毕！' % (name[:1]))

if __name__ == '__main__':
    getDiskMessage()  # 获取初始数据
    # targetRoot = chackFirst()  # 软件初始化
    threadCount = 0  # 线程计数
    printLog("开始运行")
    while True:

        newDiskList = getDiskMessage()  # 获取新数据
        if len(newDiskList) > 0:    # 检测到有新u盘插入
            printLog('当前磁盘列表：' + str(oldDiskName))
            printLog('新磁盘列表：' + str(newDiskList))
            time.sleep(10)      # 延迟等待u盘加载
            for name in newDiskList:  # 根据新获取到的数据去复制文件
                bakpath = os.path.join(name, 'USB.bak')
                t = os.path.exists(bakpath)
                if t:  # 没有文件则不是需要备份的u盘
                    # copy(name, 'thread_' + str(threadCount))
                    thread = Thread(target=copy, args=(name, 'thread_' + str(threadCount),))  # 创建线程去复制指定磁盘
                    thread.start()  # 开启线程
                    printLog('thread_' + str(threadCount) + '-开始复制%s盘文件...' % (name[:1]))
                    threadCount = threadCount + 1  # 线程计数+1
        time.sleep(10)   # 延时10秒进行下一次数据获取
```

版权声明：

本站所有资源均为站长或网友整理自互联网或站长购买自互联网，站长无法分辨资源版权出自何处，所以不承担任何版权以及其他问题带来的法律责任，如有侵权或者其他问题请联系站长删除！站长QQ754403226 谢谢。

有关影视版权：本站只供百度云网盘资源，版权均属于影片公司所有，请在下载后24小时删除，切勿用于商业用途。本站所有资源信息均从互联网搜索而来，本站不对显示的内容承担责任，如您认为本站页面信息侵犯了您的权益，请附上版权证明邮件告知【754403226@qq.com】，在收到邮件后72小时内删除。本文链接：https://www.piaodoo.com/7710.html

相关文章

搜索

文章专栏

  • Python 旋转打印各种矩形的方法
  • 对Python中小整数对象池和大整数对象池的使用详解
  • python 梯度法求解函数极值的实例
  • 通过PHP与Python代码对比的语法差异详解
  • python实现函数极小值
  • Python中面向对象你应该知道的一下知识
  • python 寻找离散序列极值点的方法
  • python 绘制拟合曲线并加指定点标识的实现
  • 使用python画社交网络图实例代码
  • 详解Python中的各种转义符\n\r\t
  • 我们为什么要减少Python中循环的使用
  • Python Matplotlib 基于networkx画关系网络图
  • Python求离散序列导数的示例
  • python卸载后再次安装遇到的问题解决
  • python networkx 包绘制复杂网络关系图的实现

    • 最近发表

    标签列表