_get_content_range()函数：每个Python开发者都应知道的秘密工具

_get_content_range()函数是一个用于获取HTTP响应头中Content-Range字段的Python函数。该函数可以帮助开发者在处理分片下载或断点续传等功能时，准确获得服务器返回的数据范围。

在HTTP协议中，Content-Range字段用于指定服务器返回的数据范围。它的格式通常为"RangeStart-RangeEnd/TotalSize"，表示服务器返回的是从RangeStart到RangeEnd的数据片段，总共的数据大小为TotalSize。

_get_content_range()函数的实现如下：

def _get_content_range(response):
    if response.status_code == 206:  # Partial Content
        content_range = response.headers.get('Content-Range')
        if content_range:
            return content_range.split(' ')[1]
    return None

该函数接受一个HTTP响应作为参数，首先判断响应的状态码是否为206（Partial Content），表示服务器返回的是部分内容。如果是部分内容，则尝试获取Content-Range字段的值，并返回该值。如果没有Content-Range字段或响应状态码不是206，则返回None。

以下是一个使用_get_content_range()函数的例子：

import requests

def download_file(url, file_path, start_byte=0):
    headers = {'Range': f'bytes={start_byte}-'}
    response = requests.get(url, stream=True, headers=headers)

    content_range = _get_content_range(response)
    if content_range:
        print(f"Downloading range: {content_range}")
    else:
        print("Downloading full content")

    with open(file_path, 'wb') as file:
        for chunk in response.iter_content(chunk_size=1024):
            file.write(chunk)

    if response.status_code == 200:
        print("Download completed successfully")
    elif response.status_code == 206:
        print("Partial download completed successfully")
    else:
        print(f"Download failed with status code: {response.status_code}")


url = 'http://example.com/bigfile.zip'
file_path = 'bigfile.zip'

download_file(url, file_path, start_byte=1024*1024)  # 从第1MB的位置开始下载

在上述示例中，我们定义了一个download_file()函数，使用requests库发送HTTP GET请求获取数据。我们传递了一个额外的Range请求头，指定了从start_byte字节位置开始下载数据。

在下载之前，我们首先调用_get_content_range()函数获取Content-Range字段的值（如果有的话）。如果服务器支持分片下载，并返回了Content-Range字段，则打印出下载的范围；否则，打印出正在下载全部内容。

最后，我们将接收到的数据写入文件，并根据响应的状态码判断下载是否成功。

使用_get_content_range()函数可以帮助我们在处理分片下载或断点续传等功能时，更加准确地控制和管理数据范围。