devscripts/update-version.py

   1 #!/usr/bin/env python3
   2 import subprocess
   3 import sys
   4 from datetime import datetime
   5
   6 with open('yt_dlp/version.py') as f:
   7     exec(compile(f.read(), 'yt_dlp/version.py', 'exec'))
   8 old_version = locals()['__version__']
   9
  10 old_version_list = old_version.split('.')
  11
  12 old_ver = '.'.join(old_version_list[:3])
  13 old_rev = old_version_list[3] if len(old_version_list) > 3 else ''
  14
  15 ver = datetime.utcnow().strftime("%Y.%m.%d")
  16
  17 rev = (sys.argv[1:] or [''])[0]  # Use first argument, if present as revision number
  18 if not rev:
  19     rev = str(int(old_rev or 0) + 1) if old_ver == ver else ''
  20
  21 VERSION = '.'.join((ver, rev)) if rev else ver
  22
  23 try:
  24     sp = subprocess.Popen(['git', 'rev-parse', '--short', 'HEAD'], stdout=subprocess.PIPE)
  25     GIT_HEAD = sp.communicate()[0].decode().strip() or None
  26 except Exception:
  27     GIT_HEAD = None
  28
  29 VERSION_FILE = f'''\
  30 # Autogenerated by devscripts/update-version.py
  31
  32 __version__ = {VERSION!r}
  33
  34 RELEASE_GIT_HEAD = {GIT_HEAD!r}
  35 '''
  36
  37 with open('yt_dlp/version.py', 'wt') as f:
  38     f.write(VERSION_FILE)
  39
  40 print('::set-output name=ytdlp_version::' + VERSION)
  41 print(f'\nVersion = {VERSION}, Git HEAD = {GIT_HEAD}')