forked from ok-oldking/ok-wuthering-waves
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.spec
155 lines (118 loc) · 4.48 KB
/
main.spec
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
# -*- mode: pythzon ; coding: utf-8 -*-
from pathlib import Path
import importlib.util
block_cipher = None
def check_package_exists(package_name):
package_spec = importlib.util.find_spec(package_name)
return package_spec is not None
# Example usage:
if check_package_exists('rapidocr_openvino'):
print("rapidocr_openvino exists")
import rapidocr_openvino
package_name = 'rapidocr_openvino'
install_dir = Path(rapidocr_openvino.__file__).resolve().parent
onnx_paths = list(install_dir.rglob('*.onnx')) + list(install_dir.rglob('*.txt'))
yaml_paths = list(install_dir.rglob('*.yaml'))
onnx_add_data = [(str(v.parent), f'{package_name}/{v.parent.name}')
for v in onnx_paths]
yaml_add_data = []
for v in yaml_paths:
if package_name == v.parent.name:
yaml_add_data.append((str(v.parent / '*.yaml'), package_name))
else:
yaml_add_data.append(
(str(v.parent / '*.yaml'), f'{package_name}/{v.parent.name}'))
import openvino
package_name = 'openvino'
install_dir = Path(openvino.__file__).resolve().parent
openvino_dll_path = list(install_dir.rglob('openvino_intel_cpu_plugin.dll')) + list(install_dir.rglob('openvino_onnx_frontend.dll'))
# Modified list comprehension with a condition check
openvino_add_data = [(str(v), f'{package_name}/{v.parent.name}')
for v in openvino_dll_path]
print(f'openvino_add_data {openvino_add_data}')
add_data = list(set(yaml_add_data + onnx_add_data + openvino_add_data))
else:
add_data = []
excludes = ['FixTk', 'tcl', 'tk', '_tkinter', 'tkinter', 'Tkinter', 'resources', 'matplotlib','numpy.lib']
add_data.append(('icon.png', '.'))
import ok
ok_dir = Path(ok.__file__).resolve().parent
binaries = os.path.join(ok_dir, 'binaries', '*')
print(f'ok_dir {ok_dir}')
add_data.append((binaries, 'ok/binaries'))
def list_files(directory, prefix='', extensions=[]):
file_list = []
for root, dirs, files in os.walk(directory):
for filename in files:
# Check the file extension
_, ext = os.path.splitext(filename)
if extensions and ext not in extensions:
continue
# Create the full filepath by joining root with the filename
filepath = os.path.join(root, filename)
# Create the relative path for the file to be used in the spec datas
relative_path = os.path.relpath(filepath, prefix)
folder_path = os.path.dirname(relative_path)
# Append the tuple (full filepath, relative path) to the file list
file_list.append((filepath, folder_path))
return file_list
if os.path.exists('assets'):
root_folder = os.getcwd() # Get the current working directory
assets = list_files(os.path.join(root_folder, 'assets'), root_folder)
add_data += assets
if os.path.exists('i18n'):
root_folder = os.getcwd() # Get the current working directory
assets = list_files(os.path.join(root_folder, 'i18n'), root_folder, ['.mo'])
add_data += assets
print(f"add_data {add_data}")
a = Analysis(
['main.py'],
pathex=[],
binaries=[],
datas=add_data,
hiddenimports=[],
hookspath=[],
hooksconfig={},
runtime_hooks=[],
excludes=[],
cipher=block_cipher,
noarchive=False,
noconsole=True,
)
# List of patterns to exclude
exclude_patterns = ['opencv_videoio_ffmpeg', 'opengl32sw.dll', 'Qt6Quick.dll','Qt6Pdf.dll','Qt6Qml.dll','Qt6OpenGL.dll','Qt6Network.dll','Qt6QmlModels.dll','Qt6VirtualKeyboard.dll','QtNetwork.pyd'
,'openvino_pytorch_frontend.dll','openvino_tensorflow_frontend.dll','py_tensorflow_frontend.cp311-win_amd64.pyd','py_pytorch_frontend.cp311-win_amd64.pyd',
]
# Optimized list comprehension using any() with a generator expression
a.binaries = [x for x in a.binaries if not any(pattern in x[0] for pattern in exclude_patterns)]
print(f'a.binaries {a.binaries}')
pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)
from config import config
exe = EXE(
pyz,
a.scripts,
[],
exclude_binaries=True,
name='ok-ww',
icon='icon.png',
debug=False,
bootloader_ignore_signals=False,
strip=False,
upx=True,
console=False,
disable_windowed_traceback=False,
argv_emulation=False,
target_arch=None,
codesign_identity=None,
entitlements_file=None,
uac_admin=True,
)
coll = COLLECT(
exe,
a.binaries,
a.datas,
strip=False,
upx=True,
upx_exclude=[],
name='bundle',
)