吾爱破解 - LCG - LSG |安卓破解|病毒分析|破解软件|www.52pojie.cn

 找回密码
 注册[Register]

QQ登录

只需一步,快速开始

搜索
查看: 177|回复: 0

[Python] python实现U盘备份脚本

  [复制链接]
fenglinger 发表于 2019-10-9 18:53

U盘备份脚本——python

想法的形成:

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


一、 主要原理

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

二、特点

  1. 开机自动运行,每次运行生成Log文件,以备出错时检查
    [md]

    log文件

    log文件

    [md]
  2. U盘插入首先判断是否为需要复制的U盘,否则不进行复制
  3. U盘内存放相应的配置文件,实现U盘内文件的版本更新(待实现)

三、 使用方法

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

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

四、 具体代码

  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]))
    
  4. 处理盘符信息:

    获取当前系统内的盘符列表并判断是否有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
  5. 整体代码:

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

免费评分

参与人数 3吾爱币 +6 热心值 +3 收起 理由
0001 + 2 + 1 热心回复!
为海尔而战 + 1 + 1 我很赞同!
苏紫方璇 + 3 + 1 感谢发布原创作品,吾爱破解论坛因你更精彩!

查看全部评分

发帖前要善用论坛搜索功能,那里可能会有你要找的答案或者已经有人发布过相同内容了,请勿重复发帖。

您需要登录后才可以回帖 登录 | 注册[Register]

本版积分规则 警告:禁止回复与主题无关内容,违者重罚!

快速回复 收藏帖子 返回列表 搜索

RSS订阅|小黑屋|联系我们|吾爱破解 - LCG - LSG ( 京ICP备16042023号 | 京公网安备 11010502030087号 )

GMT+8, 2019-10-14 22:25

Powered by Discuz!

© 2001-2017 Comsenz Inc.

快速回复 返回顶部 返回列表