X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=youtube_dl%2Fupdate.py;h=d9d95d2a14049f5b05d3887f38a05ab711fd8397;hb=b12cf31bb1493ba461ab8663a63b7f42164617da;hp=e4a1aaa641b34ed8c6a33e6b5fb725fce6cdba01;hpb=032f2326268ae810343c5a083e048f361f7a2074;p=youtube-dl diff --git a/youtube_dl/update.py b/youtube_dl/update.py index e4a1aaa64..d9d95d2a1 100644 --- a/youtube_dl/update.py +++ b/youtube_dl/update.py @@ -28,21 +28,25 @@ def rsa_verify(message, signature, key): return expected == signature -def update_self(to_screen, verbose, opener): +def update_self(to_screen, verbose, opener, prefer_insecure=False): """Update the program file with the latest version from the repository""" - UPDATE_URL = "https://rg3.github.io/youtube-dl/update/" + UPDATE_URL = '//rg3.github.io/youtube-dl/update/' VERSION_URL = UPDATE_URL + 'LATEST_VERSION' JSON_URL = UPDATE_URL + 'versions.json' UPDATES_RSA_KEY = (0x9d60ee4d8f805312fdb15a62f87b95bd66177b91df176765d13514a0f1754bcd2057295c5b6f1d35daa6742c3ffc9a82d3e118861c207995a8031e151d863c9927e304576bc80692bc8e094896fcf11b66f3e29e04e3a71e9a11558558acea1840aec37fc396fb6b65dc81a1c4144e03bd1c011de62e3f1357b327d08426fe93, 65537) - if not isinstance(globals().get('__loader__'), zipimporter) and not hasattr(sys, "frozen"): + if not isinstance(globals().get('__loader__'), zipimporter) and not hasattr(sys, 'frozen'): to_screen('It looks like you installed youtube-dl with a package manager, pip, setup.py or a tarball. Please use that to update.') return + def guess_scheme(url, insecure=False): + return 'http%s:%s' % ('' if insecure is True else 's', url) + # Check if there is a new version try: - newversion = opener.open(VERSION_URL).read().decode('utf-8').strip() + newversion = opener.open(guess_scheme( + VERSION_URL, prefer_insecure)).read().decode('utf-8').strip() except Exception: if verbose: to_screen(encode_compat_str(traceback.format_exc())) @@ -54,7 +58,8 @@ def update_self(to_screen, verbose, opener): # Download and check versions info try: - versions_info = opener.open(JSON_URL).read().decode('utf-8') + versions_info = opener.open(guess_scheme( + JSON_URL, prefer_insecure)).read().decode('utf-8') versions_info = json.loads(versions_info) except Exception: if verbose: @@ -83,19 +88,16 @@ def update_self(to_screen, verbose, opener): print_notes(to_screen, versions_info['versions']) - filename = sys.argv[0] - # Py2EXE: Filename could be different - if hasattr(sys, "frozen") and not os.path.isfile(filename): - if os.path.isfile(filename + '.exe'): - filename += '.exe' + # sys.executable is set to the full pathname of the exe-file for py2exe + filename = sys.executable if hasattr(sys, 'frozen') else sys.argv[0] if not os.access(filename, os.W_OK): to_screen('ERROR: no write permissions on %s' % filename) return # Py2EXE - if hasattr(sys, "frozen"): - exe = os.path.abspath(filename) + if hasattr(sys, 'frozen'): + exe = filename directory = os.path.dirname(exe) if not os.access(directory, os.W_OK): to_screen('ERROR: no write permissions on %s' % directory)