76 lines
1.7 KiB
Python
76 lines
1.7 KiB
Python
# -*- mode: python -*-
|
|
block_cipher = None
|
|
|
|
from glob import iglob
|
|
from os.path import basename, dirname, abspath
|
|
import os
|
|
import fnmatch
|
|
|
|
script_name = '%%SCRIPT%%'
|
|
train_libs = %%TRAIN_LIBS%%
|
|
strip = True
|
|
site_packages = '.venv/lib/python3.6/site-packages/'
|
|
hidden_imports = ['prettyparse', 'speechpy']
|
|
binaries = []
|
|
|
|
|
|
def recursive_glob(treeroot, pattern):
|
|
results = []
|
|
for base, dirs, files in os.walk(treeroot):
|
|
goodfiles = fnmatch.filter(files, pattern)
|
|
results.extend(os.path.join(base, f) for f in goodfiles)
|
|
return results
|
|
|
|
|
|
if train_libs:
|
|
binaries = [
|
|
(abspath(i), dirname(i.replace(site_packages, '')))
|
|
for i in recursive_glob(site_packages + "tensorflow/", "*.so")
|
|
]
|
|
hidden_imports += ['h5py']
|
|
|
|
a = Analysis(
|
|
[abspath('precise/scripts/{}.py'.format(script_name))],
|
|
pathex=['.'],
|
|
binaries=binaries,
|
|
datas=[],
|
|
hiddenimports=hidden_imports,
|
|
hookspath=[],
|
|
runtime_hooks=[],
|
|
excludes=['PySide', 'PyQt4', 'PyQt5', 'matplotlib'],
|
|
win_no_prefer_redirects=False,
|
|
win_private_assemblies=False,
|
|
cipher=block_cipher
|
|
)
|
|
|
|
for i in range(len(a.binaries)):
|
|
dest, origin, kind = a.binaries[i]
|
|
if '_pywrap_tensorflow_internal' in dest:
|
|
a.binaries[i] = ('tensorflow.python.' + dest, origin, kind)
|
|
|
|
pyz = PYZ(
|
|
a.pure, a.zipped_data,
|
|
cipher=block_cipher
|
|
)
|
|
|
|
exe = EXE(
|
|
pyz,
|
|
a.scripts,
|
|
exclude_binaries=True,
|
|
name='precise-{}'.format(script_name.replace('_', '-')),
|
|
debug=False,
|
|
strip=strip,
|
|
upx=True,
|
|
console=True
|
|
)
|
|
|
|
coll = COLLECT(
|
|
exe,
|
|
a.binaries,
|
|
a.zipfiles,
|
|
a.datas,
|
|
strip=strip,
|
|
upx=True,
|
|
name='precise-{}'.format(script_name.replace('_', '-'))
|
|
)
|