66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import usocket
|
|
|
|
def urlopen(url, data=None, method="GET"):
|
|
if data is not None and method == "GET":
|
|
method = "POST"
|
|
try:
|
|
proto, dummy, host, path = url.split("/", 3)
|
|
except ValueError:
|
|
proto, dummy, host = url.split("/", 2)
|
|
path = ""
|
|
if proto == "http:":
|
|
port = 80
|
|
elif proto == "https:":
|
|
import ussl
|
|
port = 443
|
|
else:
|
|
raise ValueError("Unsupported protocol: " + proto)
|
|
|
|
if ":" in host:
|
|
host, port = host.split(":", 1)
|
|
port = int(port)
|
|
|
|
ai = usocket.getaddrinfo(host, port, 0, usocket.SOCK_STREAM)
|
|
ai = ai[0]
|
|
|
|
s = usocket.socket(ai[0], ai[1], ai[2])
|
|
try:
|
|
s.connect(ai[-1])
|
|
if proto == "https:":
|
|
s = ussl.wrap_socket(s, server_hostname=host)
|
|
|
|
s.write(method)
|
|
s.write(b" /")
|
|
s.write(path)
|
|
s.write(b" HTTP/1.0\r\nHost: ")
|
|
s.write(host)
|
|
s.write(b"\r\n")
|
|
|
|
if data:
|
|
s.write(b"Content-Length: ")
|
|
s.write(str(len(data)))
|
|
s.write(b"\r\n")
|
|
s.write(b"\r\n")
|
|
if data:
|
|
s.write(data)
|
|
|
|
l = s.readline()
|
|
l = l.split(None, 2)
|
|
#print(l)
|
|
status = int(l[1])
|
|
while True:
|
|
l = s.readline()
|
|
if not l or l == b"\r\n":
|
|
break
|
|
#print(l)
|
|
if l.startswith(b"Transfer-Encoding:"):
|
|
if b"chunked" in l:
|
|
raise ValueError("Unsupported " + l)
|
|
elif l.startswith(b"Location:"):
|
|
raise NotImplementedError("Redirects not yet supported")
|
|
except OSError:
|
|
s.close()
|
|
raise
|
|
|
|
return s
|