如何实现Python类变量仅可在类内方法修改、类外无法直接赋值？

我定义了如下类，其中包含几个类变量以及一个辅助类方法，该方法会根据某些条件更新这些类变量的值：

import datetime
class Foo:
    _UPDATED_AT: datetime.datetime = datetime.datetime(2000, 1, 1)
    STATUS: bool = False

    @classmethod
    def bar(cls, some_cond):
        # updated the class variable from here only based on some condition
        if some_cond:
            cls.STATUS = True 
            cls._UPDATED_AT = datetime.datetime.now()

        return cls.STATUS

需求如下：

• 防止这些类变量在类外被直接修改（赋值）；
• 允许在类方法中更新这些类变量，语义上类似C++中的私有类变量。

我曾尝试用元类阻止属性赋值，但它同时限制了类内部的更新操作：

class MyMeta(type):
    def __setattr__(cls, key, value):
        raise AttributeError(f"Can't set {cls}.{key}")

我知道下划线开头的变量是Python约定的内部变量，但希望实现强制限制。请问该需求是否可实现？

可以实现，核心思路是通过元类结合调用栈检查，区分赋值操作是来自类内部还是外部。以下是两种可行方案：

方案一：基于调用栈帧的自动判断 #

利用inspect模块获取当前调用栈，判断赋值操作的发起者是否属于目标类的方法内部：

import datetime
import inspect

class MyMeta(type):
    def __setattr__(cls, key, value):
        # 获取上一级调用帧
        frame = inspect.currentframe().f_back
        try:
            caller_locals = frame.f_locals
            # 检查是否是类方法内部调用（存在cls参数且指向当前类）
            if 'cls' in caller_locals and caller_locals['cls'] is cls:
                super().__setattr__(key, value)
                return
            # 检查是否是实例方法内部调用（存在self参数且实例属于当前类）
            if 'self' in caller_locals and type(caller_locals['self']) is cls:
                super().__setattr__(key, value)
                return
            # 外部直接赋值则抛出错误
            raise AttributeError(f"无法从{cls.__name__}类外部修改类变量{key}")
        finally:
            # 手动删除帧引用，避免内存泄漏
            del frame

class Foo(metaclass=MyMeta):
    _UPDATED_AT: datetime.datetime = datetime.datetime(2000, 1, 1)
    STATUS: bool = False

    @classmethod
    def bar(cls, some_cond):
        if some_cond:
            cls.STATUS = True 
            cls._UPDATED_AT = datetime.datetime.now()
        return cls.STATUS

验证效果 #

• 类内部修改：正常执行

  Foo.bar(True)  # 返回True，_UPDATED_AT和STATUS成功更新
  

• 外部直接赋值：触发异常

  Foo.STATUS = False  # 抛出AttributeError: 无法从Foo类外部修改类变量STATUS
  

方案二：装饰器标记允许修改的方法 #

通过装饰器明确标记有权限修改类变量的方法，元类仅允许来自这些方法的赋值操作，更可控：

import datetime
import inspect

# 装饰器：标记允许修改类变量的方法
def allow_class_var_mod(func):
    func._allow_modify = True
    return func

class MyMeta(type):
    def __setattr__(cls, key, value):
        frame = inspect.currentframe().f_back
        try:
            # 遍历调用栈，找到实际发起调用的函数
            for frame_info in inspect.stack():
                if frame_info.function == '__setattr__':
                    continue
                caller_func = frame_info.frame.f_locals.get(frame_info.function)
                if hasattr(caller_func, '_allow_modify') and caller_func._allow_modify:
                    super().__setattr__(key, value)
                    return
            raise AttributeError(f"无法从{cls.__name__}类外部修改类变量{key}")
        finally:
            del frame

class Foo(metaclass=MyMeta):
    _UPDATED_AT: datetime.datetime = datetime.datetime(2000, 1, 1)
    STATUS: bool = False

    @classmethod
    @allow_class_var_mod
    def bar(cls, some_cond):
        if some_cond:
            cls.STATUS = True 
            cls._UPDATED_AT = datetime.datetime.now()
        return cls.STATUS

注意事项 #

• 调用栈检查存在一定性能损耗，对性能敏感的场景需谨慎使用；
• 这种限制并非绝对安全，熟练开发者可通过特定手段绕过，但足以满足绝大多数场景下的强制约束需求；
• Python的设计哲学是"约定优于强制"，下划线命名的变量已经是行业通用的内部变量标识，强制限制属于额外的约束手段。

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