build_hooks.py

import os
from typing import TYPE_CHECKING

from whey import builder
from whey.mixin import BuilderMixin

if TYPE_CHECKING:
	from domdf_python_tools.paths import PathPlus


class GettextMixin:
	def build_messages(self: BuilderMixin):
		from babel.messages.mofile import write_mo
		from babel.messages.pofile import read_po

		locales = self.pkgdir / "locales"
		if self.verbose:
			print("  Building messages")

		for po in locales.glob("*/LC_MESSAGES/pm2hw.po"):
			with po.open("rt", encoding="UTF-8") as f:
				catalog = read_po(f, po.parts[-3], po.stem)

			mo = self.build_dir / po.relative_to(self.project_dir).with_suffix(".mo")
			mo.parent.maybe_make(parents=True)
			with mo.open("wb") as f:
				write_mo(f, catalog)
			self.report_written(mo)

			if self.verbose:
				print("    Wrote language file:", mo)


class PyInstallerBuilder(GettextMixin, builder.AbstractBuilder):
	@property
	def default_build_dir(self) -> "PathPlus":
		"""
		Provides a default for the ``build_dir`` argument.
		"""

		return self.project_dir / "build" / "exe"

	def write_spec(self):
		import PyInstaller.building.makespec
		name = self.config["name"]
		self.spec_files = [
			PyInstaller.building.makespec.main(
				["main.py"],
				name=n,
				excludes=["appdirs", "pm2hw_icons"],
				datas=[(str(self.build_dir / name / "locales"), "pm2hw/locales")],
				onefile=True,
				console=console,
			)
			for n, console in [(name, True), (f"{name}w", False)] # 
		]

	def create_exe(self):
		import PyInstaller.building.build_main

		for spec in self.spec_files:
			PyInstaller.building.build_main.main(
				None,
				spec,
				False,
				workpath=str(self.build_dir.absolute()),
				distpath=str(self.out_dir.absolute()),
			)

	def copy_additional_files(self) -> None:
		# Hack to copy to dist instead
		tmp = self.build_dir
		self.build_dir = self.out_dir
		super().copy_additional_files()
		self.build_dir = tmp

	def build_exe(self) -> str:
		"""
		Build the Windows EXE distribution.

		:return: The filename of the created EXE.
		"""

		self.clear_build_dir()

		self.build_messages()
		self.copy_additional_files()
		self.write_spec()

		return self.create_exe()

	build = build_exe


class SDistBuilder(GettextMixin, builder.SDistBuilder):
	def call_additional_hooks(self):
		self.build_messages()


class WheelBuilder(GettextMixin, builder.WheelBuilder):
	def call_additional_hooks(self):
		self.build_messages()