# ----------------------------------------------------------------------------- # Copyright (c) 2005-2021, PyInstaller Development Team. # # Distributed under the terms of the GNU General Public License (version 2 # or later) with exception for distributing the bootloader. # # The full license is in the file COPYING.txt, distributed with this software. # # SPDX-License-Identifier: (GPL-2.0-or-later WITH Bootloader-exception) # ----------------------------------------------------------------------------- import io import os import re import struct from PyInstaller import log as logging from PyInstaller.archive.writers import SplashWriter from PyInstaller.building import splash_templates from PyInstaller.building.datastruct import TOC, Target from PyInstaller.building.utils import _check_guts_eq, _check_guts_toc, misc from PyInstaller.compat import is_darwin from PyInstaller.utils.hooks import exec_statement from PyInstaller.utils.hooks.tcl_tk import (TK_ROOTNAME, collect_tcl_tk_files, find_tcl_tk_shared_libs) try: from PIL import Image as PILImage except ImportError: PILImage = None logger = logging.getLogger(__name__) # These requirement files are checked against the current splash screen script. If you wish to modify the splash screen # and run into tcl errors/bad behavior, this is a good place to start and add components your implementation of the # splash screen might use. splash_requirements = [ # prepended tcl/tk binaries os.path.join(TK_ROOTNAME, "license.terms"), os.path.join(TK_ROOTNAME, "text.tcl"), os.path.join(TK_ROOTNAME, "tk.tcl"), # Used for customizable font os.path.join(TK_ROOTNAME, "ttk", "ttk.tcl"), os.path.join(TK_ROOTNAME, "ttk", "fonts.tcl"), os.path.join(TK_ROOTNAME, "ttk", "cursors.tcl"), os.path.join(TK_ROOTNAME, "ttk", "utils.tcl"), ] class Splash(Target): """ Bundles the required resources for the splash screen into a file, which will be included in the CArchive. A Splash has two outputs, one is itself and one is sored in splash.binaries. Both need to be passed to other build targets in order to enable the splash screen. """ typ = 'SPLASH' def __init__(self, image_file, binaries, datas, **kwargs): """ :param str image_file: A path-like object to the image to be used. Only the PNG file format is supported. .. note:: If a different file format is supplied and PIL (Pillow) is installed, the file will be converted automatically. .. note:: *Windows*: Due to the implementation, the color Magenta/ RGB(255, 0, 255) must not be used in the image or text. .. note:: If PIL (Pillow) is installed and the image is bigger than max_img_size, the image will be resized to fit into the specified area. :param TOC binaries: The TOC of binaries the Analysis build target found. This TOC includes all extensionmodules and their dependencies. This is required to figure out, if the users program uses tkinter. :param TOC datas: The TOC of data the Analysis build target found. This TOC includes all data-file dependencies of the modules. This is required to check if all splash screen requirements can be bundled. :keyword text_pos: An optional 2x integer tuple that represents the origin of the text on the splash screen image. The origin of the text is its lower left corner. A unit in the respective coordinate system is a pixel of the image, its origin lies in the top left corner of the image. This parameter also acts like a switch for the text feature. If omitted, no text will be displayed on the splash screen. This text will be used to show textual progress in onefile mode. :type text_pos: Tuple[int, int] :keyword text_size: The desired size of the font. If the size argument is a positive number, it is interpreted as a size in points. If size is a negative number, its absolute value is interpreted as a size in pixels. Default: ``12`` :type text_size: int :keyword text_font: An optional name of a font for the text. This font must be installed on the user system, otherwise the system default font is used. If this parameter is omitted, the default font is also used. :keyword text_color: An optional color for the text. Either RGB HTML notation or color names are supported. Default: black (Windows: Due to a implementation issue the color magenta/ rgb(255, 0, 255) is forbidden) :type text_color: str :keyword text_default: The default text which will be displayed before the extraction starts. Default: "Initializing" :type text_default: str :keyword full_tk: By default Splash bundles only the necessary files for the splash screen (some tk components). This options enables adding full tk and making it a requirement, meaning all tk files will be unpacked before the splash screen can be started. This is useful during development of the splash screen script. Default: ``False`` :type full_tk: bool :keyword minify_script: The splash screen is created by executing an Tcl/Tk script. This option enables minimizing the script, meaning removing all non essential parts from the script. Default: True :keyword rundir: The folder name in which tcl/tk will be extracted at runtime. There should be no matching folder in your application to avoid conflicts. Default: ``__splash`` :type rundir: str :keyword name: An optional alternative filename for the .res file. If not specified, a name is generated. :type name: str :keyword script_name: An optional alternative filename for the Tcl script, that will be generated. If not specified, a name is generated. :type script_name: str :keyword max_img_size: Maximum size of the splash screen image as a tuple. If the supplied image exceeds this limit, it will be resized to fit the maximum width (to keep the original aspect ratio). This option can be disabled by setting it to None. Default: (760, 480) :type max_img_size: Tuple[int, int] """ from ..config import CONF Target.__init__(self) # Splash screen is not supported on macOS. It operates in a secondary thread and macOS disallows UI operations # in any thread other than main. if is_darwin: raise SystemExit("Splash screen is not supported on macOS.") # Make image path relative to .spec file if not os.path.isabs(image_file): image_file = os.path.join(CONF['specpath'], image_file) image_file = os.path.normpath(image_file) if not os.path.exists(image_file): raise ValueError("Image file '%s' not found" % image_file) # Copy all arguments self.image_file = image_file self.full_tk = kwargs.get("full_tk", False) self.name = kwargs.get("name", None) self.script_name = kwargs.get("script_name", None) self.minify_script = kwargs.get("minify_script", True) self.rundir = kwargs.get("rundir", None) self.max_img_size = kwargs.get("max_img_size", (760, 480)) # text options self.text_pos = kwargs.get("text_pos", None) self.text_size = kwargs.get("text_size", 12) self.text_font = kwargs.get("text_font", "TkDefaultFont") self.text_color = kwargs.get("text_color", "black") self.text_default = kwargs.get("text_default", "Initializing") # Save the generated file separately so that it is not necessary to generate the data again and again root = os.path.splitext(self.tocfilename)[0] if self.name is None: self.name = root + '.res' if self.script_name is None: self.script_name = root + '_script.tcl' if self.rundir is None: self.rundir = self._find_rundir(binaries + datas) # Internal variables try: # Do not import _tkinter at the toplevel, because on some systems _tkinter will fail to load, since it is # not installed. This would cause a runtime error in PyInstaller, since this module is imported from # build_main.py, instead we just want to inform the user that the splash screen feature is not supported on # his platform import _tkinter self._tkinter_module = _tkinter self._tkinter_file = self._tkinter_module.__file__ except ModuleNotFoundError: raise SystemExit( "You platform does not support the splash screen feature, since tkinter is not installed. Please " "install tkinter and try again." ) # Calculated / analysed values self.uses_tkinter = self._uses_tkinter(binaries) self.script = self.generate_script() self.tcl_lib, self.tk_lib = find_tcl_tk_shared_libs(self._tkinter_file) if is_darwin: # Outdated Tcl/Tk 8.5 system framework is not supported. Depending on macOS version, the library path will # come up empty (hidden system libraries on Big Sur), or will be # [/System]/Library/Frameworks/Tcl.framework/Tcl if self.tcl_lib[1] is None or 'Library/Frameworks/Tcl.framework' in self.tcl_lib[1]: raise SystemExit("The splash screen feature does not support macOS system framework version of Tcl/Tk.") # Check if tcl/tk was found assert all(self.tcl_lib) assert all(self.tk_lib) logger.debug("Use Tcl Library from %s and Tk From %s" % (self.tcl_lib, self.tk_lib)) self.splash_requirements = set([self.tcl_lib[0], self.tk_lib[0]] + splash_requirements) logger.info("Collect tcl/tk binaries for the splash screen") tcltk_tree = collect_tcl_tk_files(self._tkinter_file) if self.full_tk: # The user wants a full copy of tk, so make all tk files a requirement. self.splash_requirements.update(toc[0] for toc in tcltk_tree) self.binaries = TOC() if not self.uses_tkinter: # The user's script does not use tkinter, so we need to provide a TOC of all necessary files add the shared # libraries to the binaries. self.binaries.append((self.tcl_lib[0], self.tcl_lib[1], 'BINARY')) self.binaries.append((self.tk_lib[0], self.tk_lib[1], 'BINARY')) # Only add the intersection of the required and the collected resources, or add all entries if full_tk is # true. self.binaries.extend(toc for toc in tcltk_tree if toc[0] in self.splash_requirements) # Check if all requirements were found. fnames = [toc[0] for toc in (binaries + datas + self.binaries)] def _filter(_item): if _item not in fnames: # Item is not bundled, so warn the user about it. This actually may happen on some tkinter installations # that are missing the license.terms file. logger.warning( "The local Tcl/Tk installation is missing the file %s. The behavior of the splash screen is " "therefore undefined and may be unsupported." % _item ) return False return True # Remove all files which were not found. self.splash_requirements = set(filter(_filter, self.splash_requirements)) # Test if the tcl/tk version is supported by the bootloader. self.test_tk_version() logger.debug("Splash Requirements: %s" % self.splash_requirements) self.__postinit__() _GUTS = ( # input parameters ('image_file', _check_guts_eq), ('name', _check_guts_eq), ('script_name', _check_guts_eq), ('text_pos', _check_guts_eq), ('text_size', _check_guts_eq), ('text_font', _check_guts_eq), ('text_color', _check_guts_eq), ('text_default', _check_guts_eq), ('full_tk', _check_guts_eq), ('minify_script', _check_guts_eq), ('rundir', _check_guts_eq), ('max_img_size', _check_guts_eq), # calculated/analysed values ('uses_tkinter', _check_guts_eq), ('script', _check_guts_eq), ('tcl_lib', _check_guts_eq), ('tk_lib', _check_guts_eq), ('splash_requirements', _check_guts_eq), ('binaries', _check_guts_toc), # internal value # Check if the tkinter installation changed. This is theoretically possible if someone uses two different python # installations of the same version. ('_tkinter_file', _check_guts_eq), ) def _check_guts(self, data, last_build): if Target._check_guts(self, data, last_build): return True # Check if the image has been modified. if misc.mtime(self.image_file) > last_build: logger.info("Building %s because file %s changed", self.tocbasename, self.image_file) return True return False def assemble(self): logger.info("Building Splash %s" % self.name) # Function to resize a given image to fit into the area defined by max_img_size. def _resize_image(_image, _orig_size): if PILImage: _w, _h = _orig_size _ratio_w = self.max_img_size[0] / _w if _ratio_w < 1: # Image width exceeds limit _h = int(_h * _ratio_w) _w = self.max_img_size[0] _ratio_h = self.max_img_size[1] / _h if _ratio_h < 1: # Image height exceeds limit _w = int(_w * _ratio_h) _h = self.max_img_size[1] # If a file is given it will be open if isinstance(_image, PILImage.Image): _img = _image else: _img = PILImage.open(_image) _img_resized = _img.resize((_w, _h)) # Save image into a stream _image_stream = io.BytesIO() _img_resized.save(_image_stream, format='PNG') _img.close() _img_resized.close() _image_data = _image_stream.getvalue() logger.info( "Resized image %s from dimensions %s to (%d, %d)" % (self.image_file, str(_orig_size), _w, _h) ) return _image_data else: raise ValueError( "The splash image dimensions (w: %d, h: %d) exceed max_img_size (w: %d, h:%d), but the image " "cannot be resized due to missing PIL.Image! Either install the Pillow package, adjust the " "max_img_size, or use an image of compatible dimensions." % (_orig_size[0], _orig_size[1], self.max_img_size[0], self.max_img_size[1]) ) # Open image file image_file = open(self.image_file, 'rb') # Check header of the file to identify it if image_file.read(8) == b'\x89PNG\r\n\x1a\n': # self.image_file is a PNG file image_file.seek(16) img_size = (struct.unpack("!I", image_file.read(4))[0], struct.unpack("!I", image_file.read(4))[0]) if img_size > self.max_img_size: # The image exceeds the maximum image size, so resize it image = _resize_image(self.image_file, img_size) else: image = os.path.abspath(self.image_file) elif PILImage: # Pillow is installed, meaning the image can be converted automatically img = PILImage.open(self.image_file, mode='r') if img.size > self.max_img_size: image = _resize_image(img, img.size) else: image_data = io.BytesIO() img.save(image_data, format='PNG') img.close() image = image_data.getvalue() logger.info("Converted image %s to PNG format" % self.image_file) else: raise ValueError( "The image %s needs to be converted to a PNG file, but PIL.Image is not available! Either install the " "Pillow package, or use a PNG image for you splash screen." % self.image_file ) image_file.close() SplashWriter( self.name, self.splash_requirements, self.tcl_lib[0], # tcl86t.dll self.tk_lib[0], # tk86t.dll TK_ROOTNAME, self.rundir, image, self.script ) def test_tk_version(self): tcl_version = float(self._tkinter_module.TCL_VERSION) tk_version = float(self._tkinter_module.TK_VERSION) # Test if tcl/tk version is supported if tcl_version < 8.6 or tk_version < 8.6: logger.warning( "The installed Tcl/Tk (%s/%s) version might not work with the splash screen feature of the bootloader. " "The bootloader is tested against Tcl/Tk 8.6" % (self._tkinter_module.TCL_VERSION, self._tkinter_module.TK_VERSION) ) # This should be impossible, since tcl/tk is released together with the same version number, but just in case if tcl_version != tk_version: logger.warning( "The installed version of Tcl (%s) and Tk (%s) do not match. PyInstaller is tested against matching " "versions" % (self._tkinter_module.TCL_VERSION, self._tkinter_module.TK_VERSION) ) # Test if tcl is threaded. # If the variable tcl_platform(threaded) exist, the tcl interpreter was compiled with thread support. threaded = bool(exec_statement( """ from tkinter import Tcl, TclError try: print(Tcl().getvar('tcl_platform(threaded)')) except TclError: pass """ )) # yapf: disable if not threaded: # This is a feature breaking problem, so exit. raise SystemExit( "The installed tcl version is not threaded. PyInstaller only supports the splash screen " "using threaded tcl." ) def generate_script(self): """ Generate the script for the splash screen. If minify_script is True, all unnecessary parts will be removed. """ d = {} if self.text_pos is not None: logger.debug("Add text support to splash screen") d.update({ 'pad_x': self.text_pos[0], 'pad_y': self.text_pos[1], 'color': self.text_color, 'font': self.text_font, 'font_size': self.text_size, 'default_text': self.text_default, }) script = splash_templates.build_script(text_options=d) if self.minify_script: # Remove any documentation, empty lines and unnecessary spaces script = '\n'.join( line for line in map(lambda l: l.strip(), script.splitlines()) if not line.startswith('#') # documentation and line # empty lines ) # Remove unnecessary spaces script = re.sub(' +', ' ', script) # Write script to disk, so that it is transparent to the user what script is executed. with open(self.script_name, "w") as script_file: script_file.write(script) return script @staticmethod def _uses_tkinter(binaries): # Test for _tkinter instead of tkinter, because a user might use a different wrapping library for tk. return '_tkinter' in binaries.filenames @staticmethod def _find_rundir(structure): # First try a name the user could understand, if one would find the directory. rundir = '__splash%s' candidate = rundir % "" counter = 0 # Run this loop as long as a folder exist named like rundir. In most cases __splash will be sufficient and this # loop wont enter. while any(e[0].startswith(candidate + os.sep) for e in structure): # just append to rundir a counter candidate = rundir % str(counter) counter += 1 # The SPLASH_DATA_HEADER structure limits the name to be 16 bytes at maximum. So if we exceed the limit # raise an error. This will never happen, since there are 10^8 different possibilities, but just in case. assert len(candidate) <= 16 return candidate