yt_dlp/__pyinstaller/hook-yt_dlp.py

   1 import sys
   2
   3 from PyInstaller.utils.hooks import collect_submodules
   4
   5
   6 def pycryptodome_module():
   7     try:
   8         import Cryptodome  # noqa: F401
   9     except ImportError:
  10         try:
  11             import Crypto  # noqa: F401
  12             print('WARNING: Using Crypto since Cryptodome is not available. '
  13                   'Install with: pip install pycryptodomex', file=sys.stderr)
  14             return 'Crypto'
  15         except ImportError:
  16             pass
  17     return 'Cryptodome'
  18
  19
  20 def get_hidden_imports():
  21     yield 'yt_dlp.compat._legacy'
  22     yield pycryptodome_module()
  23     yield from collect_submodules('websockets')
  24     # These are auto-detected, but explicitly add them just in case
  25     yield from ('mutagen', 'brotli', 'certifi')
  26
  27
  28 hiddenimports = list(get_hidden_imports())
  29 print(f'Adding imports: {hiddenimports}')
  30
  31 excludedimports = ['youtube_dl', 'youtube_dlc', 'test', 'ytdlp_plugins', 'devscripts']