Python高级类型提示：如何避免Mypy因类型不够具体报错？

你的问题核心是原函数的返回类型标注过于宽泛，Mypy无法识别exists和metadata之间的逻辑关联——当exists为True时，metadata必然包含有效字段。以下是几种推荐的解决方法：

方案1：使用TypedDict（静态类型检查友好，推荐） #

通过TypedDict定义两种明确的字典结构，分别对应文件存在和不存在的情况，让Mypy能通过类型窄化识别逻辑关联：

from typing import TypedDict, Literal, str
import os

# 定义文件存在时的元数据字段结构
class FileMeta(TypedDict):
    created_at: str
    # 可添加其他元数据字段，比如modified_at等

# 定义两种状态的返回结构
class FileExistsMetadata(TypedDict):
    exists: Literal[True]
    metadata: FileMeta

class FileNotExistsMetadata(TypedDict):
    exists: Literal[False]
    metadata: dict[str, str]

def get_metadata_from_file(filepath: str) -> FileExistsMetadata | FileNotExistsMetadata:
    '''Get metadata about a file if it exists'''
    if os.path.isfile(filepath):
        return {
            'exists': True,
            'metadata': {'created_at': '2024-05-20'}  # 实际业务中的元数据
        }
    else:
        return {
            'exists': False,
            'metadata': {}
        }

def get_creation_time(filepath: str) -> str | None:
    metadata = get_metadata_from_file(filepath)
    if metadata['exists']:
        # Mypy此时能识别metadata属于FileExistsMetadata，metadata['metadata']必然包含created_at
        return metadata['metadata']['created_at']
    else:
        return None

这种方式在静态检查阶段就能确保类型安全，同时保留字典的使用习惯。

方案2：使用类型断言（快速临时解决） #

如果不想修改原函数的类型定义，可以通过断言或类型转换告诉Mypy具体的类型信息，但这种方式无法在静态阶段保证created_at一定存在，运行时仍可能出现KeyError：

from typing import cast, Dict
import os

def get_metadata_from_file(filepath: str) -> dict[str, bool|dict[str, Any]]:
    '''Get metadata about a file if it exists'''
    answer = {}
    if os.path.isfile(filepath):
        answer['exists'] = True
        answer['metadata'] = {'created_at': '2024-05-20'}
    else:
        answer['exists'] = False
        answer['metadata'] = {}
    return answer

def get_creation_time(filepath: str) -> str | None:
    metadata = get_metadata_from_file(filepath)
    if metadata['exists']:
        # 强制转换metadata['metadata']为包含created_at的字典
        meta = cast(Dict[str, str], metadata['metadata'])
        # 或者用断言确保键存在
        assert 'created_at' in meta, "Created_at missing in file metadata"
        return meta['created_at']
    else:
        return None

方案3：使用dataclass（面向对象的清晰方案） #

用dataclass封装元数据状态，完全避免字典的类型歧义，代码可读性和可维护性更高：

from dataclasses import dataclass
from typing import Optional
import os

@dataclass
class FileMetadata:
    exists: bool
    created_at: Optional[str] = None

def get_metadata_from_file(filepath: str) -> FileMetadata:
    '''Get metadata about a file if it exists'''
    if os.path.isfile(filepath):
        return FileMetadata(exists=True, created_at='2024-05-20')
    else:
        return FileMetadata(exists=False)

def get_creation_time(filepath: str) -> str | None:
    metadata = get_metadata_from_file(filepath)
    return metadata.created_at if metadata.exists else None

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