# `socketserver` — Um framework para servidores de rede

Código-fonte: Lib/socketserver.py

O módulo `socketserver` simplifica a tarefa de escrever servidores de rede.

Disponibilidade: not Emscripten, not WASI.

Este módulo não funciona ou não está disponível em plataformas WebAssembly `wasm32-emscripten` e `wasm32-wasi`. Veja Plataformas WebAssembly para mais informações.

Existem quatro classes básicas de servidores concretos:

class socketserver.TCPServer(server_address, RequestHandlerClass, bind_and_activate=True)

  Isso usa o protocolo TCP da internet, que fornece fluxos contínuos de dados entre o cliente e o servidor. Se bind_and_activate for true, o construtor tenta automaticamente invocar `server_bind()` e `server_activate()`. Os outros parâmetros são passados para a classe base `BaseServer`.

class socketserver.UDPServer(server_address, RequestHandlerClass, bind_and_activate=True)

  Isso usa datagramas, que são pacotes discretos de informações que podem chegar fora de ordem ou ser perdidos durante o trânsito. Os parâmetros são os mesmos de `TCPServer`.

class socketserver.UnixStreamServer(server_address, RequestHandlerClass, bind_and_activate=True)
class socketserver.UnixDatagramServer(server_address, RequestHandlerClass, bind_and_activate=True)

  Essas classes usadas com menos frequência são semelhantes às classes TCP e UDP, mas usam soquetes de domínio Unix; elas não estão disponíveis em plataformas não Unix. Os parâmetros são os mesmos de `TCPServer`.

Essas quatro classes processam solicitações synchronously; cada solicitação deve ser concluída antes que a próxima solicitação possa ser iniciada. Isso não é adequado se cada solicitação demorar muito para ser concluída, porque exige muita computação ou porque retorna muitos dados que o cliente demora a processar. A solução é criar um processo ou thread separado para lidar com cada solicitação; as classes misturadas `ForkingMixIn` e `ThreadingMixIn` podem ser usadas para oferecer suporte ao comportamento assíncrono.

Criar um servidor requer várias etapas. Primeiro, você deve criar uma classe de manipulador de solicitação estendendo a classe `BaseRequestHandler` e substituindo seu método `handle()`; esse método processará as solicitações recebidas. Segundo, você deve instanciar uma das classes de servidor, passando a ela o endereço do servidor e a classe de manipulador de solicitação. É recomendável usar o servidor em uma instrução `with`. Em seguida, chame o método `handle_request()` ou `serve_forever()` do objeto do servidor para processar uma ou muitas solicitações. Finalmente, chame `server_close()` para fechar o soquete (a menos que você tenha usado uma instrução `with`).

Ao herdar de `ThreadingMixIn` para comportamento de conexão em threads, você deve declarar explicitamente como quer que suas threads se comportem em um desligamento abrupto. A classe `ThreadingMixIn` define um atributo daemon_threads, que indica se o servidor deve ou não esperar pelo término da thread. Você deve definir o sinalizador explicitamente se quiser que as threads se comportem de forma autônoma; o padrão é `False`, o que significa que o Python não sairá até que todas as threads criadas por `ThreadingMixIn` tenham sido encerradas.

As classes de servidor têm os mesmos métodos e atributos externos, não importa qual protocolo de rede eles usam.

## Notas sobre criação do servidor

Existem cinco classes em um diagrama de herança, quatro das quais representam servidores síncronos de quatro tipos:

    +------------+
    | BaseServer |
    +------------+
          |
          v
    +-----------+        +------------------+
    | TCPServer |------->| UnixStreamServer |
    +-----------+        +------------------+
          |
          v
    +-----------+        +--------------------+
    | UDPServer |------->| UnixDatagramServer |
    +-----------+        +--------------------+

Observe que `UnixDatagramServer` deriva de `UDPServer`, não de `UnixStreamServer` — a única diferença entre um IP e um servidor Unix é a família de endereços.

