[devscripts/pyinstaller] Analyze sub-modules of `Cryptodome`

Ref: https://github.com/yt-dlp/yt-dlp/issues/6185#issuecomment-1423523986
This commit is contained in:
pukkandan 2023-02-09 11:56:12 +05:30
parent 203a06f855
commit b85faf6ffb
No known key found for this signature in database
GPG Key ID: 7EEE9E1E817D0A39
1 changed files with 33 additions and 5 deletions

View File

@ -1,9 +1,31 @@
import ast
import os
import sys
from pathlib import Path
from PyInstaller.utils.hooks import collect_submodules
def _pycryptodome_module():
def find_attribute_accesses(node, name, path=()):
if isinstance(node, ast.Attribute):
path = [*path, node.attr]
if isinstance(node.value, ast.Name) and node.value.id == name:
yield path[::-1]
for child in ast.iter_child_nodes(node):
yield from find_attribute_accesses(child, name, path)
def collect_used_submodules(name, level):
for dirpath, _, filenames in os.walk(Path(__file__).parent.parent):
for filename in filenames:
if not filename.endswith('.py'):
continue
with open(Path(dirpath) / filename, encoding='utf8') as f:
for submodule in find_attribute_accesses(ast.parse(f.read()), name):
yield '.'.join(submodule[:level])
def pycryptodome_module():
try:
import Cryptodome # noqa: F401
except ImportError:
@ -17,13 +39,19 @@ def _pycryptodome_module():
return 'Cryptodome'
def _hidden_imports():
def get_hidden_imports():
yield 'yt_dlp.compat._legacy'
for m in [_pycryptodome_module(), 'websockets']:
yield from collect_submodules(m)
yield from collect_submodules('websockets')
crypto = pycryptodome_module()
for sm in set(collect_used_submodules('Cryptodome', 2)):
yield f'{crypto}.{sm}'
# These are auto-detected, but explicitly add them just in case
yield from ('mutagen', 'brotli', 'certifi')
hiddenimports = list(_hidden_imports())
hiddenimports = list(get_hidden_imports())
print(f'Adding imports: {hiddenimports}')
excludedimports = ['youtube_dl', 'youtube_dlc', 'test', 'ytdlp_plugins', 'devscripts']