「Python 编程」编码实现网络请求库中的 URL 解析器

1,499 阅读5分钟

摘要:怎么写出更短的代码并不是这次要讨论的话题。今天我们来研究一下:运行代码的计算机是如何找到目标服务器的?

相信各位 Python 开发者都用过 Requests 库,有些朋友还用过 WebSockets 库。这里回顾一下它们的基本用法,例如使用 Requests 库向目标网站发出 GET 请求:

import requests


url = "https://www.baidu.com"
resp = requests.get(url)
print(resp.status_code)  # output -> 200

使用起来非常简单,我们很轻松地向目标网站发出了请求并打印输出响应状态码。当然,你还可以把它缩短:

import requests


print(requests.get("https://www.baidu.com").status_code)  # output -> 200

怎么写出更短的代码并不是这次要讨论的话题。今天我们来研究一下:运行代码的计算机是如何找到目标服务器的?

显然,你的第一映象是 IP 地址和端口号。

没错,就是 IP 地址和端口号。

但你明明输入的是 URL 地址,怎么就 IP + 端口号呢?

URL 解析的原因

一下子你也回答不上来吧?

我们可以将上方代码的逻辑,即计算机向目标服务器发出请求并拿到响应信息的过程抽象成下图:

程序输入的是 https://www.baidu.com,但最终要解析出具体的 IP 地址和端口号才能访问,例如 183.232.231.172:443

网络交互实际上属于 Socket 编程的范畴,无论是 Requests 还是 WebSockets 库,最终都会通过 Socket 与目标网站的服务器进行交互。而 Socket 编程中并不能直接使用域名,而是采用 IP + 端口号这种形式进行寻址的。

假设你现在需要编写一个网络请求库,有可能是 HTTP 协议的,也有可能是 WebSocket 协议的。你要解决的第一个问题就是解析 URL,将网址转换成 IP + 端口号,甚至还需要分割出协议类型、资源路径以及是否采用更安全的传输方式等。

URL 解析格式

以 WebSocket 协议方面的客户端库为例,在双端确认连接之前有一个「握手」的过程,这个过程之前已经需要双端的 IP 和端口号等信息了。下面的代码描述了 WebSocket 发出「握手」请求之前,双端建立连接时需要用到的基本信息:

 # aiowebsocket
 reader, writer = await asyncio.open_connection(host=host, port=port, ssl=ssl)

也就是 hostportssl

大部分的 WebSocket 服务给出的都是域名,例如 wss://echo.websocket.org。「握手」时还会用到资源路径。

接下来,我们来尝试一下,如何将域名转换为 IP + 端口号和 is ssl 这样的格式。

代码实现 URL 解析

开始之前,我们先规划一下基本步骤:

然后确定要使用的标准库:解析 URL 当然要用到 urllib 库中的 url parse;解析 address 则需要用到 socket 库;为了方面取数据,可以尝试使用 collections 库中的 namedtuple。

首先引入这几个库:

# 崔庆才和韦世东邀请你关注公众号:进击的Coder
import socket
from collections import namedtuple
from urllib.parse import urlparse

然后定义输出结构,对应代码如下:

REMOTE = namedtuple('REMOTE', ['scheme', 'hostname', 'address', 'port', 'resource', 'ssl'])

然后定义一个方法,我们传入 URL,获得解析好的 REMOTE 对象。方法定义如下:

def parses(url: str) -> REMOTE:
    pass

待会我们在 pass 处编写属于该方法的代码。

最开始要解析 URL,获得 schemehostname,对应代码如下:

url = urlparse(url)

urlparse 方法会返回一个 ParseResult 对象,对象大体格式如下:

ParseResult(scheme='wss', netloc='echo.websocket.org', path='', params='', query='', fragment='')

有了 schemehostname 后,就可以得到 portis ssladdress。对应代码如下:

# 崔庆才和韦世东邀请你关注公众号:进击的Coder
scheme = url.scheme
address = url.hostname
port = url.port or (443 if scheme == 'wss' else 80)
ssl = True if scheme == 'wss' else False

WebSocket 协议中只有两种协议头:wswss。它们对应的端口分别是 80443,这里借助 scheme 的值进行判断即可得到答案。同理,也直接得到了 is ssl 答案。

拿到 hostname 后,调用 socket 库的 getbyhostname 方法就能够得到目标服务器的 IP 地址了。对应代码如下:

address = socket.gethostbyname(hostname)

至于资源路径,它早已存在于 ParseResult 对象中,直接取出即可:

resource = url.path

要注意的是,有些 URL 中还会携带请求正文(即参数和值)。所以这里需要取 query,并将其拼接到 resource 中:

if url.query:
    resource += '?' + url.query

至此,我们已经拿到了所需的所有数据。

现在将它们装在到 REMOTE 结构中,返回给调用方:

return REMOTE(scheme, hostname, address, port, resource, ssl)

此时,调用 parses 方法后就会拿到 REMOTE 结构,它的取值方式很舒服,用 . 符号取值即可。例如:

# 夜幕团队邀请你关注公众号:NightTeam
res = parses("ws://echo.websocket.org?sign=i9878")
print(res.address, res.port, res.resource)

代码运行结果如下:

174.129.224.73 
80 
?sign=i9878

这样,我们就完成了 URL 解析的代码编写。

小结

代码虽然不多,逻辑也并不复杂。但我们完整实现了网络请求库中的 URL 解析模块,这代表着完成了编写库的基石之一。

在这个过程当中,我们了解到双端通信的基本过程和要用到的信息。在编码中学会了如何将 urlparsesocketnamedtuple 结合到一起。

而且,你今天学到了 namedtuple 这个新姿势!

夜幕团队邀请你关注公众号:NightTeam

本篇文章由一文多发平台 ArtiPub 自动发布