MENU

U盘文件自动备份脚本——python

September 29, 2019 • 鼓捣其他东西

U盘备份脚本——python

想法的形成:

前几天博主不小心把U盘连带钥匙一起丢了,U盘里面全是博主从高中开始到大学的资料,博主是一个念旧的人,所以觉得好珍贵,特别可惜,于是脑海里就有了这个想法。

一、 主要原理

  • 主要原理就是通过python提供的os库进行操作系统的文件拷贝命令
  • 通过多线程技术加快文件拷贝速度

二、特点

  1. 开机自动运行,每次运行生成Log文件,以备出错时检查
  2. U盘插入首先判断是否为需要复制的U盘,否则不进行复制
  3. U盘内存放相应的配置文件,实现U盘内文件的版本更新(待实现)

三、 具体代码

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

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

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

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

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

    
     '''
     从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]))

    
    ```
  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
    ```
  1. 整体代码:

    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秒进行下一次数据获取
    ```

四、 使用方法

因为项目尚未完工,所以还没有进行打包,项目详情请访问github
  1. 首先将文件拷贝至本地文件夹,确保电脑存在D盘盘符(因为U盘文件默认储存在'D:\CopyFileRoot'目录中)
  2. 然后右键生成USB_bak.pyw文件的快捷方式,将快捷方式放入win系统的启动文件夹下。
  3. 双击USB_bak.pyw文件即可运行,因为是后台进程所以不会有界面,打开进程管理器可以看到python虚拟机在运行。
  4. 如要关闭程序只需打开进程管理器杀掉python虚拟机进程。
Archives QR Code
QR Code for this page
Tipping QR Code