forked from sanmusen214/BAAH
-
Notifications
You must be signed in to change notification settings - Fork 0
/
package.py
199 lines (162 loc) · 6.95 KB
/
package.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
import zipfile
import shutil
import os
from modules.configs.MyConfig import config
import subprocess
from pathlib import Path
import nicegui
import time
import pponnxcr
import platform
import requests
def package_download_adb(platformstr = None):
target_adb_path = os.path.join(os.getcwd(), "tools", "adb")
downloadurls = {
"Windows": "https://dl.google.com/android/repository/platform-tools-latest-windows.zip",
"Darwin": "https://dl.google.com/android/repository/platform-tools-latest-darwin.zip",
"Linux": "https://dl.google.com/android/repository/platform-tools-latest-linux.zip"
}
if not os.path.exists(target_adb_path):
if platformstr and platformstr in downloadurls.keys():
url = downloadurls[platformstr]
elif platform.system() in downloadurls.keys():
url = downloadurls[platform.system()]
else:
print(f"Unknown platform: {platform.system()}")
return
# download zip
r = requests.get(url)
with open("platform-tools-latest.zip", "wb") as f:
f.write(r.content)
target_adb_path_parent_folder = os.path.dirname(target_adb_path)
# unzip to target_adb_path, rename the upper folder "playform-tools" to "adb"
with zipfile.ZipFile("platform-tools-latest.zip", 'r') as z:
z.extractall(target_adb_path_parent_folder)
print(f"adb downloaded to: {target_adb_path_parent_folder}")
package_rename(os.path.join(target_adb_path_parent_folder, "platform-tools"), target_adb_path)
else:
print(f"adb already exists: {target_adb_path}")
def package_copyfolder(src, dst):
try:
# 拷贝文件夹
shutil.copytree(src, dst)
print(f"{dst}文件夹已拷贝")
except FileExistsError as e:
print(f"{dst}文件夹已存在!")
def package_copyfile(src, dst):
try:
shutil.copyfile(src, dst)
print(f"{dst}已拷贝")
except FileExistsError as e:
print(f"{dst}已存在!")
def package_rename(src, dst):
try:
os.rename(src, dst)
except Exception as e:
print(f"{dst}已存在!")
def package_create_folder(path):
try:
os.makedirs(path)
except Exception as e:
print(f"{path}创建时出错!")
def package_remove_file(path):
try:
os.remove(path)
except Exception as e:
print(f"{path}删除时出错!")
def package_remove_folder(path):
try:
shutil.rmtree(path)
except Exception as e:
print(f"{path}删除时出错!")
# ====================开始====================
config_version = config.NOWVERSION
# mainly for windows, download adb
package_download_adb(platformstr="Windows")
package_remove_folder("./dist")
# 打包main.py,名字为BAAH
baahcmd = [
'pyinstaller',
'main.py',
'-n', 'BAAH',
'--icon', './DATA/assets/kei.ico',
'--add-data', f'{Path(pponnxcr.__file__).parent}{os.pathsep}pponnxcr',
'-y'
]
subprocess.call(baahcmd)
# 打包GUI
guicmd = [
'pyinstaller',
'jsoneditor.py',
# '--windowed', # prevent console appearing, only use with ui.run(native=True, ...)
'--add-data', f'{Path(nicegui.__file__).parent}{os.pathsep}nicegui',
'--icon', './DATA/assets/aris.ico',
'-y'
]
subprocess.call(guicmd)
# 打包update.py,名字为BAAH_UPDATE
updatecmd = [
'pyinstaller',
'update.py',
'-n', 'BAAH_UPDATE',
'--icon', './DATA/assets/kayoko.ico',
'-y'
]
subprocess.call(updatecmd)
# 当前目录
print("当前目录:", os.getcwd())
workdir = os.getcwd()
print("开始封装")
# 遍历./dist/jsoneditor/_internal里的所有文件夹和文件,将它们拷贝到./dist/BAAH/_internal,如果已存在则跳过
for dirpath, dirnames, filenames in os.walk(os.path.join('./dist', 'jsoneditor', '_internal')):
for filename in filenames:
package_copyfile(os.path.join(dirpath, filename), os.path.join('./dist/BAAH/_internal', filename))
for dirname in dirnames:
package_copyfolder(os.path.join(dirpath, dirname), os.path.join('./dist/BAAH/_internal', dirname))
# 走一层就终止
break
package_copyfolder('./tools/adb', './dist/BAAH/tools/adb')
# pytinstall的时候已经把pponnxcr和nicegui文件拷贝进去了
# package_copyfolder('./tools/pponnxcr', './dist/BAAH/_internal/pponnxcr')
# 挪i18n进去创建下DATA文件夹
package_copyfolder("./DATA/i18n", "./dist/BAAH/DATA/i18n")
package_create_folder("./dist/BAAH/DATA/CONFIGS")
# 将LICENSE挪进去占位, 不放software.config, 防止覆盖掉用户的
package_copyfile("./LICENSE", "./dist/BAAH/DATA/CONFIGS/LICENSE")
# 这里只拷贝example.json,不拷贝其他的,因为其他的是用户的配置文件
# package_copyfolder("./BAAH_CONFIGS", "./dist/BAAH/BAAH_CONFIGS")
package_create_folder("./dist/BAAH/BAAH_CONFIGS")
package_copyfile("./BAAH_CONFIGS/example.json", "./dist/BAAH/BAAH_CONFIGS/example.json")
package_copyfolder("./DATA/assets", "./dist/BAAH/DATA/assets")
package_copyfolder("./DATA/assets_jp", "./dist/BAAH/DATA/assets_jp")
package_copyfolder("./DATA/assets_cn", "./dist/BAAH/DATA/assets_cn")
package_copyfolder("./DATA/assets_global_en", "./dist/BAAH/DATA/assets_global_en")
package_copyfolder("./DATA/grid_solution", "./dist/BAAH/DATA/grid_solution")
package_copyfile("./dist/jsoneditor/jsoneditor.exe", "./dist/BAAH/jsoneditor.exe")
package_copyfile("./dist/BAAH_UPDATE/BAAH_UPDATE.exe", "./dist/BAAH/BAAH_UPDATE.exe")
time.sleep(2)
# package_rename("./dist/BAAH/BAAH.exe", f"./dist/BAAH/BAAH{config_version}.exe")
package_rename("./dist/BAAH/jsoneditor.exe", "./dist/BAAH/BAAH_GUI.exe")
package_rename("./dist/BAAH", f"./dist/BAAH{config_version}")
package_remove_file("./BAAH.exe")
package_copyfile(f"./dist/BAAH{config_version}/BAAH.exe", "./BAAH.exe")
print("开始压缩")
time.sleep(2)
# 压缩./dist/BAAH文件夹为BAAH.zip
z = zipfile.ZipFile(f'./dist/BAAH{config_version}.zip', 'w', zipfile.ZIP_DEFLATED)
startdir = f"./dist/BAAH{config_version}"
for dirpath, dirnames, filenames in os.walk(startdir):
for filename in filenames:
z.write(os.path.join(dirpath, filename), arcname=os.path.join(dirpath, filename).replace("/dist",""))
print(f"完成,压缩包./dist/BAAH{config_version}.zip已生成")
print(f"压缩包大小为{os.path.getsize(f'./dist/BAAH{config_version}.zip')/1024/1024:.2f}MB")
# 压缩./dist/BAAH文件夹(除了_internal, tools)为BAAH_update.zip
z = zipfile.ZipFile(f'./dist/BAAH{config_version}_update.zip', 'w', zipfile.ZIP_DEFLATED)
startdir = f"./dist/BAAH{config_version}"
for dirpath, dirnames, filenames in os.walk(startdir):
if "_internal" in dirpath or "tools" in dirpath or "BAAH_CONFIGS" in dirpath:
continue
for filename in filenames:
z.write(os.path.join(dirpath, filename), arcname=os.path.join(dirpath, filename).replace("/dist",""))
print(f"完成,压缩包./dist/BAAH{config_version}_update.zip已生成")
print(f"压缩包大小为{os.path.getsize(f'./dist/BAAH{config_version}_update.zip')/1024/1024:.2f}MB")