Add PyInstaller hook

Author: bwoodsend

setup.py

@@ -52,5 +52,8 @@ def run(self):
                         "PyQt5" if sys.version_info.major >= 3 else "python_qt5",
                         "numpy-stl", "namegenerator", "PILLOW"],
       },
-      cmdclass={"build": Build}
+      cmdclass={"build": Build},
+      entry_points={
+          "pyinstaller40": ["hook-dirs = vtkplotlib.data:_get_hooks_dir"]
+      },
       ) # yapf:disable

vtkplotlib/__pyinstaller/hook-vtkplotlib.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# =============================================================================
+# Created on Tue Sep 08 23:19:14 2020
+#
+# @author: Brénainn
+#
+# hook-vtkplotlib.py
+# Copyright (C) 2019-2020  Brénainn Woodsend
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+# =============================================================================
+"""
+Hook for PyInstaller.
+"""
+
+from PyInstaller.utils.hooks import collect_data_files
+
+datas = []
+
+# These lines can be removed to skip specific unwanted data files.
+
+# Icon used by QtFigure2.add_screenshot_button().
+datas += collect_data_files("vtkplotlib", includes=["**/screenshot.png"])
+
+# Icons used by QtFigure2.add_preset_views().
+datas += collect_data_files("vtkplotlib", includes=["**/*.jpg"])
+
+# The rabbit STL. Only used for demonstrating.
+# datas += collect_data_files("vtkplotlib", includes=["**/rabbit/"])

vtkplotlib/data/__init__.py

@@ -42,11 +42,18 @@ def get_rabbit_stl():
 ICONS_FOLDER = DATA_FOLDER / "icons"
 ICONS = {i.stem: str(i) for i in ICONS_FOLDER.glob("*.jpg")}
 
+_HOOKS_DIR = DATA_FOLDER.with_name("__PyInstaller")
+
+
+def _get_hooks_dir():
+    return [str(_HOOKS_DIR)]
+
 
 def assert_ok():
     assert ICONS_FOLDER.is_dir()
     assert MODELS_FOLDER.is_dir()
     assert Path(get_rabbit_stl()).exists()
+    assert _HOOKS_DIR.exists()
 
 
 if __name__ == "__main__":