Python的dataclasses能否实现类似OCaml的参数化变体功能？

Python 3.10+ 版本的dataclasses配合内置的结构化模式匹配，完全可以实现和OCaml参数化变体（代数数据类型）相近的使用体验，只是Python本身是动态类型语言，静态类型校验需要依赖mypy之类的外置工具实现。

你提到的dataclasses基础用法示例如下：

import dataclasses


@dataclasses.dataclass
class User:
    name: str
    age: int = 0

if __name__ == "__main__":
    u = User("taro", 32)


    match u:
        case User(name, age):
            print(f"{name}, {age}")

你给出的OCaml参数化变体参考是一个泛型链表定义：

type 'a mtype =
  | Empty
  | Cons of 'a * 'a mtype
;;

let (musage: int mtype) = Cons (1, Cons (2, Empty));;
(* val musage : int mtype = Cons (1, Cons (2, Empty)) *)

具体实现 #

核心思路是定义一个公共的根类型，再为每个变体分支单独定义dataclass，需要泛型能力的话直接套Python内置的泛型语法即可：

from __future__ import annotations
from dataclasses import dataclass
from typing import Generic, TypeVar

T = TypeVar("T")

# 变体公共基类
class MType(Generic[T]):
    pass

# 空节点分支
@dataclass
class Empty(MType[T]):
    pass

# 链表节点分支
@dataclass
class Cons(MType[T]):
    head: T
    tail: MType[T]

if __name__ == "__main__":
    # 和OCaml示例完全等价的实例定义
    musage: MType[int] = Cons(1, Cons(2, Empty()))
    print(musage)
    # 输出：Cons(head=1, tail=Cons(head=2, tail=Empty()))

    # 模式匹配写法和OCaml的match逻辑一一对应
    match musage:
        case Empty():
            print("空链表")
        case Cons(first, Cons(second, Empty())):
            print(f"两元素链表，值为{first}、{second}")

使用说明 #

• 这套写法的运行时逻辑和OCaml变体完全一致，模式匹配的分支判断行为没有区别
• 写的类型标注符合PEP484规范，用mypy做静态检查时可以正确识别泛型参数，实现接近OCaml的类型安全校验
• 如果不需要静态类型检查，可以直接删掉Generic、TypeVar相关的泛型定义，只保留类继承关系即可，动态运行时不受任何影响
• 嫌定义基类麻烦的话，也可以直接用类型别名MType[T] = Empty | Cons[T]，效果完全相同。

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