如何将Python struct库格式字符串转换为对应允许的整数取值范围

问题描述 #

Python struct 库包含大量对应C语言类型的格式字符串（"h"对应int16，"H"对应uint16）。
有没有简单方法可以从格式字符串（如"h"、"H"等）得到其合法取值范围（如-32768到32767、0到65535等）？
我看到struct库提供了calcsize方法，但我实际需要的是类似calcrange的功能。
请问是否有内置解决方案，或是我忽略的优雅实现？也可以接受第三方库方案。
我自己写了一个简易的calcrange，但它仅覆盖了有限的格式字符串，还存在一些不具备通用性的假设，代码如下：

def calcrange(fmt: str) -> Tuple[int, int]:
    """Calculate the min and max possible value of a given struct format string."""
    size: int = calcsize(fmt)
    unsigned_max = int("0x" + "FF" * size, 16)
    if fmt.islower():
        # Signed case
        min_ = -1 * int("0x80" + "00" * (calcsize(fmt) - 1), 16)
        return min_, unsigned_max + min_
    # Unsigned case
    return 0, unsigned_max

Python 标准库struct没有内置的范围计算接口，以下是两种可行的实现方案：

方案1：原生标准库实现（无第三方依赖） #

你原有实现的主要局限性是未处理字节序前缀、特殊格式符（布尔、浮点、字符等），且用大小写判断符号的逻辑不够严谨。完善后的实现如下：

from struct import calcsize
from typing import Tuple, Union

def calcrange(fmt: str) -> Tuple[Union[int, float], Union[int, float]]:
    """计算给定struct格式字符串对应的取值范围
    支持整数、浮点数、布尔类型，自动忽略字节序前缀
    """
    # 移除开头的字节序/对齐标记
    fmt_char = fmt.lstrip('@=<>!')[-1]
    
    # 特殊格式符单独处理
    if fmt_char == '?': # 布尔类型
        return 0, 1
    if fmt_char == 'f': # 单精度浮点数
        return -3.4028234663852886e+38, 3.4028234663852886e+38
    if fmt_char == 'd': # 双精度浮点数
        return -1.7976931348623157e+308, 1.7976931348623157e+308
    
    size = calcsize(fmt)
    bits = size * 8
    unsigned_max = (1 << bits) - 1
    
    # 判断是否为有符号整数
    signed_formats = {'b', 'h', 'i', 'l', 'q'}
    if fmt_char in signed_formats:
        min_val = - (1 << (bits - 1))
        max_val = (1 << (bits - 1)) - 1
        return min_val, max_val
    else:
        return 0, unsigned_max

方案2：基于NumPy的极简实现（推荐） #

NumPy内置的数值类型完全匹配struct对应的C语言类型，可以直接调用其接口获取范围，无需手动计算，准确性更高：

import numpy as np
from struct import calcsize
from typing import Tuple, Union

# 格式符到numpy类型的映射表
FORMAT_TO_NUMPY = {
    'b': np.int8, 'B': np.uint8,
    'h': np.int16, 'H': np.uint16,
    'i': np.int32, 'I': np.uint32,
    'l': np.int32, 'L': np.uint32,
    'q': np.int64, 'Q': np.uint64,
    'f': np.float32, 'd': np.float64,
    '?': np.bool_
}

def calcrange(fmt: str) -> Tuple[Union[int, float], Union[int, float]]:
    fmt_char = fmt.lstrip('@=<>!')[-1]
    np_type = FORMAT_TO_NUMPY[fmt_char]
    if np.issubdtype(np_type, np.integer):
        info = np.iinfo(np_type)
    elif np.issubdtype(np_type, np.floating):
        info = np.finfo(np_type)
    elif np_type == np.bool_:
        return 0, 1
    return info.min, info.max

内容的提问来源于stack exchange，提问作者Intrastellar Explorer