class socketserver.ForkingMixIn
class socketserver.ThreadingMixIn

  Versões de fork e threads de cada tipo de servidor podem ser criadas usando essas classes mix-in. Por exemplo, `ThreadingUDPServer` é criado da seguinte forma:

      class ThreadingUDPServer(ThreadingMixIn, UDPServer):
          pass

  A classe mix-in vem primeiro, pois ela substitui um método definido em `UDPServer`. Definir os vários atributos também altera o comportamento do mecanismo de servidor subjacente.

  `ForkingMixIn` e as classes Forking mencionadas abaixo estão disponíveis apenas em plataformas POSIX que oferecem suporte a `fork()`.

  block_on_close

    `ForkingMixIn.server_close` aguarda até que todos os processos filho sejam concluídos, exceto se o atributo `block_on_close` for `False`.

    `ThreadingMixIn.server_close` aguarda até que todos as threads não daemon sejam concluídos, exceto se o atributo `block_on_close` for `False`.

  daemon_threads

    Para `ThreadingMixIn`, use threads como daemon definindo `ThreadingMixIn.daemon_threads` como `True` para não esperar até que as threads sejam concluídas.

  Alterado na versão 3.7: `ForkingMixIn.server_close` e `ThreadingMixIn.server_close` agora esperam até que todos os processos filhos e threads não daemon sejam concluídos. Adicione um novo atributo de classe `ForkingMixIn.block_on_close` para optar pelo comportamento pré-3.7.

class socketserver.ForkingTCPServer
class socketserver.ForkingUDPServer
class socketserver.ThreadingTCPServer
class socketserver.ThreadingUDPServer
class socketserver.ForkingUnixStreamServer
class socketserver.ForkingUnixDatagramServer
class socketserver.ThreadingUnixStreamServer
class socketserver.ThreadingUnixDatagramServer

  Essas classes são predefinidas usando as classes mix-in.

Adicionado na versão 3.12: As classes `ForkingUnixStreamServer` e `ForkingUnixDatagramServer` foram adicionadas.

Para implementar um serviço, você deve derivar uma classe de `BaseRequestHandler` e redefinir seu método `handle()`. Você pode então executar várias versões do serviço combinando uma das classes de servidor com sua classe de manipulador de solicitação. A classe de manipulador de solicitação deve ser diferente para serviços de datagrama ou fluxo. Isso pode ser ocultado usando as subclasses de manipulador `StreamRequestHandler` ou `DatagramRequestHandler`.

Claro, você ainda tem que usar sua cabeça! Por exemplo, não faz sentido usar um servidor de Forking se o serviço contém estado na memória que pode ser modificado por diferentes solicitações, já que as modificações no processo filho nunca atingiriam o estado inicial mantido no processo pai e passado para cada filho. Neste caso, você pode usar um servidor de Threading, mas provavelmente terá que usar travas para proteger a integridade dos dados compartilhados.

Por outro lado, se você estiver construindo um servidor HTTP onde todos os dados são armazenados externamente (por exemplo, no sistema de arquivos), uma classe síncrona essencialmente tornará o serviço “surdo” enquanto uma solicitação estiver sendo manipulada – o que pode ser por um tempo muito longo se um cliente for lento para receber todos os dados que solicitou. Aqui, um servidor de Threading ou Forking é apropriado.

Em alguns casos, pode ser apropriado processar parte de uma solicitação de forma síncrona, mas terminar o processamento em um filho bifurcado dependendo dos dados da solicitação. Isso pode ser implementado usando um servidor síncrono e fazendo uma bifurcação explícita no método da classe do manipulador de solicitações `handle()`.

Outra abordagem para lidar com múltiplas requisições simultâneas em um ambiente que não oferece suporte a threads nem `fork()` (ou onde estes são muito caros ou inapropriados para o serviço) é manter uma tabela explícita de requisições parcialmente finalizadas e usar `selectors` para decidir em qual requisição trabalhar em seguida (ou se deve lidar com uma nova requisição recebida). Isto é particularmente importante para serviços de fluxo onde cada cliente pode potencialmente ficar conectado por um longo tempo (se threads ou subprocessos não puderem ser usados).

