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))
# Debug模式起效的注解
if debug:
if r[2:] == "Debug":
res.append(f"say from file:{file}")
Console.warning(f"注意：{file}包含Debug注解")
elif r[2:].startswith("Debug "):
res.append(r[8:])
Console.warning(f"注意：{file}包含Debug指令")
else:
break
if len(res) > 1:
with open(file, 'w', encoding="utf-8") as f:
f.writelines(res)
after_size += os.path.getsize(file)
if _index > 0:
Console.success("尝试处理", _index, "次后成功解决")
_index = 0
except IOError as e:
Console.warning("处理：" + file + "发生IO错误", e)
except UnicodeDecodeError as e:
_index += 1
if _index > b_size:
Console.warning("处理：" + file + "发生无法解决的Unicode解码错误", e.reason)
_index = 0
else:
if _index == 1:
Console.warning("处理：" + file + "发生Unicode解码异常", e.reason)
file_index -= 1
finally:
file_index += 1
else:
