| | 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 sys |
| | 13 | from datetime import datetime, timezone |
| | 14 | |
| | 15 | from devscripts.utils import read_version, run_process, write_file |
| | 16 | |
| | 17 | |
| | 18 | def get_new_version(version, revision): |
| | 19 | if not version: |
| | 20 | version = datetime.now(timezone.utc).strftime('%Y.%m.%d') |
| | 21 | |
| | 22 | if revision: |
| | 23 | assert revision.isdecimal(), 'Revision must be a number' |
| | 24 | else: |
| | 25 | old_version = read_version().split('.') |
| | 26 | if version.split('.') == old_version[:3]: |
| | 27 | revision = str(int((old_version + [0])[3]) + 1) |
| | 28 | |
| | 29 | return f'{version}.{revision}' if revision else version |
| | 30 | |
| | 31 | |
| | 32 | def get_git_head(): |
| | 33 | with contextlib.suppress(Exception): |
| | 34 | return run_process('git', 'rev-parse', 'HEAD').stdout.strip() |
| | 35 | |
| | 36 | |
| | 37 | VERSION_TEMPLATE = '''\ |
| | 38 | # Autogenerated by devscripts/update-version.py |
| | 39 | |
| | 40 | __version__ = {version!r} |
| | 41 | |
| | 42 | RELEASE_GIT_HEAD = {git_head!r} |
| | 43 | |
| | 44 | VARIANT = None |
| | 45 | |
| | 46 | UPDATE_HINT = None |
| | 47 | |
| | 48 | CHANNEL = {channel!r} |
| | 49 | |
| | 50 | ORIGIN = {origin!r} |
| | 51 | |
| | 52 | _pkg_version = {package_version!r} |
| | 53 | ''' |
| | 54 | |
| | 55 | if __name__ == '__main__': |
| | 56 | parser = argparse.ArgumentParser(description='Update the version.py file') |
| | 57 | parser.add_argument( |
| | 58 | '-c', '--channel', default='stable', |
| | 59 | help='Select update channel (default: %(default)s)') |
| | 60 | parser.add_argument( |
| | 61 | '-r', '--origin', default='local', |
| | 62 | help='Select origin/repository (default: %(default)s)') |
| | 63 | parser.add_argument( |
| | 64 | '-s', '--suffix', default='', |
| | 65 | help='Add an alphanumeric suffix to the package version, e.g. "dev"') |
| | 66 | parser.add_argument( |
| | 67 | '-o', '--output', default='yt_dlp/version.py', |
| | 68 | help='The output file to write to (default: %(default)s)') |
| | 69 | parser.add_argument( |
| | 70 | 'version', nargs='?', default=None, |
| | 71 | help='A version or revision to use instead of generating one') |
| | 72 | args = parser.parse_args() |
| | 73 | |
| | 74 | git_head = get_git_head() |
| | 75 | version = ( |
| | 76 | args.version if args.version and '.' in args.version |
| | 77 | else get_new_version(None, args.version)) |
| | 78 | write_file(args.output, VERSION_TEMPLATE.format( |
| | 79 | version=version, git_head=git_head, channel=args.channel, origin=args.origin, |
| | 80 | package_version=f'{version}{args.suffix}')) |
| | 81 | |
| | 82 | print(f'version={version} ({args.channel}), head={git_head}') |
projects / yt-dlp.git / blame_incremental