X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;ds=sidebyside;f=bin%2Fwget-download;h=a851395a6b36aa8d610c146663beea21712ba78c;hb=a9641b9731213a7d70247dcc46dcfc1c846df03b;hp=da7ea395a36077f75e16201eb4825532f3e377e9;hpb=bc64e79021fc792c18f18a989bc6079a679a7738;p=dotfiles.git diff --git a/bin/wget-download b/bin/wget-download index da7ea39..a851395 100755 --- a/bin/wget-download +++ b/bin/wget-download @@ -1,13 +1,17 @@ -#! /usr/bin/env python +#! /usr/bin/env python3 -from cgi import parse_qsl from getopt import getopt, GetoptError -import os, posixpath -import sys, urllib +from urllib.parse import urlsplit, parse_qsl, quote, quote_plus, urlencode +import os +import posixpath +import sys from m_lib.defenc import default_encoding + def usage(): - sys.exit('Usage: %s [-e|--encoding=encoding] [-n|--newwin|-t|--tab] URL' % sys.argv[0]) + sys.exit('Usage: %s [-e|--encoding=encoding] [-n|--newwin|-t|--tab] URL' + % sys.argv[0]) + try: options, arguments = getopt(sys.argv[1:], 'e:', ['encoding=']) @@ -27,43 +31,53 @@ if not encoding: encoding = default_encoding url = arguments[0] -protocol, request = urllib.splittype(url) -user, password, port = None, None, None -host, path = urllib.splithost(request) -if host: - user, host = urllib.splituser(host) - if user: - user, password = urllib.splitpasswd(user) - host, port = urllib.splitport(host) - if port: port = int(port) -path, tag = urllib.splittag(path) -path, query = urllib.splitquery(path) -path = urllib.unquote(path) + +split_results = urlsplit(url) +protocol, netloc, path, query, tag = split_results +user = split_results.username +password = split_results.password +host = split_results.hostname +port = split_results.port if query: - qlist = [] - for name, value in parse_qsl(query): - name = unicode(name, default_encoding).encode(encoding) - value = unicode(value, default_encoding).encode(encoding) - qlist.append((name, value)) + qlist = [] + for name, value in parse_qsl(query): + if isinstance(name, bytes): + name = name.decode(default_encoding) + value = value.decode(default_encoding) + name = name.encode(encoding) + value = value.encode(encoding) + qlist.append((name, value)) url = protocol + "://" if user: - url += urllib.quote(unicode(user, default_encoding).encode(encoding)) - if password: - url += ':' + urllib.quote(unicode(password, default_encoding).encode(encoding)) - url += '@' + if isinstance(user, bytes): + user = user.decode(default_encoding) + url += quote(user.encode(encoding)) + if password: + if isinstance(password, bytes): + password = password.decode(default_encoding) + url += ':' + quote(password.encode(encoding)) + url += '@' if host: - url += host.decode(encoding).encode('idna') - if port: - url += ':%d' % port + if isinstance(host, bytes): + host = host.decode(encoding) + url += host.encode('idna').decode('ascii') + if port: + url += ':%d' % port if path: - if protocol == "file": - url += urllib.quote(path) - else: - url += urllib.quote(unicode(path, default_encoding).encode(encoding)) -#if query: -# url += '?' + urllib.urlencode(qlist) + if protocol == "file": + url += quote(path) + else: + if isinstance(path, bytes): + path = path.decode(default_encoding) + url += quote(path.encode(encoding)) +if query: + url += '?' + urlencode(qlist) +if tag: + if isinstance(tag, bytes): + tag = tag.decode(default_encoding) + url += '#' + quote_plus(tag.encode(encoding)) name = posixpath.basename(path) os.system('exec wget-wrapper -O "%s" "%s"' % (name, url))