Python2.7 AppEngine应用迁移至Python3.9时NDB ContextError问题

问题原因 #

你遇到的ContextError核心原因是NDB Query对象的延迟执行特性：
在get_books()函数中，你在client.context()上下文管理器内创建了Book.query()，但返回的是未执行的Query对象本身。当Django视图main尝试序列化这个对象时，上下文已经被管理器关闭，此时查询才开始执行，自然会因为缺少NDB上下文报错。

解决方案 #

必须在NDB上下文管理器内部完成查询执行，将Query对象转换为具体的结果集合，而非返回Query对象。

修正后的基础代码 #

from django.shortcuts import render, redirect
from django.http import HttpResponse  # 补充原代码缺失的导入
from google.cloud import ndb


class Book(ndb.Model):
    title = ndb.StringProperty()

client = ndb.Client()


def get_books():
    with client.context():
        # 在上下文内执行查询，获取实际结果列表
        books = Book.query().fetch()
        return books

def main(request):
    books = get_books()
    # 将结果转换为Django可返回的格式
    book_titles = "\n".join([book.title for book in books])
    return HttpResponse(book_titles)

进阶优化：用Django中间件统一管理NDB上下文 #

如果项目中有多处使用NDB，手动在每个函数里写with client.context()会很繁琐，可以用Django中间件全局注入上下文：

1. 创建中间件文件myapp/middleware.py：

from google.cloud import ndb

client = ndb.Client()

class NDBContextMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        with client.context():
            response = self.get_response(request)
        return response

2. 在settings.py的MIDDLEWARE列表中添加该中间件：

MIDDLEWARE = [
    # 保留原有中间件
    'myapp.middleware.NDBContextMiddleware',
]

之后所有视图都会自动处于NDB上下文内，无需再手动管理。

内容的提问来源于stack exchange，提问作者Nazih AIT BENSAID