使用Google Cloud Datastore批量上传实体时遇Batch must be in progress to put()错误

嘿，这个问题我之前也碰到过！你遇到的错误是因为Datastore的batch还没进入活跃状态就调用了put()方法。咱们来一步步解决它：

问题根源 #

创建batch实例后，必须先启动它的执行上下文，才能添加put、delete这类操作。你的代码里只初始化了batch，但没有开启它的"运行状态"，所以调用batch.put()时会报错。

两种正确的解决写法 #

方法1：使用with语句（推荐）

with语句会自动帮你管理batch的生命周期——自动启动、提交，遇到异常还会自动回滚，代码更简洁安全：

from google.cloud import datastore

client = datastore.Client()

# 用with块包裹batch操作，自动处理begin和commit
with client.batch() as batch:
    key = client.key('test', 'key1')
    entity = datastore.Entity(key, exclude_from_indexes=['attribute2'])
    entity.update({
        'attribute1': 'hello',
        'attribute2': 'hello again',
        'attribute3': 0.98,
    })
    batch.put(entity)
# 离开with块后，batch会自动完成提交

方法2：手动调用begin()和commit()

如果需要更精细地控制流程，可以手动启动和提交batch，记得加上异常处理：

from google.cloud import datastore

client = datastore.Client()
batch = client.batch()

try:
    batch.begin()  # 手动启动batch，进入活跃状态
    key = client.key('test', 'key1')
    entity = datastore.Entity(key, exclude_from_indexes=['attribute2'])
    entity.update({
        'attribute1': 'hello',
        'attribute2': 'hello again',
        'attribute3': 0.98,
    })
    batch.put(entity)
    batch.commit()  # 提交所有batch内的操作
except Exception as e:
    batch.rollback()  # 出错时回滚所有未提交的操作
    print(f"Batch操作失败: {e}")

额外注意点 #

• 一旦调用commit()或rollback()，当前的batch实例就失效了，不能再用来添加新操作，需要重新创建新的batch。
• 尽量优先用with语句，它能避免忘记提交或回滚的问题，减少出错概率。

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