1#!/usr/bin/env python3 2from __future__ import unicode_literals 3 4import json 5import os 6import re 7import sys 8 9sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) 10 11from yt_dlp.compat import compat_urllib_request 12 13 14# usage: python3 ./devscripts/update-formulae.py <path-to-formulae-rb> <version> 15# version can be either 0-aligned (yt-dlp version) or normalized (PyPl version) 16 17filename, version = sys.argv[1:] 18 19normalized_version = '.'.join(str(int(x)) for x in version.split('.')) 20 21pypi_release = json.loads(compat_urllib_request.urlopen( 22 'https://pypi.org/pypi/yt-dlp/%s/json' % normalized_version 23).read().decode('utf-8')) 24 25tarball_file = next(x for x in pypi_release['urls'] if x['filename'].endswith('.tar.gz')) 26 27sha256sum = tarball_file['digests']['sha256'] 28url = tarball_file['url'] 29 30with open(filename, 'r') as r: 31 formulae_text = r.read() 32 33formulae_text = re.sub(r'sha256 "[0-9a-f]*?"', 'sha256 "%s"' % sha256sum, formulae_text) 34formulae_text = re.sub(r'url "[^"]*?"', 'url "%s"' % url, formulae_text) 35 36with open(filename, 'w') as w: 37 w.write(formulae_text) 38