如何在 django-piston 处理程序中为 JSON 消息设置 HTTP 标头？

Question

在活塞处理程序中，我需要返回 django.db.models.query.QuerySet 作为正确的 JSON 消息（反映底层模型和查询），同时还添加我自己的 HttpResponse 标头。到目前为止，我可以执行其中之一，但不能同时执行两者（并获得看起来正确的 JSON 响应）。

下面生成一个正确的 JSON 格式的响应，但没有添加 HttpResponse 标头（未显示）：

class PollHandlerSearch(BaseHandler):
    allowed_methods = ('POST')
    model = Poll
    fields = ('id', 'question', 'pub_date')
    KEYS =  ( 'question', 'start-date', 'end-date' )

    def create(self, request):
        post = Poll.objects.all()
        for skey in self.KEYS:
            if len(post) and request.POST.has_key(skey) and len(request.POST[skey]):
                if skey == self.KEYS[0]:
                        post = post.filter(question__icontains=request.POST[skey])
                elif skey == self.KEYS[1]:
                        post = post.filter(pub_date__gte=request.POST[skey])
                elif skey == self.KEYS[2]:
                        post = post.filter(pub_date__lte=request.POST[skey])
        return post

生成正确格式的 JSON 消息：

[
    {
        "pub_date": "2010-08-23 22:15:07", 
        "question": "What's up?", 
        "id": 1
    }
]

下面使用添加的标头实现 HttpResponse 并生成一个看起来像 JSON 的响应，但不是实际的响应预期或想要的，加上不反映 django 的“DateTimeAwareJSONEncoder”所做的任何事情（由活塞的 JSONEmitter 使用）。

class PollHandlerSearch(BaseHandler):
    allowed_methods = ('POST')
    model = Poll
    fields = ('id', 'question', 'pub_date')
    KEYS =  ( 'question', 'start-date', 'end-date' )

    def create(self, request):
        resp = HttpResponse()
        post = Poll.objects.all()
        for skey in self.KEYS:
            if len(post) and request.POST.has_key(skey) and len(request.POST[skey]):
                if skey == self.KEYS[0]:
                        post = post.filter(question__icontains=request.POST[skey])
                elif skey == self.KEYS[1]:
                        post = post.filter(pub_date__gte=request.POST[skey])
                elif skey == self.KEYS[2]:
                        post = post.filter(pub_date__lte=request.POST[skey])
        json_serializer = serializers.get_serializer("json")()
        json_serializer.serialize(post, ensure_ascii=False, indent=4, stream=resp)
        resp['MYHEADER'] = 'abc123'
        return resp

导致格式不正确的 JSONish 消息：

[
    {
        "pk": 1, 
        "model": "polls.poll", 
        "fields": {
            "pub_date": "2010-08-23 22:15:07", 
            "question": "What's up?"
        }
    }
]

这无疑是发生的，因为我正在执行自己的 JSON 序列化，绕过了活塞的 JSONEmitter，从而绕过了它正确呈现“post”的任何操作。

我一直在研究piston的emitters.py，但基本上无法理解它（我在OOP/Python/django/piston方面还很新）。如何让活塞传递格式正确的 JSON 消息，其中包含 HTTP 标头并补充了我提供的标头？

Answer 1

您可以子类化piston.resource.Resource，并在__call__ 方法中添加您想要的任何标头。
frompiston.resource import Resource

class MyResource(Resource):
    def __call__(self, request, *args, **kwargs):
        resp = super(MyResource, self).__call__(request, *args, **kwargs)
        resp['HEADER'] = "abc123"
        return resp

然后，在你的 urls.py 中：

def BasicResource(handler):
    return resource.MyResource(handler=handler, authentication=basic)

your_handler = BasicResource(YourHandlerClass)
another_handler = BasicResource(AnotherHandlerClass)