diff options
Diffstat (limited to 'generate-download')
-rwxr-xr-x | generate-download | 39 |
1 files changed, 19 insertions, 20 deletions
diff --git a/generate-download b/generate-download index 4fb6ce004..f19729f59 100755 --- a/generate-download +++ b/generate-download @@ -1,34 +1,33 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 import hashlib -import os.path +import shutil import subprocess +import tempfile +import urllib.request -try: - from subprocess import check_output -except ImportError: # Python < 2.7 - def check_output(*args, **kwargs): - p = subprocess.Popen(*args, stdout=subprocess.PIPE, **kwargs) - out,err = p.communicate() - if p.returncode != 0: - raise subprocess.CalledProcessError(p.returncode, args) - return out +URL = 'https://github.com/downloads/rg3/youtube-dl/youtube-dl' -youtubeDlDir = os.path.join(os.path.dirname(__file__), '..', 'youtube-dl') +with tempfile.NamedTemporaryFile(suffix='youtube-dl', delete=True) as ytdl_file: + with urllib.request.urlopen(URL) as dl: + shutil.copyfileobj(dl, ytdl_file) -# Read template page -template = file('download.html.in', 'r').read() + ytdl_file.seek(0) + data = ytdl_file.read() + + ytdl_file.flush() + version = subprocess.check_output(['python3', ytdl_file.name, '--version']).decode('ascii').strip() -# Build replacement strings -version = check_output([os.path.join(youtubeDlDir, 'youtube-dl'), '--version']).strip() -data = check_output(['git', 'show', '%s:youtube-dl' % version], cwd=youtubeDlDir) +# Read template page +with open('download.html.in', 'r', encoding='utf-8') as tmplf: + template = tmplf.read() -url = 'https://github.com/rg3/youtube-dl/raw/%s/youtube-dl' % version md5sum = hashlib.md5(data).hexdigest() sha1sum = hashlib.sha1(data).hexdigest() sha256sum = hashlib.sha256(data).hexdigest() template = template.replace('@PROGRAM_VERSION@', version) -template = template.replace('@PROGRAM_URL@', url) +template = template.replace('@PROGRAM_URL@', URL) template = template.replace('@PROGRAM_MD5SUM@', md5sum) template = template.replace('@PROGRAM_SHA1SUM@', sha1sum) template = template.replace('@PROGRAM_SHA256SUM@', sha256sum) -file('download.html', 'w').write(template) +with open('download.html', 'w', encoding='utf-8') as dlf: + dlf.write(template) |