Python TypedDict：定义互斥字段与复杂组合类型

from typing import Literal, TypedDict, Union

class _FileLocal(TypedDict):
    local_filepath: str

class _FileCloud(TypedDict):
    cloud_url: str

# 尝试定义一个包含互斥字段的联合类型
_FileCloudOrLocal = _FileLocal | _FileCloud # 这是一个类型别名，不是一个类

class _FileTextProcess(_FileCloudOrLocal): # 错误：TypedDict不能继承Union
    filetype: Literal['txt']

class _FileCSVProcess(_FileCloudOrLocal): # 错误：TypedDict不能继承Union
    filetype: Literal['csv']
    delimeter: str

# FileProcess = _FileTextProcess | _FileCSVProcess

from typing import Literal, TypedDict, Union

# 1. 定义基础的、互斥的TypedDict片段
class _FileLocal(TypedDict):
    """表示本地文件路径的数据结构片段"""
    local_filepath: str

class _FileCloud(TypedDict):
    """表示云端URL的数据结构片段"""
    cloud_url: str

# 2. 定义基础的、通用的TypedDict片段
class _FileTextProcess(TypedDict):
    """表示文本文件类型的数据结构片段"""
    filetype: Literal['txt']

class _FileCSVProcess(TypedDict):
    """表示CSV文件类型的数据结构片段，包含分隔符"""
    filetype: Literal['csv']
    delimeter: str

# 3. 创建组合TypedDict：通过多重继承将互斥片段和通用片段结合
# 组合 A: 本地文本文件
class LocalTextFile(
    _FileLocal,      # 来源：本地文件
    _FileTextProcess # 类型：文本文件
):
    """本地文本文件类型定义"""
    pass

# 组合 B: 本地CSV文件
class LocalCSVFile(
    _FileLocal,      # 来源：本地文件
    _FileCSVProcess  # 类型：CSV文件，包含delimeter
):
    """本地CSV文件类型定义"""
    pass

# 组合 C: 云端文本文件
class CloudTextFile(
    _FileCloud,      # 来源：云端文件
    _FileTextProcess # 类型：文本文件
):
    """云端文本文件类型定义"""
    pass

# 组合 D: 云端CSV文件
class CloudCSVFile(
    _FileCloud,      # 来源：云端文件
    _FileCSVProcess  # 类型：CSV文件，包含delimeter
):
    """云端CSV文件类型定义"""
    pass

# 4. 使用Union将所有组合类型联合起来，形成最终的类型
FileProcess = Union[
    LocalTextFile,
    LocalCSVFile,
    CloudTextFile,
    CloudCSVFile
]

# 示例数据与类型检查
def process_file_data(data: FileProcess):
    """
    处理文件数据，根据其类型和来源进行操作。
    """
    if 'local_filepath' in data:
        print(f"处理本地文件: {data['local_filepath']}")
        if data['filetype'] == 'csv':
            print(f"  - 文件类型: CSV, 分隔符: {data['delimeter']}")
        else: # data['filetype'] == 'txt'
            print(f"  - 文件类型: TXT")
    elif 'cloud_url' in data:
        print(f"处理云端文件: {data['cloud_url']}")
        if data['filetype'] == 'csv':
            print(f"  - 文件类型: CSV, 分隔符: {data['delimeter']}")
        else: # data['filetype'] == 'txt'
            print(f"  - 文件类型: TXT")
    else:
        # 理论上，由于类型定义，这里不会发生
        print("未知文件数据结构")

# 有效的示例
data1: FileProcess = {"cloud_url": "https://example.com/file.txt", "filetype": "txt"}
data2: FileProcess = {"local_filepath": "./file.csv", "filetype": "csv", "delimeter": ","}
data3: FileProcess = {"cloud_url": "https://example.com/data.csv", "filetype": "csv", "delimeter": ";"}

process_file_data(data1)
process_file_data(data2)
process_file_data(data3)

# 无效的示例 (类型检查器会报错)
# data_invalid1: FileProcess = {"cloud_url": "url", "local_filepath": "path", "filetype": "txt"} # 既有云端又有本地
# data_invalid2: FileProcess = {"local_filepath": "path", "filetype": "csv"} # CSV缺少delimeter