## Objetos Server

class socketserver.BaseServer(server_address, RequestHandlerClass)

  Esta é a superclasse de todos os objetos Server no módulo. Ela define a interface, dada abaixo, mas não implementa a maioria dos métodos, o que é feito em subclasses. Os dois parâmetros são armazenados nos respectivos atributos `server_address` e `RequestHandlerClass`.

  fileno()

    Retorna um descritor de arquivo inteiro para o soquete no qual o servidor está escutando. Esta função é mais comumente passada para `selectors`, para permitir o monitoramento de múltiplos servidores no mesmo processo.

  handle_request()

    Processa uma única solicitação. Esta função chama os seguintes métodos em ordem: `get_request()`, `verify_request()` e `process_request()`. Se o método `handle()` fornecido pelo usuário da classe manipuladora levantar uma exceção, o método `handle_error()` do servidor será chamado. Se nenhuma solicitação for recebida dentro de `timeout` segundos, `handle_timeout()` será chamado e `handle_request()` retornará.

  serve_forever(poll_interval=0.5)

    Manipula solicitações até uma solicitação explícita `shutdown()`. Sonda para desligamento a cada poll_interval segundos. Ignora o atributo `timeout`. Ele também chama `service_actions()`, que pode ser usado por uma subclasse ou mixin para fornecer ações específicas para um determinado serviço. Por exemplo, a classe `ForkingMixIn` usa `service_actions()` para limpar processos filhos zumbis.

    Alterado na versão 3.3: Adicionada chamada `service_actions` ao método `serve_forever`.

  service_actions()

    Isso é chamado no laço `serve_forever()`. Este método pode ser substituído por subclasses ou classes mixin para executar ações específicas para um determinado serviço, como ações de limpeza.

    Adicionado na versão 3.3.

  shutdown()

    Diz ao laço `serve_forever()` para parar e esperar até que isso aconteça. `shutdown()` deve ser chamado enquanto `serve_forever()` estiver em execução em um thread diferente, caso contrário, ocorrerá um impasse.

  server_close()

    Limpa o servidor. Pode ser substituído.

  address_family

    A família de protocolos à qual o soquete do servidor pertence. Exemplos comuns são `socket.AF_INET`, `socket.AF_INET6` e `socket.AF_UNIX`. Subclasse as classes de servidor TCP ou UDP neste módulo com o atributo de classe `address_family = AF_INET6` definido se você quiser classes de servidor IPv6.

  RequestHandlerClass

    A classe de manipulador de solicitações fornecida pelo usuário; uma instância dessa classe é criada para cada solicitação.

  server_address

    O endereço no qual o servidor está escutando. O formato dos endereços varia dependendo da família do protocolo; veja a documentação do módulo `socket` para detalhes. Para protocolos de internet, esta é uma tupla contendo uma string dando o endereço e um número de porta inteiro: `('127.0.0.1', 80)`, por exemplo.

  socket

    O objeto soquete no qual o servidor ouve para solicitações recebidas.

  As classes de servidor oferecem suporte às seguintes variáveis de classe:

  allow_reuse_address

    Se o servidor permitirá a reutilização de um endereço. O padrão é `False`, e pode ser definido em subclasses para alterar a política.

  request_queue_size

    O tamanho da fila de requisições. Se levar muito tempo para processar uma única requisição, todas as requisições que chegarem enquanto o servidor estiver ocupado serão colocadas em uma fila, até `request_queue_size` requisições. Quando a fila estiver cheia, outras requisições de clientes receberão um erro “Conexão negada”. O valor padrão é geralmente 5, mas isso pode ser substituído por subclasses.

  socket_type

    O tipo de soquete usado pelo servidor; `socket.SOCK_STREAM` e `socket.SOCK_DGRAM` são dois valores comuns.

  timeout

    Duração do tempo limite, medida em segundos, ou `None` se nenhum tempo limite for desejado. Se `handle_request()` não receber nenhuma requisição de entrada dentro do período de tempo limite, o método `handle_timeout()` será chamado.

  Existem vários métodos de servidor que podem ser substituídos por subclasses de classes de servidor base, como `TCPServer`; esses métodos não são úteis para usuários externos do objeto de servidor.

  finish_request(request, client_address)

    Na verdade, processa a solicitação instanciando `RequestHandlerClass` e chamando seu método `handle()`.

  get_request()

    Deve aceitar uma solicitação do soquete e retornar uma tupla de 2 elementos contendo o novo objeto de soquete a ser usado para se comunicar com o cliente e o endereço do cliente.

  handle_error(request, client_address)

    Esta função é chamada se o método `handle()` de uma instância `RequestHandlerClass` levanta uma exceção. A ação padrão é exibir o traceback para erro padrão e continuar manipulando solicitações futuras.

    Alterado na versão 3.6: Agora é chamada apenas para exceções derivadas da classe `Exception`.

  handle_timeout()

    Esta função é chamada quando o atributo `timeout` foi definido para um valor diferente de `None` e o período de timeout passou sem que nenhuma solicitação tenha sido recebida. A ação padrão para servidores de bifurcação é coletar o status de quaisquer processos filhos que tenham saído, enquanto em servidores de threading esse método não faz nada.

  process_request(request, client_address)

    Chama `finish_request()` para criar uma instância de `RequestHandlerClass`. Se desejado, esta função pode criar um novo processo ou thread para manipular a solicitação; as classes `ForkingMixIn` e `ThreadingMixIn` fazem isso.

  server_activate()

    Chamado pelo construtor do servidor para ativar o servidor. O comportamento padrão para um servidor TCP apenas invoca `listen()` no soquete do servidor. Pode ser substituído.

  server_bind()

    Chamada pelo construtor do servidor para vincular o soquete ao endereço desejado. Pode ser substituída.

  verify_request(request, client_address)

    Deve retornar um valor booleano; se o valor for `True`, a solicitação será processada, e se for `False`, a solicitação será negada. Esta função pode ser substituída para implementar controles de acesso para um servidor. A implementação padrão sempre retorna `True`.

  Alterado na versão 3.6: Foi adicionado suporte para o protocolo gerenciador de contexto. Sair do gerenciador de contexto é equivalente a chamar `server_close()`.

