def zip_file_path(input_path, output_path, output_name, compress_level=None):
"""
压缩文件
:param compress_level: 压缩等级
:param input_path: 压缩的文件夹路径
:param output_path: 解压（输出）的路径
:param output_name: 压缩包名称
:return:
"""
f = zipfile.ZipFile(output_path + '/' + output_name, 'w', zipfile.ZIP_DEFLATED, compresslevel=compress_level)
file_list = []
get_zip_file(input_path, file_list)
before_size = 0
file_count = len(file_list)
file_index = 0
while file_index < file_count:
file = file_list[file_index]
before_size += os.path.getsize(file)
arc_name = file[len(input_path):]
# 添加文件而非完整路径
f.write(file, arc_name)
sys.stdout.write(str(int((file_index / file_count) * 100)) + "%\r")
sys.stdout.flush()
file_index += 1
f.close()
after_size = os.path.getsize(output_path + '/' + output_name)
if before_size != 0:
Console.success("源文件总大小：", before_size, "压缩后大小：", after_size, "压缩率：", after_size / before_size * 100, "%")
else:
Console.info("文件总大小为0")
return len(file_list)
def main(argv):
"""
:param argv:
"""
err_msg = 'zip.py -i <input_path> -o <output_path> -n <output_name> [-l <compress_level>]'
input_path = ''
output_path = ''
output_name = ''
compress_level = None
try:
opts, args = getopt.getopt(argv, "hi:o:n:l:",
["input_path=", "output_path=", "output_name=", "compress_level="])
if len(argv) < 6:
Console.error(err_msg)
sys.exit(2)
except getopt.GetoptError as e:
Console.error(e.msg, e)
sys.exit(2)
before = time.time()
for opt, arg in opts:
if opt == '-h':
Console.info(err_msg)
sys.exit()
elif opt in ("-i", "--input_path"):
if arg == "":
Console.error("目标路径不能为空！")
sys.exit(2)
input_path = arg
elif opt in ("-o", "--output_path"):
if arg == "":
Console.error("输出路径不能为空！")
sys.exit(2)
output_path = arg
elif opt in ("-n", "--output_name"):
if arg == "":
Console.error("输出文件名不能为空！")
sys.exit(2)
output_name = arg
elif opt in ("-l", "--compress_level"):
if not arg == "":
try:
compress_level = int(arg)
except ValueError:
compress_level = arg
r = zip_file_path(input_path, output_path, output_name, compress_level)
Console.success("完成压缩，共：", r, "个项目", "\r\n输出文件：", output_path + '/' + output_name, "耗时：",
time_escape(time.time() - before))
file_list = []
argv = sys.argv[1:]
input_path = ''
_type = ''
debug = True
try:
opts, args = getopt.getopt(argv, "hi:t:r", ["input_path=", "type="])
if len(argv) < 4:
Console.error(err_msg)
sys.exit(2)
except getopt.GetoptError as e:
Console.error(e.msg, e)
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
Console.info(err_msg)
sys.exit()
elif opt in ("-i", "--input_path"):
if arg == "":
Console.error("目标路径不能为空！")
sys.exit(2)
input_path = arg
elif opt in ("-t", "--type"):
if arg == "":
Console.error("压缩格式不能为空！")
sys.exit(2)
_type = arg
elif opt == '-r':
debug = False
before = time.time()
before_size = 0
