61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
|
from hashlib import sha256
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import zipfile
|
|||
|
|
|
|||
|
|
# 获取当前目录
|
|||
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
|
|||
|
|
# 构建输出目录路径
|
|||
|
|
package_dir = os.path.join(current_dir, "output", "package")
|
|||
|
|
|
|||
|
|
shutil.rmtree(package_dir, ignore_errors=True)
|
|||
|
|
shutil.rmtree(os.path.join(current_dir, "output", "package.zip"), ignore_errors=True)
|
|||
|
|
|
|||
|
|
# 确保输出目录存在
|
|||
|
|
os.makedirs(package_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 复制文件
|
|||
|
|
shutil.copy(os.path.join(current_dir, "output", "LPT262_hilink_UPGRADE.bin"),
|
|||
|
|
os.path.join(package_dir, "image2_all_ota1.bin"))
|
|||
|
|
shutil.copy(os.path.join(current_dir, "output", "LPT262_hilink_UPGRADE.bin"),
|
|||
|
|
os.path.join(package_dir, "image2_all_ota2.bin"))
|
|||
|
|
|
|||
|
|
print("文件已复制到输出目录")
|
|||
|
|
|
|||
|
|
json_file = os.path.join(current_dir, "output", "package", "filelist.json")
|
|||
|
|
json_data = {}
|
|||
|
|
|
|||
|
|
def calc_sha256(filepath):
|
|||
|
|
h = sha256()
|
|||
|
|
with open(filepath, "rb") as f:
|
|||
|
|
for chunk in iter(lambda: f.read(4096), b""):
|
|||
|
|
h.update(chunk)
|
|||
|
|
return h.hexdigest()
|
|||
|
|
|
|||
|
|
file1 = os.path.join(package_dir, "image2_all_ota1.bin")
|
|||
|
|
file2 = os.path.join(package_dir, "image2_all_ota2.bin")
|
|||
|
|
|
|||
|
|
json_data['image2_all_ota1.bin'] = {}
|
|||
|
|
json_data['image2_all_ota1.bin']['sha256'] = calc_sha256(file1)
|
|||
|
|
json_data['image2_all_ota2.bin'] = {}
|
|||
|
|
json_data['image2_all_ota2.bin']['sha256'] = calc_sha256(file2)
|
|||
|
|
|
|||
|
|
print("file1 sha1: ", json_data['image2_all_ota1.bin']['sha256'])
|
|||
|
|
print("file2 sha2: ", json_data['image2_all_ota2.bin']['sha256'])
|
|||
|
|
|
|||
|
|
with open(json_file, "w") as f:
|
|||
|
|
json.dump(json_data, f)
|
|||
|
|
|
|||
|
|
package_file = os.path.join(current_dir, "output", "package.zip")
|
|||
|
|
with zipfile.ZipFile(package_file, "w") as zip:
|
|||
|
|
# 遍历package_dir目录及其所有子文件夹，将所有文件打包，并保留目录结构
|
|||
|
|
for root, dirs, files in os.walk(package_dir):
|
|||
|
|
for file in files:
|
|||
|
|
file_path = os.path.join(root, file)
|
|||
|
|
# 归一化存储在zip中的路径，使其以package_dir为根目录
|
|||
|
|
arcname = os.path.relpath(file_path, os.path.dirname(package_dir))
|
|||
|
|
zip.write(file_path, arcname)
|
|||
|
|
|
|||
|
|
print("升级包：package.zip sha256: ", calc_sha256(package_file))
|