pyinstaller: add package hooks

Author: pmrowla

pydrive2/__pyinstaller/__init__.py

@@ -0,0 +1,9 @@
+import os
+
+
+def get_hook_dirs():
+    return [os.path.dirname(__file__)]
+
+
+def get_PyInstaller_tests():
+    return [os.path.dirname(__file__)]

pydrive2/__pyinstaller/hook-googleapiclient.py

@@ -0,0 +1,9 @@
+from PyInstaller.utils.hooks import (  # pylint: disable=import-error
+    copy_metadata,
+    collect_data_files,
+)
+
+datas = copy_metadata("google-api-python-client")
+datas += collect_data_files(
+    "googleapiclient", excludes=["*.txt", "**/__pycache__"]
+)

pydrive2/__pyinstaller/test_hook-googleapiclient.py

@@ -0,0 +1,35 @@
+import subprocess
+
+from PyInstaller import __main__ as pyi_main
+
+
+_APP_SOURCE = """import importlib.resources
+
+import pydrive2.files
+
+
+cache_files = importlib.resources.contents(
+    "googleapiclient.discovery_cache.documents"
+)
+assert len(cache_files) > 0
+"""
+
+
+def test_pyi_hook_google_api_client(tmp_path):
+    app_name = "userapp"
+    workpath = tmp_path / "build"
+    distpath = tmp_path / "dist"
+    app = tmp_path / f"{app_name}.py"
+    app.write_text(_APP_SOURCE)
+    pyi_main.run(
+        [
+            "--workpath",
+            str(workpath),
+            "--distpath",
+            str(distpath),
+            "--specpath",
+            str(tmp_path),
+            str(app),
+        ],
+    )
+    subprocess.run([str(distpath / app_name / app_name)], check=True)

setup.py

@@ -8,6 +8,7 @@
     "flake8",
     "flake8-docstrings",
     "pytest-mock",
+    "pyinstaller",
 ]
 
 tests_requirements.append("black==22.10.0")
@@ -18,7 +19,12 @@
     author_email="[email protected]",
     maintainer="DVC team",
     maintainer_email="[email protected]",
-    packages=["pydrive2", "pydrive2.test", "pydrive2.fs"],
+    packages=[
+        "pydrive2",
+        "pydrive2.test",
+        "pydrive2.fs",
+        "pydrive2.__pyinstaller",
+    ],
     url="https://github.com/iterative/PyDrive2",
     project_urls={
         "Documentation": "https://docs.iterative.ai/PyDrive2",
@@ -52,4 +58,10 @@
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
     ],
+    entry_points={
+        "pyinstaller40": [
+            "hook-dirs = pydrive2.__pyinstaller:get_hook_dirs",
+            "tests = pydrive2.__pyinstaller:get_PyInstaller_tests",
+        ]
+    },
 )