## Objetos manipulador de requisições

class socketserver.BaseRequestHandler

  Esta é a superclasse de todos os objetos manipulador de requisições. Ela define a interface, dada abaixo. Uma subclasse concreta do manipulador de requisições deve definir um novo método `handle()` e pode substituir qualquer um dos outros métodos. Uma nova instância da subclasse é criada para cada requisição.

  setup()

    Chamada antes do método `handle()` para executar quaisquer ações de inicialização necessárias. A implementação padrão não faz nada.

  handle()

    Esta função deve fazer todo o trabalho necessário para atender a uma solicitação. A implementação padrão não faz nada. Vários atributos de instância estão disponíveis para ela; a solicitação está disponível como `request`; o endereço do cliente como `client_address`; e a instância do servidor como `server`, caso precise de acesso a informações por servidor.

    O tipo de `request` é diferente para serviços de datagrama ou fluxo. Para serviços de fluxo, `request` é um objeto soquete; para serviços de datagrama, `request` é um par de string e soquete.

  finish()

    Chamada após o método `handle()` para executar quaisquer ações de limpeza necessárias. A implementação padrão não faz nada. Se `setup()` levanta uma exceção, esta função não será chamada.

  request

    O novo objeto `socket.socket` a ser usado para se comunicar com o cliente.

  client_address

    Endereço do cliente retornado por `BaseServer.get_request()`.

  server

    Objeto `BaseServer` usado para manipular a solicitação.

