Google App Engine PolyModel重写实例方法失效问题咨询

这是Google App Engine Datastore（尤其是旧版NDB/DB API）里非常常见的多态实体反序列化问题，我之前做GAE项目时也踩过一模一样的坑，咱们一步步拆解解决：

问题根源 #

GAE的Datastore会把实体的类继承层次存在_class属性里，用来在读取时实例化对应的类。但如果出现以下情况，就会导致反序列化出错：

1. 子类B是在父类A之后部署的，现有实体的_class只记录了A；
2. 类加载顺序不对，Datastore反序列化器先加载了父类A，找不到子类B的定义；
3. 旧版NDB/DB对多态的处理有bug，导致即使_class包含['A','B']，也优先实例化父类A。

你看到B().__class__.__name__返回'A'，说明读取出来的实体实际上是A的实例，自然会调用父类的call_me方法。

解决方案 #

1. 确保子类优先加载 #

GAE的实体反序列化器依赖类的注册顺序，子类必须在父类之前被导入。比如在项目的模型入口文件（比如models/__init__.py）里，调整导入顺序：

# 先导入子类B，再导入父类A
from .my_models import B
from .my_models import A

这样Datastore在读取实体时，能先识别到B的类定义，正确实例化子类对象。

2. 批量更新现有实体的_class属性 #

对于已经存在Datastore里的实体，它们的_class字段可能没有正确包含子类信息，或者顺序不对。你需要批量更新这些实体：

from google.appengine.ext.ndb import Query

# 分批处理，避免一次性加载过多实体超时
cursor = None
while True:
    entities, cursor, more = A.query().fetch_page(100, start_cursor=cursor)
    for entity in entities:
        # 用子类B特有的字段或属性判断是否属于B
        if hasattr(entity, 'b_specific_field'):
            # 将_class设置为['B', 'A']，确保优先识别子类
            entity._class = ['B', 'A']
            entity.put()
    if not more:
        break

注意：一定要用子类特有的属性判断，避免把纯A的实体错误改成B。

3. 验证NDB多态配置（如果使用NDB） #

确保你的模型类正确继承自ndb.Model，NDB默认支持多态，但如果是自定义了模型元类，可能会破坏这个特性：

# 正确的多态模型定义
from google.appengine.ext import ndb

class A(ndb.Model):
    def call_me(self):
        print("super class called")

class B(A):
    # 可以添加子类特有的字段
    b_specific_field = ndb.StringProperty()
    
    def call_me(self):
        print("sub class method called")

如果用的是旧版DB API（不是NDB），建议迁移到NDB，DB的多态支持非常有限，容易出问题。

4. 临时调试方案（不推荐长期使用） #

如果需要快速验证，可以在读取实体后手动修正类类型：

entity = B.get_by_id(your_entity_id)
if isinstance(entity, A) and not isinstance(entity, B):
    # 强制将实例类改为B
    entity.__class__ = B
entity.call_me()  # 现在会调用B的方法

这只是临时 workaround，最终还是要解决根源问题。

验证修复 #

修复后，重新读取实体，检查：

b_instance = B.get_by_id(your_entity_id)
print(b_instance.__class__.__name__)  # 应该返回'B'
b_instance.call_me()  # 应该输出"sub class method called"

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