-#!/usr/bin/env python
-# coding: utf-8
+#!/usr/bin/env python3
-from __future__ import unicode_literals
+# Allow direct execution
+import os
import sys
-# import os
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
import platform
-from PyInstaller.utils.win32.versioninfo import (
- VarStruct, VarFileInfo, StringStruct, StringTable,
- StringFileInfo, FixedFileInfo, VSVersionInfo, SetVersion,
-)
-import PyInstaller.__main__
-
-arch = sys.argv[1] if len(sys.argv) > 1 else platform.architecture()[0][:2]
-assert arch in ('32', '64')
-print('Building %sbit version' % arch)
-_x86 = '_x86' if arch == '32' else ''
-
-FILE_DESCRIPTION = 'Media Downloader%s' % (' (32 Bit)' if _x86 else '')
-
-# root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
-# print('Changing working directory to %s' % root_dir)
-# os.chdir(root_dir)
-
-exec(compile(open('yt_dlp/version.py').read(), 'yt_dlp/version.py', 'exec'))
-VERSION = locals()['__version__']
-
-VERSION_LIST = VERSION.split('.')
-VERSION_LIST = list(map(int, VERSION_LIST)) + [0] * (4 - len(VERSION_LIST))
-
-print('Version: %s%s' % (VERSION, _x86))
-print('Remember to update the version using devscipts\\update-version.py')
-
-VERSION_FILE = 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.' % _x86),
- StringStruct('CompanyName', 'https://github.com/yt-dlp'),
- StringStruct('FileDescription', FILE_DESCRIPTION),
- StringStruct('FileVersion', VERSION),
- StringStruct('InternalName', 'yt-dlp%s' % _x86),
- StringStruct(
- 'LegalCopyright',
- 'pukkandan.ytdlp@gmail.com | UNLICENSE',
- ),
- StringStruct('OriginalFilename', 'yt-dlp%s.exe' % _x86),
- StringStruct('ProductName', 'yt-dlp%s' % _x86),
- StringStruct('ProductVersion', '%s%s' % (VERSION, _x86)),
- ])]),
- VarFileInfo([VarStruct('Translation', [0, 1200])])
+from PyInstaller.__main__ import run as run_pyinstaller
+
+from devscripts.utils import read_version
+
+OS_NAME, MACHINE, ARCH = sys.platform, platform.machine().lower(), platform.architecture()[0][:2]
+if MACHINE in ('x86', 'x86_64', 'amd64', 'i386', 'i686'):
+ MACHINE = 'x86' if ARCH == '32' else ''
+
+
+def main():
+ opts, version = parse_options(), read_version()
+
+ 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} for {OS_NAME} {platform.machine()} 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',
+ '--additional-hooks-dir=yt_dlp/__pyinstaller',
+ *opts,
+ 'yt_dlp/__main__.py',
]
-)
-
-PyInstaller.__main__.run([
- '--name=yt-dlp%s' % _x86,
- '--onefile',
- '--icon=devscripts/cloud.ico',
- '--exclude-module=youtube_dl',
- '--exclude-module=test',
- '--exclude-module=ytdlp_plugins',
- '--hidden-import=mutagen',
- '--hidden-import=Crypto',
- '--upx-exclude=vcruntime140.dll',
- 'yt_dlp/__main__.py',
-])
-SetVersion('dist/yt-dlp%s.exe' % _x86, VERSION_FILE)
+
+ print(f'Running PyInstaller with {opts}')
+ run_pyinstaller(opts)
+ set_version_info(final_file, version)
+
+
+def parse_options():
+ # Compatibility 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
+
+
+def exe(onedir):
+ """@returns (name, path)"""
+ name = '_'.join(filter(None, (
+ 'yt-dlp',
+ {'win32': '', 'darwin': 'macos'}.get(OS_NAME, OS_NAME),
+ MACHINE,
+ )))
+ 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 set_version_info(exe, version):
+ if OS_NAME == 'win32':
+ windows_set_version(exe, version)
+
+
+def windows_set_version(exe, version):
+ from PyInstaller.utils.win32.versioninfo import (
+ FixedFileInfo,
+ StringFileInfo,
+ StringStruct,
+ StringTable,
+ VarFileInfo,
+ VarStruct,
+ VSVersionInfo,
+ )
+
+ try:
+ from PyInstaller.utils.win32.versioninfo import SetVersion
+ except ImportError: # Pyinstaller >= 5.8
+ from PyInstaller.utils.win32.versioninfo import write_version_info_to_executable as SetVersion
+
+ version_list = version_to_list(version)
+ suffix = MACHINE and f'_{MACHINE}'
+ 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' % (MACHINE and f' ({MACHINE})')),
+ 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()