+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:
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']