标签：devices led about pat success stat os.walk close targe
# coding:utf-8
import sys, os, time
import zipfile
import shutil
def print_usage():
print ‘‘‘usage:
python software_upgrade.py subZipFullPath, targetZipFullPath, mainVersionZipPath[, relativePath]\n
for example:
linux platform:
python software_upgrade.py /test/software_upgrade/subVersionDir/readme.txt /test/software_upgrade/latestVersionDir/myAPP_latest.zip /test/software_upgrade/mainVersionDir/myAPP_main.zip /readme.txt
python software_upgrade.py /test/software_upgrade/subVersionDir/myAPP_subVersion_0.1.zip /test/software_upgrade/latestVersionDir/myAPP_latest.zip /test/software_upgrade/mainVersionDir/myAPP_main.zip
windows platform:
python software_upgrade.py C:\software_upgrade\subVersionDir\readme.txt C:\software_upgrade\latestVersionDir\myAPP_latest.zip C:\software_upgrade\mainVersionDir\myAPP_main.zip \readme.txt
python software_upgrade.py C:\software_upgrade\subVersionDir\myAPP_subVersion_0.1.zip C:\software_upgrade\latestVersionDir\myAPP_latest.zip C:\software_upgrade\mainVersionDir\myAPP_main.zip
‘‘‘
def my_copytree(src, dst, symlinks=False):
"""
若同样的目标已存在则不动
Args:
src:
dst:
symlinks:
Returns:
"""
names = os.listdir(src)
if not os.path.isdir(dst):
os.makedirs(dst)
errors = []
for name in names:
srcname = os.path.join(src, name)
dstname = os.path.join(dst, name)
try:
if symlinks and os.path.islink(srcname):
linkto = os.readlink(srcname)
os.symlink(linkto, dstname)
elif os.path.isdir(srcname):
my_copytree(srcname, dstname, symlinks)
else:
if os.path.isdir(dstname):
os.rmdir(dstname)
elif os.path.isfile(dstname):
os.remove(dstname)
shutil.copy2(srcname, dstname)
# XXX What about devices, sockets etc.?
except (IOError, os.error) as why:
errors.append((srcname, dstname, str(why)))
# catch the Error from the recursive copytree so that we can
# continue with other files
except OSError as err:
errors.extend(err.args[0])
try:
shutil.copystat(src, dst)
except WindowsError:
# can‘t copy file access times on Windows
pass
except OSError as why:
errors.extend((src, dst, str(why)))
if errors:
pass
# raise Error(errors)
def unzip(filename, filedir):
"""
Args:
filename: ‘foobar.zip‘ #要解压的文件
filedir: 解压后放入的目录
Returns:
"""
r = zipfile.is_zipfile(filename)
if r:
starttime = time.time()
fz = zipfile.ZipFile(filename, ‘r‘)
for file in fz.namelist():
# print(file) # 打印zip归档中目录
fz.extract(file, filedir)
endtime = time.time()
times = endtime - starttime
else:
print(‘This file is not zip file‘)
print(‘[unzip file] time costs:‘ + str(times))
def zip(path, filename):
"""
默认情况下，zip压缩会保留根目录,我们这里不保留根目录
Args:
path: 要进行压缩的文档目录
filename: ‘foobar.zip‘ # 压缩后的文件名
Returns:
"""
try:
import zlib
compression = zipfile.ZIP_DEFLATED # 压缩方法
except:
compression = zipfile.ZIP_STORED
starttime = time.time()
# start = path.rfind(os.sep) + 1 # os.sep 分隔符 (保留根目录)
start = len(path) # （不保留根目录）
z = zipfile.ZipFile(filename, mode="w", compression=compression)
try:
for dirpath, dirs, files in os.walk(path):
for file in files:
if file == filename or file == "zip.py":
continue
# print(file)
z_path = os.path.join(dirpath, file)
z.write(z_path, z_path[start:])
z.close()
endtime = time.time()
times = endtime - starttime
except:
if z:
z.close()
print(‘[create zip file] time costs:‘ + str(times))
def create_latest_package(subVersionFullPath, latestVersionFullPath, mainVersionFullPath, relativePath):
"""
Args:
subVersionFullPath:
latestVersionFullPath: latest_Zip_path
mainVersionFullPath:
relativePath:
Returns:
"""
if os.path.exists(subVersionFullPath) and os.path.exists(mainVersionFullPath): # 判断子版本和主版本zip包是否存在
if not os.path.exists(latestVersionFullPath):
# 如果不存在最新版本的zip包，将子版本的最新文件与主版本的文件进行组装，然后将组装后的文件压缩到最新版本所在的目录
oper(subVersionFullPath, mainVersionFullPath, latestVersionFullPath, relativePath)
else:
# 如果存在最新版本的zip包，将子版本的最新文件与最新版本的文件进行组装，然后将组装后的文件压缩到最新版本所在的目录
oper(subVersionFullPath, latestVersionFullPath, latestVersionFullPath, relativePath)
else:
print "not exists {0} or {1}".format(subVersionFullPath, mainVersionFullPath)
def oper(srvZipPath, destZipPath, outZipPath, relativePath):
"""
注意：多文件替换时，一般多文件存在压缩包中，所以需要解压
基本思路 1.将最新版本的zip文件解压到同级的tmp目录中
2.如果是单文件升级，则将子版本的文件拷贝到1步骤中的tmp目录里；如果是多文件升级，则首先需要将多文件解压，然后再拷贝
3.将步骤2替换后的tmp文件重新压缩，并放到新版本的位置
Args:
srvZipPath: 子版本zip路径
destZipPath: 目标版本zip路径
outZipPath: 组装后，生成新的zip文件的完整路径
relativePath: 单文件和多文件标识，若为None则表明多文件替换，否则单文件替换
Returns:
"""
# 若最终生成的zip文件所在的路径不存在，则创建
outZipPathDir = os.path.dirname(outZipPath)
os.makedirs(outZipPathDir) if not os.path.exists(outZipPathDir) else ‘‘
# 确定临时文件的路径
srvZipPathTmp = os.path.dirname(srvZipPath) + os.sep + "tmp"
destZipPathTmp = os.path.dirname(destZipPath) + os.sep + "tmp"
print "[unzip target zip] srv:{0} dest:{1}".format(destZipPath, destZipPathTmp)
unzip(destZipPath, destZipPathTmp) # 解压目标zip文件
if relativePath: # 单文件和多文件标识，若为None则表明多文件替换，否则单文件替换
print "[single file replace] srv:{0} dest:{1}".format(srvZipPath, destZipPathTmp + relativePath)
shutil.copy(srvZipPath, destZipPathTmp + relativePath) # 将要升级的文件拷贝到已经解压开的目标zip文件destZipPathTmp中
else:
print "[unzip source zip] srv:{0} dest:{1}".format(srvZipPath, srvZipPathTmp)
unzip(srvZipPath, srvZipPathTmp)
print "[replace files] srvZipPathTmp:{0} destZipPathTmp:{1}".format(srvZipPathTmp, destZipPathTmp)
my_copytree(srvZipPathTmp, destZipPathTmp) # 将要升级的文件拷贝到已经解压开的目标zip文件destZipPathTmp中
print "[create new zip file] srv:{0} dest:{1}".format(destZipPathTmp, outZipPath)
zip(destZipPathTmp, outZipPath)
# 删除临时文件
shutil.rmtree(srvZipPathTmp) if os.path.exists(srvZipPathTmp) else ‘‘
shutil.rmtree(destZipPathTmp) if os.path.exists(destZipPathTmp) else ‘‘
print "success"
if __name__ == ‘__main__‘:
number = 0
for i in sys.argv:
print "arg{0}: {1}".format(number, i)
number += 1
if len(sys.argv) < 4 or len(sys.argv) > 5:
print_usage()
if len(sys.argv) == 4:
create_latest_package(sys.argv[1], sys.argv[2], sys.argv[3], None)
elif len(sys.argv) == 5:
create_latest_package(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4])
标签：devices led about pat success stat os.walk close targe
原文地址：http://www.cnblogs.com/Stubborn-Ant/p/7251645.html