class socketserver.StreamRequestHandler
class socketserver.DatagramRequestHandler

  Essas subclasses `BaseRequestHandler` substituem os métodos `setup()` e `finish()` e fornecem os atributos `rfile` e `wfile`.

  rfile

    Um objeto arquivo do qual recebe a solicitação é lido. Oferece suporte à interface legível `io.BufferedIOBase`.

  wfile

    Um objeto arquivo no qual a resposta é escrita. Oferece suporte à interface gravável `io.BufferedIOBase`

  Alterado na versão 3.6: `wfile` também oferece suporte à interface gravável `io.BufferedIOBase`.

## Exemplos

### Exemplo de `socketserver.TCPServer`

Este é o lado do servidor:

    import socketserver

    class MyTCPHandler(socketserver.BaseRequestHandler):
        """
        A classe de manipulador de solicitação para nosso servidor.

        Ela é instanciada uma vez por conexão com o servidor e deve
        substituir o método handle() para implementar a comunicação
        com o cliente.
        """

        def handle(self):
            # self.request é o soquete TCP conectado ao cliente
            pieces = [b'']
            total = 0
            while b'\n' not in pieces[-1] and total < 10_000:
                pieces.append(self.request.recv(2000))
                total += len(pieces[-1])
            self.data = b''.join(pieces)
            print(f"Received from {self.client_address[0]}:")
            print(self.data.decode("utf-8"))
            # só envia de volta os mesmos dados, mas em caixa alta
            self.request.sendall(self.data.upper())
            # após retornarmos, o soquete será fechado.

    if __name__ == "__main__":
        HOST, PORT = "localhost", 9999

        # Cria o servidor, ligando-o ao localhost na porta 9999
        with socketserver.TCPServer((HOST, PORT), MyTCPHandler) as server:
            # Ativa o servidor; isso vai mantê-lo em execução até você
            # interromper o programa com Ctrl-C
            server.serve_forever()

Uma classe de manipulador de solicitações alternativa que faz uso de fluxos (objetos arquivo ou similar que simplificam a comunicação ao fornecer a interface de arquivo padrão):

    class MyTCPHandler(socketserver.StreamRequestHandler):

        def handle(self):
            # self.rfile é um objeto arquivo ou similar criado pelo manipulador.
            # Agora podemos usar, por exemplo, readline() em vez de chamadas
            # brutas a recv().
            # Nos limitamos a 10000 bytes para evitar abuso pelo remetente.
            self.data = self.rfile.readline(10000).rstrip()
            print(f"{self.client_address[0]} wrote:")
            print(self.data.decode("utf-8"))
            # Da mesma forma, self.wfile é um objeto arquivo ou similar para
            # escrever de volta para o cliente
            self.wfile.write(self.data.upper())

A diferença é que a chamada `readline()` no segundo manipulador chamará `recv()` várias vezes até encontrar um caractere de nova linha, enquanto o primeiro manipulador teve que usar um laço `recv()` para acumular dados até uma nova linha. Se tivesse usado apenas um único `recv()` sem o laço, ele teria retornado apenas o que foi recebido até agora do cliente. O TCP é baseado em fluxo: os dados chegam na ordem em que foram enviados, mas não há correlação entre as chamadas `send()` ou `sendall()` do cliente e o número de chamadas `recv()` no servidor necessárias para recebê-los.

Este é o lado do cliente:

    import socket
    import sys

    HOST, PORT = "localhost", 9999
    data = " ".join(sys.argv[1:])

    # Cria um soquete (SOCK_STREAM significa um soquete TCP)
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
        # Conecta a um servidor e envia dados
        sock.connect((HOST, PORT))
        sock.sendall(bytes(data, "utf-8"))
        sock.sendall(b"\n")

        # Recebe dados do servidor e desliga
        received = str(sock.recv(1024), "utf-8")

    print("Sent:    ", data)
    print("Received:", received)

