是否存在模式可为ndb.Model自动设置属性拼接的计算式Key？

当然有办法实现！Google Cloud NDB虽然没有直接提供类似ComputedProperty来自动设置实体ID的机制，但我们可以通过重写模型的构造函数和前置存储钩子来完美解决你的需求——既自动生成基于bar和baz的ID，又能确保这两个属性不可修改，保证实体唯一性。

实现思路 #

1. 自动生成ID：在模型的构造函数中，当用户没有手动传入id参数时，自动用bar和baz的拼接值作为ID。
2. 锁定不可修改属性：通过重写_pre_put_hook（存储前的钩子函数），检查如果是已存在的实体，bar和baz是否被篡改；同时确保ID始终与bar-baz拼接值一致，防止手动传入错误ID。

完整示例代码 #

from google.cloud import ndb

class Foo(ndb.Model):
    bar: ndb.StringProperty(required=True)
    baz: ndb.StringProperty(required=True)
    # 其他可自由修改的属性示例
    description: ndb.TextProperty()

    def __init__(self, *args, **kwargs):
        # 自动生成ID：如果没有传入id，且bar和baz已提供
        if 'id' not in kwargs and 'bar' in kwargs and 'baz' in kwargs:
            kwargs['id'] = f"{kwargs['bar']}-{kwargs['baz']}"
        # 调用父类构造函数
        super().__init__(*args, **kwargs)

    def _pre_put_hook(self):
        # 检查ID是否与bar-baz拼接值一致，防止手动设置错误ID
        expected_id = f"{self.bar}-{self.baz}"
        if self.key.id() != expected_id:
            raise ValueError(f"实体ID必须为'{expected_id}'，当前ID不匹配bar='{self.bar}'和baz='{self.baz}'")
        
        # 如果是已存在的实体，检查bar和baz是否被修改
        if self.key is not None:
            existing_entity = self.key.get()
            if existing_entity:
                if existing_entity.bar != self.bar or existing_entity.baz != self.baz:
                    raise ValueError("bar和baz属性为写入后不可修改，禁止变更")

使用示例 #

现在你不需要手动传入ID，直接创建实体即可：

# 自动生成ID为"bar1-baz1"
foo = Foo(bar="bar1", baz="baz1", description="测试实体")
foo.put()

# 尝试修改bar会抛出错误
foo.bar = "bar2"
try:
    foo.put()
except ValueError as e:
    print(e)  # 输出：bar和baz属性为写入后不可修改，禁止变更

# 尝试手动传入错误ID也会抛出错误
invalid_foo = Foo(id="wrong-id", bar="bar1", baz="baz1")
try:
    invalid_foo.put()
except ValueError as e:
    print(e)  # 输出：实体ID必须为'bar1-baz1'，当前ID不匹配bar='bar1'和baz='baz1'

为什么这么做？ #

• 保证唯一性：因为NDB的实体Key是全局唯一的，用bar-baz作为ID，天然确保了同一个bar和baz组合只会存在一个实体。
• 简化开发：不需要手动拼接ID，构造实体时只需传入bar和baz即可。
• 强制约束：通过_pre_put_hook的检查，彻底阻止了对bar和baz的修改，符合你的需求。

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