-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathDEV-PACK.py
267 lines (240 loc) · 11.4 KB
/
DEV-PACK.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
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
import os
import sys
import subprocess
from colorama import init, Fore
# ------------------------- 警告 -------------------------
# 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！
# 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！
# 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！
# 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！
# 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！
# 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！
# -------------------------------------------------------
init(autoreset=True)
os.chdir(os.path.dirname(os.path.abspath(sys.argv[0])))
print(f"{Fore.BLUE}[!]{Fore.RESET} 将使用 {Fore.BLUE}Pyinstaller{Fore.RESET} 打包。")
print(f"{Fore.YELLOW}⚠{Fore.RESET} 本工具仅作为开发人员打包发行版时使用，请勿单独使用！！！！")
# 计数
fail = 0 # 失败的文件个数
countd = 0 # 已删除的文件个数
acount = 0 # 总文件个数
fcount = 0 # 已打包的文件个数
pyw_acount = 0
py_acount = 0
silent = False
if sys.argv[4].lower() in ["-s", "--silent", "silent", "quiet", "-q", "--quiet", "/s", "/q", "/silent", "/quiet"]:
silent = True
else:
from plyer import notification
if not sys.argv[1]:
# 文件夹路径
folder_path = input("请输入文件夹路径：")
else:
folder_path = sys.argv[1]
if folder_path.startswith(("'", '"')) and folder_path.endswith(("'", '"')):
folder_path = folder_path[1:-1]
if not folder_path.endswith('\\'):
folder_path += '\\'
if not os.path.exists(folder_path):
print(f"{Fore.RED}✕{Fore.RESET} 指定的目录路径不存在，请重新运行程序并输入有效的目录路径。")
if not silent:
input("按 ENTER 键继续...")
exit(1)
if not sys.argv[2]:
# 图标文件路径
icon_path = input("请输入图标文件路径：")
else:
icon_path = sys.argv[2]
if not sys.argv[3]:
# 日志文件存放文件夹
log_path = input("请输入日志文件存放文件夹：")
else:
log_path = sys.argv[3]
if not icon_path:
icon_path = "None"
print(f"{Fore.YELLOW}⚠{Fore.RESET} 将执行无图标打包！")
elif icon_path.startswith(("'", '"')) and icon_path.endswith(("'", '"')):
icon_path = icon_path[1:-1]
if not log_path:
log_path = "None"
print(f"{Fore.YELLOW}⚠{Fore.RESET} 将执行无日志打包！")
elif not log_path.endswith('\\'):
log_path += '\\'
os.makedirs(log_path, exist_ok=True)
for root, dirs, files in os.walk(folder_path):
for file in files:
if file.endswith('.py'):
file_path = os.path.join(root, file)
print(f"{Fore.BLUE}找到 py 文件: {file_path}")
py_acount += 1
elif file.endswith('.pyw'):
file_path = os.path.join(root, file)
print(f"{Fore.BLUE}找到 pyw 文件: {file_path}")
pyw_acount += 1
acount = py_acount + pyw_acount
print(f"{Fore.GREEN}一共找到了{Fore.BLUE}{acount}{Fore.GREEN}个py/pyw文件。\n其中有{Fore.BLUE}{py_acount}{Fore.GREEN}个py文件/{Fore.BLUE}{pyw_acount}{Fore.GREEN}个pyw文件。")
# 函数：记录日志并添加分隔线
def log_message(message, log_file):
# 日志中不应存在颜色
message = f"{message}"
log_file.write(message + "\n")
log_file.write("-" * 50 + "\n") # 添加分隔线
# 函数：打包 Python 文件
def package_py(file_path, log_file="None"):
global fcount
global fail
try:
if log_file != "None":
log_message(f"\n开始打包：{file_path}", log_file)
print(f"\n{Fore.GREEN}开始打包：{Fore.BLUE}{file_path}{Fore.RESET}")
output_dir = os.path.dirname(file_path) # 设置输出目录为 Python 文件所在目录
# 使用notification的
if os.path.basename(file_path) in ["网络连接检测.py", "GitHub连接检测.py", "使用Pyinstaller.py", "使用Nuitka.py", "连续push尝试.py", "连续pull尝试.py", "git连续尝试.py"]:
print(f"{Fore.YELLOW}⚠{Fore.RESET} 使用notification！")
if icon_path == "None":
command = f"pyinstaller --hidden-import plyer.platforms.win.notification --onefile --distpath=\"{output_dir}\" \"{file_path}\""
else:
command = f"pyinstaller --hidden-import plyer.platforms.win.notification --onefile -i \"{icon_path}\" --distpath=\"{output_dir}\" \"{file_path}\""
else:
if icon_path == "None":
command = f"pyinstaller --onefile --distpath=\"{output_dir}\" \"{file_path}\""
else:
command = f"pyinstaller --onefile -i \"{icon_path}\" --distpath=\"{output_dir}\" \"{file_path}\""
subprocess.run(command, shell=True, check=True)
if log_file != "None":
log_message(f"打包完成: {file_path}", log_file)
print(f"{Fore.GREEN}打包完成: {Fore.BLUE}{file_path}{Fore.RESET}")
fcount += 1
print(f"{Fore.GREEN}还剩 {Fore.BLUE}{acount-fcount}{Fore.GREEN} 个文件待打包。")
except subprocess.CalledProcessError as e:
if log_file != "None":
log_message(f"打包 {file_path} 时出错:\n{e}", log_file)
print(f"{Fore.RED}打包 {Fore.BLUE}{file_path}{Fore.RED} 时出错:\n{e}")
fail += 1
fcount += 1
if not silent:
notification.notify(
title='Pyinstaller快速打包程序提醒您',
message=f'打包程序炸啦！到现在一共炸了{fail}次。',
timeout=10
)
print(f"{Fore.GREEN}还剩 {Fore.BLUE}{acount-fcount}{Fore.GREEN} 个文件待打包。")
if silent or input(f"{Fore.BLUE}?{Fore.RESET} 是否继续打包 [Y/N]:").lower() not in ["y", "yes", "是", "继续"]:
sys.exit(1)
return file_path
# 函数：打包 Pythonw 文件
def package_pyw(file_path, log_file="None"):
global fcount
global fail
try:
if log_file != "None":
log_message(f"\n开始打包: {file_path}", log_file)
print(f"\n{Fore.GREEN}开始打包: {Fore.BLUE}{file_path}{Fore.RESET}")
output_dir = os.path.dirname(file_path) # 设置输出目录为 Pythonw 文件所在目录
# 使用notification的
if os.path.basename(file_path) in ["休息一下.pyw", "目录复制.pyw"]:
print(f"{Fore.YELLOW}⚠{Fore.RESET} 使用notification！")
if icon_path == "None":
command = f"pyinstaller --hidden-import plyer.platforms.win.notification --noconsole --onefile --distpath={output_dir} {file_path}"
else:
command = f"pyinstaller --hidden-import plyer.platforms.win.notification --noconsole --onefile -i \"{icon_path}\" --distpath={output_dir} {file_path}"
else:
if icon_path == "None":
command = f"pyinstaller --noconsole --onefile --distpath=\"{output_dir}\" \"{file_path}\""
else:
command = f"pyinstaller --noconsole --onefile -i \"{icon_path}\" --distpath=\"{output_dir}\" \"{file_path}\""
subprocess.run(command, shell=True, check=True)
if log_file != "None":
log_message(f"打包完成：{file_path}", log_file)
print(f"{Fore.GREEN}打包完成：{Fore.BLUE}{file_path}{Fore.RESET}")
fcount += 1
print(f"{Fore.GREEN}还剩 {Fore.BLUE}{acount-fcount}{Fore.GREEN} 个文件待打包。")
except subprocess.CalledProcessError as e:
if log_file != "None":
log_message(f"打包 {file_path} 时出错:\n{e}", log_file)
print(f"{Fore.RED}打包 {Fore.BLUE}{file_path}{Fore.RED} 时出错:\n{e}")
fail += 1
fcount += 1
if not silent:
notification.notify(
title='Pyinstaller快速打包程序提醒您',
message=f'打包程序炸啦！到现在一共炸了{fail}次。',
timeout=10
)
print(f"{Fore.GREEN}还剩 {Fore.BLUE}{acount-fcount}{Fore.GREEN} 个文件待打包。")
if silent or input(f"{Fore.BLUE}?{Fore.RESET} 是否继续打包 [Y/N]:").lower() not in ["y", "yes", "是", "继续"]:
sys.exit(1)
return file_path
if log_path == "None":
failed_files = [] # 存储打包失败的文件名
# 遍历文件夹中的所有文件
for root, dirs, files in os.walk(folder_path):
for file in files:
file_path = os.path.join(root, file)
# 根据文件后缀选择打包方式
if file.endswith(".py"):
failed_file = package_py(file_path)
if failed_file:
failed_files.append(failed_file)
elif file.endswith(".pyw"):
failed_file = package_pyw(file_path)
if failed_file:
failed_files.append(failed_file)
else:
with open(os.path.join(log_path, "packaging.log"), "a", encoding="utf-8") as log_file:
# 打开日志文件，准备记录日志
log_message(f"开始打包，需要打包的文件数量：{acount}", log_file)
failed_files = [] # 存储打包失败的文件名
# 遍历文件夹中的所有文件
for root, dirs, files in os.walk(folder_path):
for file in files:
file_path = os.path.join(root, file)
# 根据文件后缀选择打包方式
if file.endswith(".py"):
failed_file = package_py(file_path, log_file)
if failed_file:
failed_files.append(failed_file)
log_message(f"剩余待打包文件数量：{acount-fcount}", log_file)
elif file.endswith(".pyw"):
failed_file = package_pyw(file_path, log_file)
if failed_file:
failed_files.append(failed_file)
log_message(f"剩余待打包文件数量：{acount-fcount}", log_file)
# 提示用户打包完成
if fail != 0:
if not silent:
input(f"打包完成，一共炸了{fail}次。请按 Enter 键继续清除原文件...")
notification.notify(
title='Pyinstaller快速打包程序提醒您',
message=f'打包完成，一共炸了{fail}次。',
timeout=10
)
# 输出打包失败的文件
print("以下文件打包失败：")
for failed_file in failed_files:
print(failed_file)
else:
if not silent:
notification.notify(
title='Pyinstaller快速打包程序提醒您',
message=f'打包完成，没炸！',
timeout=10
)
input(f"打包完成，没炸！请按 Enter 键继续清除原文件...")
# 删除指定格式的文件
for root, dirs, files in os.walk(folder_path):
for file in files:
if file.endswith('.py') or file.endswith('.pyw'):
file_path = os.path.join(root, file)
countd += 1
os.remove(file_path)
print(f'{Fore.GREEN}✓{Fore.RESET} 已删除源文件: {file_path} (还剩 {acount-countd} 个源文件)')
if not silent:
notification.notify(
title='Pyinstaller快速打包程序提醒您',
message=f'文件删除完成！总共删除了{countd}个原文件',
timeout=10
)
print(f"{Fore.GREEN}文件删除完成！总共删除了 {Fore.BLUE}{countd}{Fore.RESET} 个原文件")
if not silent:
input("按 ENTER 键继续...")