如何将Python函数参数类型限定为特定socket地址族常量？

方案1：使用Literal结合常量值（兼容多数场景） #

直接用Literal指定允许的socket常量，同时添加运行时检查确保输入合法。如果mypy报错，通常是因为这些常量被标注为socket.AddressFamily类型而非字面量，此时可升级mypy到0.910+版本（该版本开始支持枚举成员作为Literal参数），或者改用常量对应的整数值（可读性稍差但能快速解决报错）。

示例代码：

import socket
from typing import Literal

# 定义允许的地址家族类型别名
AllowedAddressFamily = Literal[socket.AF_INET, socket.AF_INET6, socket.AF_UNSPEC]

def my_function(family: AllowedAddressFamily) -> None:
    # 运行时校验，防止非预期值传入
    if family not in (socket.AF_INET, socket.AF_INET6, socket.AF_UNSPEC):
        raise ValueError(f"仅支持AF_INET、AF_INET6、AF_UNSPEC，当前传入: {family}")
    # 函数业务逻辑
    print(f"使用地址家族: {family}")

若mypy版本较低，可改用整数值替代：

AllowedAddressFamily = Literal[2, 10, 0]  # 对应AF_INET, AF_INET6, AF_UNSPEC的整数值

方案2：自定义枚举类型（兼顾类型安全与可读性） #

自定义枚举类封装允许的地址家族，既能解决类型标注问题，也能让文档工具（包括intersphinx）更好识别类型。

示例代码：

import socket
from enum import Enum
from typing import Union

class ValidAddressFamily(Enum):
    """允许使用的地址家族枚举"""
    AF_INET = socket.AF_INET
    AF_INET6 = socket.AF_INET6
    AF_UNSPEC = socket.AF_UNSPEC

# 允许传入枚举成员或原始socket常量
AllowedAFType = Union[ValidAddressFamily, Literal[socket.AF_INET, socket.AF_INET6, socket.AF_UNSPEC]]

def my_function(family: AllowedAFType) -> None:
    # 统一转换为枚举值处理
    if isinstance(family, ValidAddressFamily):
        family = family.value
    # 运行时校验
    if family not in (socket.AF_INET, socket.AF_INET6, socket.AF_UNSPEC):
        raise ValueError(f"仅支持AF_INET、AF_INET6、AF_UNSPEC，当前传入: {family}")
    # 函数业务逻辑
    print(f"使用地址家族: {family}")

方案3：使用TypeVar与Union（兼容旧版Python） #

如果使用Python 3.8及以下版本，Literal支持有限，可通过TypeVar结合Union限定类型：

import socket
from typing import TypeVar, Union

AF_INET = socket.AF_INET
AF_INET6 = socket.AF_INET6
AF_UNSPEC = socket.AF_UNSPEC

AllowedAF = TypeVar("AllowedAF", AF_INET, AF_INET6, AF_UNSPEC)

def my_function(family: AllowedAF) -> None:
    if family not in (AF_INET, AF_INET6, AF_UNSPEC):
        raise ValueError(f"仅支持AF_INET、AF_INET6、AF_UNSPEC，当前传入: {family}")
    # 函数业务逻辑
    pass

关于intersphinx识别问题 #

自定义类型别名（如AllowedAddressFamily）后，在函数文档字符串中直接引用该别名，intersphinx会识别到自定义类型，避免因socket.AddressFamily未文档化导致的链接失效问题。示例：

def my_function(family: AllowedAddressFamily) -> None:
    """
    处理指定地址家族的逻辑

    :param family: 地址家族，必须是AllowedAddressFamily类型中的值
    """
    pass

内容的提问来源于stack exchange，提问作者Remco Haszing