169 lines
7.4 KiB
Python
169 lines
7.4 KiB
Python
# This Python file uses the following encoding: utf-8
|
||
import sys
|
||
import uuid
|
||
import re
|
||
import shutil
|
||
import zipfile
|
||
import hashlib
|
||
import os
|
||
import json
|
||
from PyQt6 import *
|
||
from PyQt6.QtCore import *
|
||
from PyQt6.QtGui import *
|
||
from PyQt6.QtWidgets import *
|
||
|
||
from ui_form import Ui_Widget
|
||
|
||
class Widget(QWidget):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.ui = Ui_Widget()
|
||
self.ui.setupUi(self)
|
||
self.block = False
|
||
self.appjs = ""
|
||
self.ui.pbPackage.clicked.connect(self.compressPackage)
|
||
self.ui.leappId.textChanged.connect(self.updateInfo)
|
||
self.ui.leDes.textChanged.connect(self.updateInfo)
|
||
self.ui.leName.textChanged.connect(self.updateInfo)
|
||
self.ui.leVersion.textChanged.connect(self.updateInfo)
|
||
self.ui.pbOpenFile.clicked.connect(self.openFileDialog)
|
||
self.ui.pbOpen.clicked.connect(self.openDir)
|
||
|
||
def openFileDialog(self):
|
||
file_dialog = QFileDialog(self)
|
||
file_dialog.setWindowTitle("选择工程")
|
||
file_dialog.setFileMode(QFileDialog.FileMode.Directory)
|
||
if file_dialog.exec():
|
||
file_path = file_dialog.selectedFiles()[0]
|
||
# 获取info文件的路径
|
||
info_path = os.path.join(file_path, "info")
|
||
self.block = True
|
||
if os.path.exists(info_path):
|
||
# 读取文件内容
|
||
with open(info_path, 'r', encoding="utf-8") as f:
|
||
content = f.read()
|
||
info = json.loads(content)
|
||
# 将文件内容保存到列表中
|
||
self.ui.leappId.setText(info["General"]["AppId"])
|
||
self.ui.leName.setText(info["General"]["Name"])
|
||
self.ui.leDes.setText(info["General"]["Description"])
|
||
self.ui.leVersion.setText(info["General"]["Version"])
|
||
else:
|
||
self.ui.leappId.setText(str(uuid.uuid4()))
|
||
self.ui.leName.setText("")
|
||
self.ui.leDes.setText("")
|
||
self.ui.leVersion.setText("")
|
||
self.block = False
|
||
self.ui.lePro.setText(file_path)
|
||
|
||
def updateInfo(self):
|
||
if self.block:
|
||
return
|
||
# 获取info文件的路径
|
||
info_path = os.path.join(self.ui.lePro.text(), "info")
|
||
info = {
|
||
"General": {
|
||
"AppId": self.ui.leappId.text(),
|
||
"Name": self.ui.leName.text(),
|
||
"Description": self.ui.leDes.text(),
|
||
"Version": self.ui.leVersion.text(),
|
||
"Show": False
|
||
}
|
||
}
|
||
# 将文件内容保存到列表中
|
||
with open(info_path, 'w', encoding="utf-8") as f:
|
||
json.dump(info, f, ensure_ascii=False, indent=4)
|
||
|
||
def calculate_file_md5(self, file_path):
|
||
# 计算给定文件的MD5值。
|
||
md5 = hashlib.md5()
|
||
with open(file_path, 'rb') as f:
|
||
for chunk in iter(lambda: f.read(4096), b""):
|
||
md5.update(chunk)
|
||
return md5.hexdigest()
|
||
|
||
def calculate_directory_md5(self, directory_path):
|
||
#计算给定目录下除md5.txt文件以外的全部文件的MD5值。
|
||
md5_values = {}
|
||
with os.scandir(directory_path) as it:
|
||
for entry in it:
|
||
if entry.is_file() and entry.name != 'md5.txt':
|
||
file_md5 = self.calculate_file_md5(entry.path)
|
||
md5_values[entry.name] = file_md5
|
||
elif entry.is_dir():
|
||
sub_directory_md5 = self.calculate_directory_md5(entry.path)
|
||
md5_values[entry.name] = sub_directory_md5
|
||
return {k: v for k, v in md5_values.items() if v != {} }
|
||
|
||
def openDir(self):
|
||
directory_path = self.ui.leDir.text()
|
||
os.makedirs( directory_path, exist_ok=True)
|
||
os.startfile(directory_path)
|
||
def compressPackage(self):
|
||
try:
|
||
filename = self.ui.leName.text()+".tpa"
|
||
packeagePath = os.path.join(self.ui.leDir.text(), filename)
|
||
fname = QFileDialog.getSaveFileName(self, '保存文件', packeagePath, 'TPA files (*.tpa)')
|
||
if fname[0]:
|
||
packeagePath = fname[0]
|
||
if os.path.exists(packeagePath):
|
||
os.remove(packeagePath)
|
||
print('文件已删除')
|
||
filename_with_extension = os.path.basename(fname[0]) # 获取带有后缀的文件名
|
||
filename = filename_with_extension.replace('.tpa', '') # 删除后缀，获取没有后缀的文件名
|
||
#更新app.js
|
||
js_path = os.path.join(self.ui.lePro.text(), "app.js")
|
||
with open(js_path, 'r') as file:
|
||
self.appjs = file.read()
|
||
|
||
with open(js_path, 'w', encoding="utf-8") as file:
|
||
file.write('var appId = \"' + self.ui.leappId.text() + '\"\nvar plusFile = \"plus\"')
|
||
|
||
self.ui.lbStatus.setText("开始打包")
|
||
directory_path = self.ui.lePro.text()
|
||
target_path = os.path.dirname(fname[0])
|
||
directory_md5 = self.calculate_directory_md5(directory_path)
|
||
json_str = json.dumps(directory_md5, indent=4, sort_keys=True)
|
||
md5_str = ""
|
||
md5_str = hashlib.md5(json_str.encode('utf-8')).hexdigest()
|
||
if md5_str != "":
|
||
with open(os.path.join(self.ui.lePro.text(),"md5.txt"), 'w') as f:
|
||
f.write(md5_str)
|
||
os.makedirs( self.ui.leDir.text(), exist_ok=True)
|
||
zip_path = os.path.join(self.ui.leDir.text(), filename +".zip")
|
||
# 创建ZIP文件并添加目录内容
|
||
with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
|
||
# 递归添加目录和文件到ZIP文件中
|
||
self.add_dir_to_zip(directory_path, zipf, directory_path)
|
||
filename, ext = os.path.splitext(zip_path)
|
||
os.rename( zip_path, os.path.join( target_path, filename_with_extension) )
|
||
#还原debug app.js
|
||
with open(js_path, 'w', encoding="utf-8") as file:
|
||
file.write(self.appjs)
|
||
self.ui.lbStatus.setText("打包成功")
|
||
else:
|
||
self.ui.lbStatus.setText("md5为空")
|
||
except Exception as e:
|
||
try:
|
||
os.remove(zip_path)
|
||
except:
|
||
pass
|
||
self.ui.lbStatus.setText(f"打包失败{e}")
|
||
|
||
def add_dir_to_zip(self, dir_path, zipf, parent_dir):
|
||
for root, dirs, files in os.walk(dir_path):
|
||
for file in files:
|
||
file_path = os.path.join(root, file)
|
||
# 使用os.path.relpath获取相对于parent_dir的路径作为归档名称
|
||
arcname = os.path.relpath(file_path, parent_dir)
|
||
zipf.write(file_path, arcname=arcname)
|
||
for subdir in dirs:
|
||
subdir_path = os.path.join(root, subdir)
|
||
# 递归调用add_dir_to_zip()函数，将空目录也添加到ZIP文件中
|
||
self.add_dir_to_zip(subdir_path, zipf, parent_dir)
|
||
if __name__ == "__main__":
|
||
app = QApplication(sys.argv)
|
||
widget = Widget()
|
||
widget.show()
|
||
sys.exit(app.exec())
|