Python递归类定义问题：文件系统建模报错及优化需求

我尝试用Python建模文件系统结构，该系统包含两种对象：

• File：始终拥有containing_folder属性
• Folder：可选拥有containing_folder属性

现有代码如下：

class Folder:
    def __init__(self, name: str, containing_folder: Folder):
        self.name = name
        self.containing_folder = containing_folder

class File:
    def __init__(self, name: str, size: int, containing_folder: Folder):
        self.name = name
        self.containing_folder = containing_folder

定义Folder类的__init__方法时出现报错：

NameError: name 'Folder' is not defined

我理解报错原因，但希望保留显式类型注解，不愿省略参数类型，现提出两个问题：

1. 如何定义该递归结构？
2. 如何让Folder类的containing_folder参数可选？

问题1：解决递归类型注解报错 #

Python在类定义过程中，类本身还未完成创建，直接用类名做类型注解会触发NameError。以下三种方法可以解决，且都能保留显式类型注解：

方法1：用字符串包裹类名（Python 3.7+兼容） #

直接把Folder写成字符串形式，类型检查工具（如mypy）完全能识别：

class Folder:
    def __init__(self, name: str, containing_folder: 'Folder'):
        self.name = name
        self.containing_folder = containing_folder

方法2：使用ForwardRef（Python 3.6+） #

从typing模块导入ForwardRef，提前声明类型引用：

from typing import ForwardRef

FolderType = ForwardRef('Folder')

class Folder:
    def __init__(self, name: str, containing_folder: FolderType):
        self.name = name
        self.containing_folder = containing_folder

方法3：启用annotations未来特性（Python 3.7+） #

导入from __future__ import annotations后，所有类型注解会自动被当作字符串处理，不用手动加引号：

from __future__ import annotations

class Folder:
    def __init__(self, name: str, containing_folder: Folder):
        self.name = name
        self.containing_folder = containing_folder

问题2：让Folder的containing_folder参数可选 #

给参数设置默认值None，同时用typing.Optional标注类型（表示参数可以是Folder实例或None）。结合递归类型的解决方案，最终代码如下：

from __future__ import annotations
from typing import Optional

class Folder:
    def __init__(self, name: str, containing_folder: Optional[Folder] = None):
        self.name = name
        self.containing_folder = containing_folder

class File:
    def __init__(self, name: str, size: int, containing_folder: Folder):
        self.name = name
        self.containing_folder = containing_folder

这样处理后：

• 根文件夹创建时可以不传入containing_folder，默认值为None，符合“可选”要求
• File的containing_folder没有默认值，保证始终被传入，满足“始终拥有”的规则

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