A saída do exemplo deve se parecer com algo como isso:

Servidor:

    $ python TCPServer.py
    127.0.0.1 wrote:
    b'hello world with TCP'
    127.0.0.1 wrote:
    b'python is nice'

Cliente:

    $ python TCPClient.py hello world with TCP
    Sent:     hello world with TCP
    Received: HELLO WORLD WITH TCP
    $ python TCPClient.py python is nice
    Sent:     python is nice
    Received: PYTHON IS NICE

### Exemplo `socketserver.UDPServer`

Este é o lado do servidor:

    import socketserver

    class MyUDPHandler(socketserver.BaseRequestHandler):
        """
        Esta classe funciona de forma semelhante à classe do manipulador TCP,
        exceto que self.request consiste em um par de dados e soquete do cliente
        e, como não há conexão, o endereço do cliente deve ser fornecido
        explicitamente ao enviar dados de volta via sendto().
        """

        def handle(self):
            data = self.request[0].strip()
            socket = self.request[1]
            print(f"{self.client_address[0]} wrote:")
            print(data)
            socket.sendto(data.upper(), self.client_address)

    if __name__ == "__main__":
        HOST, PORT = "localhost", 9999
        with socketserver.UDPServer((HOST, PORT), MyUDPHandler) as server:
            server.serve_forever()

Este é o lado do cliente:

    import socket
    import sys

    HOST, PORT = "localhost", 9999
    data = " ".join(sys.argv[1:])

    # SOCK_DGRAM é o tipo de soquete para usar para soquetes UDP
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    # Como você pode ver, não há uma chamada connect(); UDP não tem conexão.
    # Em vez disso, dados são enviados diretamente para o destinatário via sendto().
    sock.sendto(bytes(data + "\n", "utf-8"), (HOST, PORT))
    received = str(sock.recv(1024), "utf-8")

    print("Sent:    ", data)
    print("Received:", received)

A saída do exemplo deve ser exatamente igual à do exemplo do servidor TCP.

### Mixins assíncronos

Para construir manipuladores assíncronos, use as classes `ThreadingMixIn` e `ForkingMixIn`.

Um exemplo para a classe `ThreadingMixIn`:

    import socket
    import threading
    import socketserver

    class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler):

        def handle(self):
            data = str(self.request.recv(1024), 'ascii')
            cur_thread = threading.current_thread()
            response = bytes("{}: {}".format(cur_thread.name, data), 'ascii')
            self.request.sendall(response)

    class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
        pass

    def client(ip, port, message):
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
            sock.connect((ip, port))
            sock.sendall(bytes(message, 'ascii'))
            response = str(sock.recv(1024), 'ascii')
            print("Received: {}".format(response))

    if __name__ == "__main__":
        # Porta 0 significa selecionar uma porta não usada arbitrária
        HOST, PORT = "localhost", 0

        server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler)
        with server:
            ip, port = server.server_address

            # Inicia uma thread com o servidor -- essa thread vai então
            # iniciar mais um thread para cada requisição
            server_thread = threading.Thread(target=server.serve_forever)
            # Sai da thread do servidor quando a thread principal é encerrada
            server_thread.daemon = True
            server_thread.start()
            print("Server loop running in thread:", server_thread.name)

            client(ip, port, "Hello World 1")
            client(ip, port, "Hello World 2")
            client(ip, port, "Hello World 3")

            server.shutdown()

A saída do exemplo deve se parecer com algo como isso:

    $ python ThreadedTCPServer.py
    Server loop running in thread: Thread-1
    Received: Thread-2: Hello World 1
    Received: Thread-3: Hello World 2
    Received: Thread-4: Hello World 3

A classe `ForkingMixIn` é usada da mesma forma, exceto que o servidor gerará um novo processo para cada solicitação. Disponível apenas em plataformas POSIX que oferecem suporte a `fork()`.