Demomgr/setup_nuitka.py at master · Square789/Demomgr · GitHub

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
from itertools import product
import os
from pathlib import Path
import sys
import pkg_resources
import shutil
import subprocess

ASSUME_YES = os.getenv("NUITKA_ASSUME_YES") is not None


# files that the program still runs without (according to ProcessExplorer) and
# that are relatively large in size.
# lzma and bz2 do get imported when included, but no compression is ever performed,
# so it's probably fine to remove them.
PROBABLY_UNUSED_FILES = (
	"_asyncio.pyd",
	"_bz2.pyd",
	"_decimal.pyd",
	"_elementtree.pyd",
	"_hashlib.pyd",
	"_msi.pyd",
	"_multiprocessing.pyd",
	"_overlapped.pyd",
	"_lzma.pyd",
	"_sqlite3.pyd",
	"_ssl.pyd",
	"libcrypto-1_1.dll",
	"libssl-1_1.dll",
	"pyexpat.pyd",
	"sqlite3.dll",
)

PROBABLY_UNUSED_DIRS = (
	"lib2to3",
)

def copy_license(package_name, target_dir, license_suffix = "", license_names = None):
	license_text = None
	module = pkg_resources.get_distribution(package_name)
	names = (
		map("".join, product(("LICENSE", "LICENCE"), ("", ".txt")))
		if license_names is None else license_names
	)
	filename = None
	for _name in names:
		try:
			# Look, I know this var would persist after the loop anyways,
			# but it's mega clunky
			filename = _name
			license_text = module.get_metadata(filename)
			break
		except FileNotFoundError:
			pass
	else:
		raise FileNotFoundError(f"Could not find license for module {package_name!r}")
	with (target_dir / (filename + license_suffix)).open("w") as license_fp:
		license_fp.write(license_text)


cur_path = Path.cwd()
build_dir = cur_path / "nuitka_build"
dist_dir = build_dir / "demomgr.dist"
regex_dist_dir = dist_dir / "regex"
target = cur_path / "demomgr"

def main():
	if build_dir.exists():
		shutil.rmtree(build_dir)

	# Starting from Nuitka 2.6.2 (latest 2.6.9 at time of writing), the
	# `ui_themes` directory could not be loaded by the build anymore without
	# the two `--include-package` switches. The package was directly copied
	# into the build directory before.
	# It is worth noting that Nuitka does not include the `__init__.py` file,
	# which makes me uneasy as interpreted Python 3.8 usually fails when
	# importing such packages via `importlib.resources` functions.
	# But as long as it works.

	options = [sys.executable, "-m", "nuitka"]
	if ASSUME_YES:
		options.append("--assume-yes-for-downloads")
	options.extend((
		"--standalone",
		"--include-package=demomgr.ui_themes",
		"--include-package-data=demomgr.ui_themes",
		"--plugin-enable=tk-inter",
		f"--output-dir={build_dir}",
		"--remove-output",
		str(target),
	))

	nuitka_process = subprocess.run(options)
	if nuitka_process.returncode != 0:
		raise RuntimeError("Nuitka failed.")

	# Delete probably unused stuff
	for file in PROBABLY_UNUSED_FILES:
		if (to_delete := dist_dir / file).exists():
			to_delete.unlink()
	for dir in PROBABLY_UNUSED_DIRS:
		if (to_delete := dist_dir / dir).exists():
			shutil.rmtree(str(to_delete))

	# Copy licenses
	with (cur_path / "LICENSE").open("r") as license_fp:
		license = license_fp.read()
	with (dist_dir / "LICENSE").open("w") as license_fp:
		license_fp.write(license)

	copy_license("regex", regex_dist_dir)
	copy_license("vdf", dist_dir, "-ValvePython_vdf")
	copy_license("schema", dist_dir, "-keleshev_schema", ("LICENSE-MIT",))


if __name__ == "__main__":
	main()