如何确保Python类必须通过with语句（上下文管理器）调用？

你需要确保类必须通过with语句使用，直接实例化调用方法则触发异常，以下是几种实用方案，均能满足“降低用户误操作概率”的需求：

方案1：状态标记检查（你的初始思路优化） #

在__enter__中设置私有状态标记，业务方法统一检查该标记，未进入上下文则抛出异常：

class MyClass:
    def __init__(self, params):
        self.params = params
        self._entered = False

    def __enter__(self):
        self._entered = True
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self._entered = False
        # 这里可添加资源清理逻辑
        pass

    def do_things(self):
        if not self._entered:
            raise RuntimeError("MyClass必须在'with'语句中使用")
        # 实际业务逻辑
        print(f"使用参数 {self.params} 执行操作")

这种方案简单直接，用户误操作会立刻收到明确的错误提示。

方案2：代理对象隔离（更优雅的接口限制） #

让MyClass实例本身不暴露业务方法，仅在进入上下文后返回包含业务方法的代理对象，从接口层面直接阻止错误调用：

class MyClass:
    def __init__(self, params):
        self.params = params

    def __enter__(self):
        # 内部类封装所有业务方法
        class _MyClassProxy:
            def do_things(proxy_self):
                print(f"使用参数 {self.params} 执行操作")
            # 其他业务方法都定义在此
        return _MyClassProxy()

    def __exit__(self, exc_type, exc_val, exc_tb):
        pass

# 正确用法
with MyClass("test") as mc:
    mc.do_things()  # 正常执行

# 错误用法会直接报错
mc = MyClass("test")
mc.do_things()  # AttributeError: 'MyClass' object has no attribute 'do_things'

这种方案无需状态检查，用户直接实例化后根本找不到业务方法，从根源上降低误操作可能。

方案3：装饰器统一检查（多方法场景适配） #

如果类有多个业务方法，用装饰器封装检查逻辑，避免重复代码：

from functools import wraps

def require_context(func):
    @wraps(func)
    def wrapper(self, *args, **kwargs):
        if not getattr(self, "_entered", False):
            raise RuntimeError("MyClass必须在'with'语句中使用")
        return func(self, *args, **kwargs)
    return wrapper

class MyClass:
    def __init__(self, params):
        self.params = params
        self._entered = False

    def __enter__(self):
        self._entered = True
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self._entered = False

    @require_context
    def do_things(self):
        print(f"使用参数 {self.params} 执行操作")

    @require_context
    def another_method(self):
        print("另一个需要上下文的方法")

适合业务方法较多的场景，统一管理检查逻辑，代码更简洁。

方案选择建议 #

• 追求接口直观性：选方案2，直接从结构上限制错误调用；
• 需要保留实例本身的属性/方法：选方案1或方案3，灵活度更高。

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