urllib.urequest: Add support for HTTPS.

pull/76/head
Paul Sokolovsky 2016-05-30 21:24:41 +03:00
rodzic a85b23058e
commit 880c6a7011
1 zmienionych plików z 10 dodań i 4 usunięć

Wyświetl plik

@ -8,10 +8,14 @@ def urlopen(url, data=None, method="GET"):
except ValueError: except ValueError:
proto, dummy, host = url.split("/", 2) proto, dummy, host = url.split("/", 2)
path = "" path = ""
if proto != "http:": if proto == "http:":
port = 80
elif proto == "https:":
import ussl
port = 443
else:
raise ValueError("Unsupported protocol: " + proto) raise ValueError("Unsupported protocol: " + proto)
port = 80
if ":" in host: if ":" in host:
host, port = host.split(":", 1) host, port = host.split(":", 1)
port = int(port) port = int(port)
@ -20,13 +24,15 @@ def urlopen(url, data=None, method="GET"):
addr = ai[0][4] addr = ai[0][4]
s = usocket.socket() s = usocket.socket()
s.connect(addr) s.connect(addr)
if proto == "https:":
s = ussl.wrap_socket(s)
if data: if data:
req = b"%s /%s HTTP/1.0\r\nHost: %s\r\nContent-Length: %d\r\n\r\n" % (method, path, host, len(data)) req = b"%s /%s HTTP/1.0\r\nHost: %s\r\nContent-Length: %d\r\n\r\n" % (method, path, host, len(data))
else: else:
req = b"%s /%s HTTP/1.0\r\nHost: %s\r\n\r\n" % (method, path, host) req = b"%s /%s HTTP/1.0\r\nHost: %s\r\n\r\n" % (method, path, host)
s.send(req) s.write(req)
if data: if data:
s.send(data) s.write(data)
l = s.readline() l = s.readline()
protover, status, msg = l.split(None, 2) protover, status, msg = l.split(None, 2)