summaryrefslogtreecommitdiffstats
path: root/yt_dlp/__pyinstaller/hook-yt_dlp.py
diff options
context:
space:
mode:
Diffstat (limited to 'yt_dlp/__pyinstaller/hook-yt_dlp.py')
-rw-r--r--yt_dlp/__pyinstaller/hook-yt_dlp.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/yt_dlp/__pyinstaller/hook-yt_dlp.py b/yt_dlp/__pyinstaller/hook-yt_dlp.py
new file mode 100644
index 0000000..63dcdff
--- /dev/null
+++ b/yt_dlp/__pyinstaller/hook-yt_dlp.py
@@ -0,0 +1,31 @@
+import sys
+
+from PyInstaller.utils.hooks import collect_submodules
+
+
+def pycryptodome_module():
+ try:
+ import Cryptodome # noqa: F401
+ except ImportError:
+ try:
+ import Crypto # noqa: F401
+ print('WARNING: Using Crypto since Cryptodome is not available. '
+ 'Install with: pip install pycryptodomex', file=sys.stderr)
+ return 'Crypto'
+ except ImportError:
+ pass
+ return 'Cryptodome'
+
+
+def get_hidden_imports():
+ yield 'yt_dlp.compat._legacy'
+ yield pycryptodome_module()
+ yield from collect_submodules('websockets')
+ # These are auto-detected, but explicitly add them just in case
+ yield from ('mutagen', 'brotli', 'certifi')
+
+
+hiddenimports = list(get_hidden_imports())
+print(f'Adding imports: {hiddenimports}')
+
+excludedimports = ['youtube_dl', 'youtube_dlc', 'test', 'ytdlp_plugins', 'devscripts']