使用python-ndb实现Cloud Datastore事务性创建实体报错求助

错误原因 #

你遇到的TypeError: transactional_wrapper() missing 1 required positional argument: 'wrapped'是因为代码缩进错误：@ndb.transactional装饰器和内部_transaction函数没有正确嵌套在with client.context()的代码块内，导致装饰器无法正确绑定目标函数，引发参数缺失异常。

正确实现方式 #

以下是修正后的事务性创建Account和关联Profile实体的代码，确保任一实体创建失败则事务回滚，两者都不保存：

from google.cloud import ndb

# 定义Account和Profile模型（根据你的实际业务调整字段）
class Account(ndb.Model):
    email = ndb.StringProperty(required=True)
    created_at = ndb.DateTimeProperty(auto_now_add=True)

class Profile(ndb.Model):
    display_name = ndb.StringProperty()
    bio = ndb.TextProperty()

client = ndb.Client()

def create_new_account(email, display_name=None, bio=None):
    with client.context():
        @ndb.transactional(retries=3)  # 配置重试次数，处理Datastore并发冲突
        def _transaction():
            # 创建Account实体并持久化，获取其主键
            account = Account(email=email)
            account_key = account.put()
            
            # 以Account的主键作为父键，创建关联的Profile实体
            profile = Profile(parent=account_key, display_name=display_name, bio=bio)
            profile.put()
            
            return account, profile
        
        try:
            return _transaction()
        except Exception as e:
            print(f"创建Account和Profile失败: {str(e)}")
            raise

关键说明 #

• 缩进修正：事务装饰器和内部函数必须完全嵌套在client.context()代码块中，ndb事务依赖客户端上下文环境才能正常运行。
• 事务原子性：@ndb.transactional装饰器会自动维护事务边界，任何一步操作失败（如Datastore写入错误、并发锁冲突）都会触发回滚，两个实体都不会被持久化。
• 实体组关联：用Account的主键作为Profile的父键，确保两者属于同一实体组——这是Datastore事务的硬性要求，事务内操作的实体必须属于同一实体组。
• 并发重试：添加retries=3参数可自动重试因并发冲突导致的失败，提升创建操作的成功率。

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