python - Pylons 自定义中间件返回 404

Question

我在 pylons 应用程序中有以下代码作为中间件：

import testing.model as model
import re
from pylons.controllers.util import abort

class SubdomainCheckMiddleware(object):
    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        if 'subdomaincheck' in environ:
            return self.app(environ, start_response)

        p = re.compile('([a-z0-9\-]+)', re.IGNORECASE)
        subdomain = p.match(environ['HTTP_HOST']).group(0)
        query = "SELECT \"nspname\" FROM \"pg_namespace\" WHERE \"nspname\" = '%s';" % subdomain
        result = model.meta.Session.execute(query)
        for row in result:
            if row['nspname'] == subdomain:
                environ['subdomaincheck'] = 'done'
                return self.app(environ, start_response)

它的基本作用是检查给定子域中是否存在 postgresql 中的模式，但是如果模式不存在，我需要它返回 404 not found，我该怎么做？

Answer 1

在 WSGI 应用程序中返回基本的 404 非常简单：

if error:
    start_response("404 Not Found", [('Content-type', 'text/plain')])
    return ['Page not found']

如果您想要更精细的东西，您可以使用其他一些中间件来为您处理错误。它实际上看起来像 Pylons 带有一个StatusCodeRedirect中间件，如果你需要它可以帮助你。