文件数据备份脚本

目的：实现多层文件夹文件的检索，筛选需要的文件，完成备份路径的创建（根据时间和传参[项目名 版本名 等]），完成文件的copy，然后对备份的所有文件进行压缩

py文件增加了3个方法：

1.路径创建

2.文件检索与copy

3.文件压缩

 

使用.bat文件带3个参数调用该py文件

#! /usr/bin/env python
#coding=utf-8
#Edit：Sandy Zheng
#date:2021/7/6

import os
import time
import os, shutil
import os.path
import time
from pathlib import Path
import zipfile
import sys

Projectname=sys.argv[1]
Type =sys.argv[2]
VERSION=sys.argv[3]

# 格式化
CurrentTime = (time.strftime("%Y-%m-%d %H-%M-%S", time.localtime()))

CurrentTime1 = (time.strftime("%Y %m %d %H %M %S", time.localtime()))
print (CurrentTime)
def mkdir(path):

    # 去除首位空格
    path = path.strip()
    # 去除尾部  符号
    path = path.rstrip("\")

    # 判断路径是否存在
    # 存在     True
    # 不存在   False
    isExists = os.path.exists(path)

    # 判断结果
    if not isExists:
        # 如果不存在则创建目录
        # 创建目录操作函数
        os.makedirs(path)

        print(path + ' 创建成功')
        return True
    else:
        # 如果目录存在则不创建，并提示目录已存在
        print(path + ' 目录已存在')
        return False

def copy_search_file(srcDir, desDir):
    ls = os.listdir(srcDir)
    for line in ls:
        filePath = os.path.join(srcDir, line)
        if os.path.isfile(filePath):
            shutil.copy(filePath, desDir)

        if os.path.isdir(filePath) and not(filePath.endswith("ExtFlash")):
            ls1 = os.listdir(filePath)
            for line1 in ls1:
                filePath2 = os.path.join(filePath,line1)
                if os.path.isfile(filePath2):
                    mkdir(desDir+"\"+line)
                    shutil.copy(filePath2,desDir+"\"+line)

            if Path(filePath + "\ExtFlash").exists():
                ls3 = os.listdir(filePath + "\ExtFlash")
                for line2 in ls3:
                    filePath0 = os.path.join(filePath + "\ExtFlash", line2)
                    mkdir(filePath+"\"+line)
                    if os.path.isfile(filePath0):
                        if filePath0.endswith(".hex"):
                            shutil.copy(filePath0, desDir+"\"+line)

    if Path(srcDir+"\ExtFlash").exists():
        ls = os.listdir(srcDir+"\ExtFlash")
        for line in ls:
            filePath = os.path.join(srcDir+"\ExtFlash", line)
            if os.path.isfile(filePath):
                if filePath.endswith(".hex"):
                    shutil.copy(filePath, desDir)

def zip_ya(startdir,file_news):
    #startdir要压缩的文件夹路径
    #file_news是要压缩后压缩文件名，需要带后缀
    z = zipfile.ZipFile(file_news,'w',zipfile.ZIP_DEFLATED) #参数一：文件夹名
    for dirpath, dirnames, filenames in os.walk(startdir):
        fpath = dirpath.replace(startdir,'') #这一句很重要，不replace的话，就从根目录开始复制
        fpath = fpath and fpath + os.sep or ''#，实现当前文件夹以及包含的所有文件的压缩
        for filename in filenames:
            z.write(os.path.join(dirpath, filename),fpath+filename)
            print ('压缩成功')
    z.close()

#copy_search_file("D:\test\src","D:\test\out")
#zip_ya("D:\test\out","D:\test\out"+".zip")


FinalPath = "D:FirmwareBackUp"+"\"+Projectname+"\"+CurrentTime+"\"+Type+VERSION
print ("固件备份路径是："+FinalPath)
mkdir(FinalPath)
copy_search_file(".\"+"\out" ,FinalPath)
zip_ya(FinalPath,FinalPath+".zip")
oldname = FinalPath+".zip"
newname = Projectname+" "+CurrentTime+" "+Type+VERSION+".zip"
os.rename(oldname,newname)
shutil.copy(newname,".\"+"\out")
shutil.move(newname,FinalPath)