setup.py

   1 #!/usr/bin/env python3
   2
   3 # Allow execution from anywhere
   4 import os
   5 import sys
   6
   7 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
   8
   9 import subprocess
  10 import warnings
  11
  12 try:
  13     from setuptools import Command, find_packages, setup
  14     setuptools_available = True
  15 except ImportError:
  16     from distutils.core import Command, setup
  17     setuptools_available = False
  18
  19 from devscripts.utils import read_file, read_version
  20
  21 VERSION = read_version()
  22
  23 DESCRIPTION = 'A youtube-dl fork with additional features and patches'
  24
  25 LONG_DESCRIPTION = '\n\n'.join((
  26     'Official repository: <https://github.com/yt-dlp/yt-dlp>',
  27     '**PS**: Some links in this document will not work since this is a copy of the README.md from Github',
  28     read_file('README.md')))
  29
  30 REQUIREMENTS = read_file('requirements.txt').splitlines()
  31
  32
  33 def packages():
  34     if setuptools_available:
  35         return find_packages(exclude=('youtube_dl', 'youtube_dlc', 'test', 'ytdlp_plugins', 'devscripts'))
  36
  37     return [
  38         'yt_dlp', 'yt_dlp.extractor', 'yt_dlp.downloader', 'yt_dlp.postprocessor', 'yt_dlp.compat',
  39     ]
  40
  41
  42 def py2exe_params():
  43     warnings.warn(
  44         'py2exe builds do not support pycryptodomex and needs VC++14 to run. '
  45         'It is recommended to run "pyinst.py" to build using pyinstaller instead')
  46
  47     return {
  48         'console': [{
  49             'script': './yt_dlp/__main__.py',
  50             'dest_base': 'yt-dlp',
  51             'icon_resources': [(1, 'devscripts/logo.ico')],
  52         }],
  53         'version_info': {
  54             'version': VERSION,
  55             'description': DESCRIPTION,
  56             'comments': LONG_DESCRIPTION.split('\n')[0],
  57             'product_name': 'yt-dlp',
  58             'product_version': VERSION,
  59         },
  60         'options': {
  61             'bundle_files': 0,
  62             'compressed': 1,
  63             'optimize': 2,
  64             'dist_dir': './dist',
  65             'excludes': ['Crypto', 'Cryptodome'],  # py2exe cannot import Crypto
  66             'dll_excludes': ['w9xpopen.exe', 'crypt32.dll'],
  67             # Modules that are only imported dynamically must be added here
  68             'includes': ['yt_dlp.compat._legacy'],
  69         },
  70         'zipfile': None,
  71     }
  72
  73
  74 def build_params():
  75     files_spec = [
  76         ('share/bash-completion/completions', ['completions/bash/yt-dlp']),
  77         ('share/zsh/site-functions', ['completions/zsh/_yt-dlp']),
  78         ('share/fish/vendor_completions.d', ['completions/fish/yt-dlp.fish']),
  79         ('share/doc/yt_dlp', ['README.txt']),
  80         ('share/man/man1', ['yt-dlp.1'])
  81     ]
  82     data_files = []
  83     for dirname, files in files_spec:
  84         resfiles = []
  85         for fn in files:
  86             if not os.path.exists(fn):
  87                 warnings.warn(f'Skipping file {fn} since it is not present. Try running " make pypi-files " first')
  88             else:
  89                 resfiles.append(fn)
  90         data_files.append((dirname, resfiles))
  91
  92     params = {'data_files': data_files}
  93
  94     if setuptools_available:
  95         params['entry_points'] = {
  96             'console_scripts': ['yt-dlp = yt_dlp:main'],
  97             'pyinstaller40': ['hook-dirs = yt_dlp.__pyinstaller:get_hook_dirs'],
  98         }
  99     else:
 100         params['scripts'] = ['yt-dlp']
 101     return params
 102
 103
 104 class build_lazy_extractors(Command):
 105     description = 'Build the extractor lazy loading module'
 106     user_options = []
 107
 108     def initialize_options(self):
 109         pass
 110
 111     def finalize_options(self):
 112         pass
 113
 114     def run(self):
 115         if self.dry_run:
 116             print('Skipping build of lazy extractors in dry run mode')
 117             return
 118         subprocess.run([sys.executable, 'devscripts/make_lazy_extractors.py'])
 119
 120
 121 def main():
 122     if sys.argv[1:2] == ['py2exe']:
 123         params = py2exe_params()
 124         try:
 125             from py2exe import freeze
 126         except ImportError:
 127             import py2exe  # noqa: F401
 128             warnings.warn('You are using an outdated version of py2exe. Support for this version will be removed in the future')
 129             params['console'][0].update(params.pop('version_info'))
 130             params['options'] = {'py2exe': params.pop('options')}
 131         else:
 132             return freeze(**params)
 133     else:
 134         params = build_params()
 135
 136     setup(
 137         name='yt-dlp',
 138         version=VERSION,
 139         maintainer='pukkandan',
 140         maintainer_email='pukkandan.ytdlp@gmail.com',
 141         description=DESCRIPTION,
 142         long_description=LONG_DESCRIPTION,
 143         long_description_content_type='text/markdown',
 144         url='https://github.com/yt-dlp/yt-dlp',
 145         packages=packages(),
 146         install_requires=REQUIREMENTS,
 147         python_requires='>=3.7',
 148         project_urls={
 149             'Documentation': 'https://github.com/yt-dlp/yt-dlp#readme',
 150             'Source': 'https://github.com/yt-dlp/yt-dlp',
 151             'Tracker': 'https://github.com/yt-dlp/yt-dlp/issues',
 152             'Funding': 'https://github.com/yt-dlp/yt-dlp/blob/master/Collaborators.md#collaborators',
 153         },
 154         classifiers=[
 155             'Topic :: Multimedia :: Video',
 156             'Development Status :: 5 - Production/Stable',
 157             'Environment :: Console',
 158             'Programming Language :: Python',
 159             'Programming Language :: Python :: 3.7',
 160             'Programming Language :: Python :: 3.8',
 161             'Programming Language :: Python :: 3.9',
 162             'Programming Language :: Python :: 3.10',
 163             'Programming Language :: Python :: 3.11',
 164             'Programming Language :: Python :: Implementation',
 165             'Programming Language :: Python :: Implementation :: CPython',
 166             'Programming Language :: Python :: Implementation :: PyPy',
 167             'License :: Public Domain',
 168             'Operating System :: OS Independent',
 169         ],
 170         cmdclass={'build_lazy_extractors': build_lazy_extractors},
 171         **params
 172     )
 173
 174
 175 main()