Compartilhamento de tecnologia

한어Русский языкEnglishFrançaisIndonesianSanskrit日本語DeutschPortuguêsΕλληνικάespañolItalianoSuomalainenLatina

Tutorial do servidor Python WSGI:start_response Análise de método

Neste artigo analisaremos detalhadamente um servidor WSGI start_response método.Este método é responsável por processar o código de status e os cabeçalhos de resposta da resposta HTTP e retornar umwrite Função usada para enviar dados de resposta. Explicaremos como o método funciona linha por linha e forneceremos algumas informações básicas para ajudar a compreender seus recursos.

conhecimento prévio

WSGI (Web Server Gateway Interface) é uma interface padrão usada para conectar um servidor web a uma aplicação ou estrutura web. Através do WSGI, você pode se comunicar entre o servidor e o aplicativo e lidar com solicitações e respostas HTTP.

Ao implementar um servidor WSGI,start_response O método é uma parte importante chamada pelo aplicativo WSGI para definir o status da resposta HTTP e os cabeçalhos de resposta e retorna uma função para gravar os dados de resposta.

start_response Implementação de método

O seguinte é um típico start_response Implementação do método:

def start_response(status, response_headers, exc_info=None):
    if exc_info:
        try:
            if headers_sent:
                reraise(*exc_info)
        finally:
            exc_info = None
    elif headers_set:
        raise AssertionError("Headers already set")
    headers_set[:] = [status, response_headers]
    return write
1
2
3
4
5
6
7
8
9
10
11

Explicação passo a passo

1. tratar exc_info parâmetro

if exc_info:
    try:
        if headers_sent:
            reraise(*exc_info)
    finally:
        exc_info = None
1
2
3
4
5
6

• exc_info O argumento geralmente contém uma tupla de exceção no formato(type, value, traceback) , usado para tratamento de exceções.seexc_info Não paraNone, indicando que há uma exceção que precisa ser tratada.
• try Verificações em blocoheaders_sent Se o cabeçalho de resposta foi enviado.Se os cabeçalhos de resposta foram enviados, chamereraise(*exc_info) Relançar a exceção.
• finally bloco garante que depois de tratar a exceção, oexc_info definido comoNonepara evitar duplicação de processamento.

2. Verifique se o cabeçalho de resposta foi definido

elif headers_set:
    raise AssertionError("Headers already set")
1
2

• se headers_set A lista já contém valores (ou seja, o cabeçalho da resposta foi definido), aumentaAssertionError anormal.Esta verificação garantestart_response Só pode ser chamado uma vez para definir cabeçalhos de resposta.

3. Definir status e cabeçalhos de resposta

headers_set[:] = [status, response_headers]
1

• Use atribuição de fatia para status eresponse_headers atribuído aheaders_set lista.O objetivo disso é reterheaders_set referência, mas atualiza seu conteúdo.
• status é uma string que representa o código e a mensagem do status da resposta HTTP, por exemplo"200 OK"。
• response_headers é uma lista contendo tuplas, cada tupla representa um par de valores-chave de um cabeçalho de resposta, por exemplo[("Content-Type", "text/html"), ("Content-Length", "123")]。

4. retornar write função

return write
1

• start_response O método retorna umwrite função. Esta função é chamada pelo aplicativo WSGI para enviar dados de resposta.

Exemplo de uso

Aqui está um exemplo completo mostrando como usar start_response Métodos para lidar com respostas WSGI:

import socketserver

class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        self.headers_set = []
        self.headers_sent = []

        def start_response(status, response_headers, exc_info=None):
            if exc_info:
                try:
                    if self.headers_sent:
                        raise exc_info[1]
                finally:
                    exc_info = None
            elif self.headers_set:
                raise AssertionError("Headers already set")
            self.headers_set[:] = [status, response_headers]
            return self.write

        def write(data):
            assert self.headers_set, "write() before start_response"
            if not self.headers_sent:
                status, response_headers = self.headers_sent[:] = self.headers_set
                try:
                    code, msg = status.split(None, 1)
                except ValueError:
                    code, msg = status, ""
                code = int(code)
                self.send_response(code, msg)
                header_keys = set()
                for key, value in response_headers:
                    self.send_header(key, value)
                    key = key.lower()
                    header_keys.add(key)
                if not (
                    "content-length" in header_keys
                    or self.environ["REQUEST_METHOD"] == "HEAD"
                    or code < 200
                    or code in (204, 304)
                ):
                    self.close_connection = True
                    self.send_header("Connection", "close")
                if "server" not in header_keys:
                    self.send_header("Server", self.version_string())
                if "date" not in header_keys:
                    self.send_header("Date", self.date_time_string())
                self.end_headers()

            assert isinstance(data, bytes), "applications must write bytes"
            self.wfile.write(data)
            self.wfile.flush()

        self.write = write
        self.environ = self.make_environ()

        try:
            result = self.server.app(self.environ, start_response)
            try:
                for data in result:
                    write(data)
                if not self.headers_sent:
                    write(b"")
            finally:
                if hasattr(result, "close"):
                    result.close()
        except Exception as e:
            self.send_error(500, str(e))

if __name__ == "__main__":
    HOST, PORT = "localhost", 9999
    with socketserver.TCPServer((HOST, PORT), MyTCPHandler) as server:
        print("Server started at {}:{}".format(HOST, PORT))
        server.serve_forever()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

Resumir

Através deste tutorial, analisamos detalhadamente um servidor WSGI start_response Método que explica como trata o código de status e os cabeçalhos de resposta de uma resposta HTTP e retorna uma função para gravar os dados da resposta. A compreensão desse conteúdo o ajudará a compreender melhor a especificação WSGI e a implementar um servidor WSGI customizado. Espero que este tutorial seja útil para você.Para mais detalhes e exemplos, consulteDocumentação oficial。