devscripts/update-version.py

   1 #!/usr/bin/env python3
   2
   3 # Allow direct execution
   4 import os
   5 import sys
   6
   7 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
   8
   9
  10 import argparse
  11 import contextlib
  12 import subprocess
  13 import sys
  14 from datetime import datetime
  15
  16 from devscripts.utils import read_version, write_file
  17
  18
  19 def get_new_version(version, revision):
  20     if not version:
  21         version = datetime.utcnow().strftime('%Y.%m.%d')
  22
  23     if revision:
  24         assert revision.isdigit(), 'Revision must be a number'
  25     else:
  26         old_version = read_version().split('.')
  27         if version.split('.') == old_version[:3]:
  28             revision = str(int((old_version + [0])[3]) + 1)
  29
  30     return f'{version}.{revision}' if revision else version
  31
  32
  33 def get_git_head():
  34     with contextlib.suppress(Exception):
  35         return subprocess.check_output(['git', 'rev-parse', 'HEAD'], text=True).strip() or None
  36
  37
  38 VERSION_TEMPLATE = '''\
  39 # Autogenerated by devscripts/update-version.py
  40
  41 __version__ = {version!r}
  42
  43 RELEASE_GIT_HEAD = {git_head!r}
  44
  45 VARIANT = None
  46
  47 UPDATE_HINT = None
  48
  49 CHANNEL = {channel!r}
  50 '''
  51
  52 if __name__ == '__main__':
  53     parser = argparse.ArgumentParser(description='Update the version.py file')
  54     parser.add_argument(
  55         '-c', '--channel', choices=['stable', 'nightly'], default='stable',
  56         help='Select update channel (default: %(default)s)')
  57     parser.add_argument(
  58         '-o', '--output', default='yt_dlp/version.py',
  59         help='The output file to write to (default: %(default)s)')
  60     parser.add_argument(
  61         'version', nargs='?', default=None,
  62         help='A version or revision to use instead of generating one')
  63     args = parser.parse_args()
  64
  65     git_head = get_git_head()
  66     version = (
  67         args.version if args.version and '.' in args.version
  68         else get_new_version(None, args.version))
  69     write_file(args.output, VERSION_TEMPLATE.format(
  70         version=version, git_head=git_head, channel=args.channel))
  71
  72     print(f'version={version} ({args.channel}), head={git_head}')