Skip to content

Commit

Permalink
Add hook for fastparquet
Browse files Browse the repository at this point in the history
  • Loading branch information
rokm committed May 13, 2023
1 parent 1ec0eff commit 59092ef
Show file tree
Hide file tree
Showing 4 changed files with 41 additions and 0 deletions.
1 change: 1 addition & 0 deletions news/583.new.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Add hook for ``fastparquet``.
1 change: 1 addition & 0 deletions requirements-test-libraries.txt
Original file line number Diff line number Diff line change
Expand Up @@ -125,6 +125,7 @@ minecraft-launcher-lib==5.3; python_version >= '3.8'
scikit-learn==1.2.2; python_version >= '3.8'
scikit-image==0.20.0; python_version >= '3.8'
customtkinter==5.1.3
fastparquet==2023.4.0

# ------------------- Platform (OS) specifics

Expand Down
32 changes: 32 additions & 0 deletions src/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-fastparquet.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
# ------------------------------------------------------------------
# Copyright (c) 2023 PyInstaller Development Team.
#
# This file is distributed under the terms of the GNU General Public
# License (version 2.0 or later).
#
# The full license is available in LICENSE.GPL.txt, distributed with
# this software.
#
# SPDX-License-Identifier: GPL-2.0-or-later
# ------------------------------------------------------------------
import os

from PyInstaller.compat import is_win
from PyInstaller.utils.hooks import get_package_paths

# In all versions for which fastparquet provides Windows wheels (>= 0.7.0), delvewheel is used,
# so we need to collect the external site-packages/fastparquet.libs directory.
if is_win:
pkg_base, pkg_dir = get_package_paths("fastparquet")
lib_dir = os.path.join(pkg_base, "fastparquet.libs")
if os.path.isdir(lib_dir):
# We collect DLLs as data files instead of binaries to suppress binary
# analysis, which would result in duplicates (because it collects a copy
# into the top-level directory instead of preserving the original layout).
# In addition to DLls, this also collects .load-order* file (required on
# python < 3.8), and ensures that fastparquet.libs directory exists (required
# on python >= 3.8 due to os.add_dll_directory call).
datas = [
(os.path.join(lib_dir, lib_file), 'fastparquet.libs')
for lib_file in os.listdir(lib_dir)
]
7 changes: 7 additions & 0 deletions src/_pyinstaller_hooks_contrib/tests/test_libraries.py
Original file line number Diff line number Diff line change
Expand Up @@ -1559,3 +1559,10 @@ def test_customtkinter(pyi_builder):
pyi_builder.test_source("""
import customtkinter
""")


@importorskip('fastparquet')
def test_fastparquet(pyi_builder):
pyi_builder.test_source("""
import fastparquet
""")

0 comments on commit 59092ef

Please sign in to comment.