Python类型提示：非字面量对象限制的策略与最佳实践

from typing import Literal

def process_status(status: Literal["success", "failure", "pending"]):
    """根据传入的状态字面量执行操作。"""
    if status == "success":
        print("操作成功")
    else:
        print(f"操作状态: {status}")

process_status("success")
# process_status("unknown") # 类型检查器会报错，因为 "unknown" 不是字面量之一

from enum import Enum
import numpy as np
from typing import Callable, Any

# 定义一个枚举，其成员的值是具体的函数对象
class TrigFunction(Enum):
    SIN = np.sin
    COS = np.cos

    # 可以添加一个方法来获取名称，便于打印或日志
    def get_name(self) -> str:
        return self.name.lower() # 返回 'sin' 或 'cos'

def process_trig_function(func_choice: TrigFunction, value: float) -> float:
    """
    根据枚举选择的三角函数执行计算。
    """
    actual_function: Callable[[float], float] = func_choice.value
    print(f"正在使用 {func_choice.get_name()} 函数进行计算...")
    return actual_function(value)

# 示例调用
result_sin = process_trig_function(TrigFunction.SIN, np.pi / 2)
print(f"sin(pi/2) = {result_sin}")

result_cos = process_trig_function(TrigFunction.COS, np.pi)
print(f"cos(pi) = {result_cos}")

# 尝试传入非 TrigFunction 类型会导致类型检查器错误
# process_trig_function(np.sin, 0.0) # Type checker error: Expected TrigFunction, got function

from typing import Protocol, runtime_checkable
import numpy as np

@runtime_checkable
class MathOperation(Protocol):
    """
    定义一个数学操作的协议，要求实现 calculate 方法和 name 属性。
    """
    def calculate(self, x: float) -> float:
        ...
    @property
    def name(self) -> str:
        ...

class SinOperation:
    """封装 np.sin 操作的类。"""
    def calculate(self, x: float) -> float:
        return np.sin(x)
    @property
    def name(self) -> str:
        return "Sinus"

class CosOperation:
    """封装 np.cos 操作的类。"""
    def calculate(self, x: float) -> float:
        return np.cos(x)
    @property
    def name(self) -> str:
        return "Cosinus"

def perform_math_operation(op: MathOperation, value: float) -> float:
    """
    执行一个通用的数学操作。
    """
    print(f"执行 {op.name} 操作...")
    return op.