mirror of https://github.com/yt-dlp/yt-dlp
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
155 lines
4.9 KiB
155 lines
4.9 KiB
#!/usr/bin/env python3 |
|
import os |
|
import platform |
|
import sys |
|
|
|
from PyInstaller.__main__ import run as run_pyinstaller |
|
|
|
|
|
OS_NAME, ARCH = sys.platform, platform.architecture()[0][:2] |
|
|
|
|
|
def main(): |
|
opts = parse_options() |
|
version = read_version('yt_dlp/version.py') |
|
|
|
onedir = '--onedir' in opts or '-D' in opts |
|
if not onedir and '-F' not in opts and '--onefile' not in opts: |
|
opts.append('--onefile') |
|
|
|
name, final_file = exe(onedir) |
|
print(f'Building yt-dlp v{version} {ARCH}bit for {OS_NAME} with options {opts}') |
|
print('Remember to update the version using "devscripts/update-version.py"') |
|
if not os.path.isfile('yt_dlp/extractor/lazy_extractors.py'): |
|
print('WARNING: Building without lazy_extractors. Run ' |
|
'"devscripts/make_lazy_extractors.py" to build lazy extractors', file=sys.stderr) |
|
print(f'Destination: {final_file}\n') |
|
|
|
opts = [ |
|
f'--name={name}', |
|
'--icon=devscripts/logo.ico', |
|
'--upx-exclude=vcruntime140.dll', |
|
'--noconfirm', |
|
# NB: Modules that are only imported dynamically must be added here. |
|
# --collect-submodules may not work correctly if user has a yt-dlp installed via PIP |
|
'--hidden-import=yt_dlp.compat._legacy', |
|
*dependency_options(), |
|
*opts, |
|
'yt_dlp/__main__.py', |
|
] |
|
|
|
print(f'Running PyInstaller with {opts}') |
|
run_pyinstaller(opts) |
|
set_version_info(final_file, version) |
|
|
|
|
|
def parse_options(): |
|
# Compatability with older arguments |
|
opts = sys.argv[1:] |
|
if opts[0:1] in (['32'], ['64']): |
|
if ARCH != opts[0]: |
|
raise Exception(f'{opts[0]}bit executable cannot be built on a {ARCH}bit system') |
|
opts = opts[1:] |
|
return opts |
|
|
|
|
|
# Get the version from yt_dlp/version.py without importing the package |
|
def read_version(fname): |
|
with open(fname, encoding='utf-8') as f: |
|
exec(compile(f.read(), fname, 'exec')) |
|
return locals()['__version__'] |
|
|
|
|
|
def exe(onedir): |
|
"""@returns (name, path)""" |
|
name = '_'.join(filter(None, ( |
|
'yt-dlp', |
|
OS_NAME == 'darwin' and 'macos', |
|
ARCH == '32' and 'x86' |
|
))) |
|
return name, ''.join(filter(None, ( |
|
'dist/', |
|
onedir and f'{name}/', |
|
name, |
|
OS_NAME == 'win32' and '.exe' |
|
))) |
|
|
|
|
|
def version_to_list(version): |
|
version_list = version.split('.') |
|
return list(map(int, version_list)) + [0] * (4 - len(version_list)) |
|
|
|
|
|
def dependency_options(): |
|
# Due to the current implementation, these are auto-detected, but explicitly add them just in case |
|
dependencies = [pycryptodome_module(), 'mutagen', 'brotli', 'certifi', 'websockets'] |
|
excluded_modules = ['test', 'ytdlp_plugins', 'youtube_dl', 'youtube_dlc'] |
|
|
|
yield from (f'--hidden-import={module}' for module in dependencies) |
|
yield '--collect-submodules=websockets' |
|
yield from (f'--exclude-module={module}' for module in excluded_modules) |
|
|
|
|
|
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 set_version_info(exe, version): |
|
if OS_NAME == 'Windows': |
|
windows_set_version(exe, version) |
|
|
|
|
|
def windows_set_version(exe, version): |
|
from PyInstaller.utils.win32.versioninfo import ( |
|
FixedFileInfo, |
|
SetVersion, |
|
StringFileInfo, |
|
StringStruct, |
|
StringTable, |
|
VarFileInfo, |
|
VarStruct, |
|
VSVersionInfo, |
|
) |
|
|
|
version_list = version_to_list(version) |
|
suffix = '_x86' if ARCH == '32' else '' |
|
SetVersion(exe, VSVersionInfo( |
|
ffi=FixedFileInfo( |
|
filevers=version_list, |
|
prodvers=version_list, |
|
mask=0x3F, |
|
flags=0x0, |
|
OS=0x4, |
|
fileType=0x1, |
|
subtype=0x0, |
|
date=(0, 0), |
|
), |
|
kids=[ |
|
StringFileInfo([StringTable('040904B0', [ |
|
StringStruct('Comments', 'yt-dlp%s Command Line Interface.' % suffix), |
|
StringStruct('CompanyName', 'https://github.com/yt-dlp'), |
|
StringStruct('FileDescription', 'yt-dlp%s' % (' (32 Bit)' if ARCH == '32' else '')), |
|
StringStruct('FileVersion', version), |
|
StringStruct('InternalName', f'yt-dlp{suffix}'), |
|
StringStruct('LegalCopyright', 'pukkandan.ytdlp@gmail.com | UNLICENSE'), |
|
StringStruct('OriginalFilename', f'yt-dlp{suffix}.exe'), |
|
StringStruct('ProductName', f'yt-dlp{suffix}'), |
|
StringStruct( |
|
'ProductVersion', f'{version}{suffix} on Python {platform.python_version()}'), |
|
])]), VarFileInfo([VarStruct('Translation', [0, 1200])]) |
|
] |
|
)) |
|
|
|
|
|
if __name__ == '__main__': |
|
main()
|
|
|