SpaceLife-Updater/venv/lib64/python3.12/site-packages/PyInstaller/hooks/hook-sqlalchemy.py

80 lines
3.0 KiB
Python
Raw Normal View History

2024-01-23 13:52:41 -05:00
#-----------------------------------------------------------------------------
# Copyright (c) 2005-2023, 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 re
import importlib.util
from PyInstaller import isolated
from PyInstaller.lib.modulegraph.modulegraph import SourceModule
from PyInstaller.utils.hooks import check_requirement, logger
# 'sqlalchemy.testing' causes bundling a lot of unnecessary modules.
excludedimports = ['sqlalchemy.testing']
# Include most common database bindings some database bindings are detected and include some are not. We should
# explicitly include database backends.
hiddenimports = ['pysqlite2', 'MySQLdb', 'psycopg2', 'sqlalchemy.ext.baked']
if check_requirement('sqlalchemy >= 1.4'):
hiddenimports.append("sqlalchemy.sql.default_comparator")
@isolated.decorate
def _get_dialect_modules(module_name):
import importlib
module = importlib.import_module(module_name)
return [f"{module_name}.{submodule_name}" for submodule_name in module.__all__]
# In SQLAlchemy >= 0.6, the "sqlalchemy.dialects" package provides dialects.
# In SQLAlchemy <= 0.5, the "sqlalchemy.databases" package provides dialects.
if check_requirement('sqlalchemy >= 0.6'):
hiddenimports += _get_dialect_modules("sqlalchemy.dialects")
else:
hiddenimports += _get_dialect_modules("sqlalchemy.databases")
def hook(hook_api):
"""
SQLAlchemy 0.9 introduced the decorator 'util.dependencies'. This decorator does imports. E.g.:
@util.dependencies("sqlalchemy.sql.schema")
This hook scans for included SQLAlchemy modules and then scans those modules for any util.dependencies and marks
those modules as hidden imports.
"""
if not check_requirement('sqlalchemy >= 0.9'):
return
# this parser is very simplistic but seems to catch all cases as of V1.1
depend_regex = re.compile(r'@util.dependencies\([\'"](.*?)[\'"]\)')
hidden_imports_set = set()
known_imports = set()
for node in hook_api.module_graph.iter_graph(start=hook_api.module):
if isinstance(node, SourceModule) and node.identifier.startswith('sqlalchemy.'):
known_imports.add(node.identifier)
# Read the source...
with open(node.filename, 'rb') as f:
source_code = f.read()
source_code = importlib.util.decode_source(source_code)
# ... and scan it
for match in depend_regex.findall(source_code):
hidden_imports_set.add(match)
hidden_imports_set -= known_imports
if len(hidden_imports_set):
logger.info(" Found %d sqlalchemy hidden imports", len(hidden_imports_set))
hook_api.add_imports(*list(hidden_imports_set))