# ------------------------------------------------------------------
|
# Copyright (c) 2020 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
|
# ------------------------------------------------------------------
|
|
"""
|
Hook for PyCryptodome library: https://pypi.python.org/pypi/pycryptodome
|
|
PyCryptodome is an almost drop-in replacement for the now unmaintained
|
PyCrypto library. The two are mutually exclusive as they live under
|
the same package ("Crypto").
|
|
PyCryptodome distributes dynamic libraries and builds them as if they were
|
Python C extensions (even though they are not extensions - as they can't be
|
imported by Python). It might sound a bit weird, but this decision is rooted
|
in PyPy and its partial and slow support for C extensions. However, this also
|
invalidates several of the existing methods used by PyInstaller to decide the
|
right files to pull in.
|
|
Even though this hook is meant to help with PyCryptodome only, it will be
|
triggered also when PyCrypto is installed, so it must be tested with both.
|
|
Tested with PyCryptodome 3.5.1, PyCrypto 2.6.1, Python 2.7 & 3.6, Fedora & Windows
|
"""
|
|
import os
|
import glob
|
|
from PyInstaller.compat import EXTENSION_SUFFIXES
|
from PyInstaller.utils.hooks import get_module_file_attribute
|
|
# Include the modules as binaries in a subfolder named like the package.
|
# Cryptodome's loader expects to find them inside the package directory for
|
# the main module. We cannot use hiddenimports because that would add the
|
# modules outside the package.
|
|
binaries = []
|
binary_module_names = [
|
'Crypto.Math', # First in the list
|
'Crypto.Cipher',
|
'Crypto.Util',
|
'Crypto.Hash',
|
'Crypto.Protocol',
|
'Crypto.PublicKey',
|
]
|
|
try:
|
for module_name in binary_module_names:
|
m_dir = os.path.dirname(get_module_file_attribute(module_name))
|
for ext in EXTENSION_SUFFIXES:
|
module_bin = glob.glob(os.path.join(m_dir, '_*%s' % ext))
|
for f in module_bin:
|
binaries.append((f, module_name.replace('.', os.sep)))
|
except ImportError:
|
# Do nothing for PyCrypto (Crypto.Math does not exist there)
|
pass
|
