Сервер разработки на Python: от статики до кастомных обработчиков

Расширенные примеры настройки сервера

REST-подобный сервер с маршрутизацией

Пример обработчика, который распознает разные URL и возвращает JSON-ответы.

import json
from http.server import HTTPServer, BaseHTTPRequestHandler

class RESTHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        if self.path == '/api/status':
            self.send_json({'status': 'OK', 'version': '1.0'})
        elif self.path.startswith('/api/items/'):
            item_id = self.path.split('/')[-1]
            self.send_json({'id': item_id, 'name': f'Item {item_id}'})
        else:
            self.send_error(404, 'Not Found')

    def send_json(self, data):
        self.send_response(200)
        self.send_header('Content-Type', 'application/json')
        self.end_headers()
        self.wfile.write(json.dumps(data).encode())

server = HTTPServer(('', 8080), RESTHandler)
print('REST сервер запущен на порту 8080')
server.serve_forever()

Результат: при запросе /api/status возвращается JSON {"status":"OK","version":"1.0"}. Для /api/items/42 получается {"id":"42","name":"Item 42"}.

Сервер с поддержкой POST-запросов для загрузки данных

Обработчик, который принимает POST-запросы и сохраняет тело запроса в файл.

import os
from http.server import HTTPServer, BaseHTTPRequestHandler

class UploadHandler(BaseHTTPRequestHandler):
    def do_POST(self):
        content_length = int(self.headers.get('Content-Length', 0))
        body = self.rfile.read(content_length)
        filename = f'uploads/data_{len(os.listdir("uploads"))}.bin'
        os.makedirs('uploads', exist_ok=True)
        with open(filename, 'wb') as f:
            f.write(body)
        self.send_response(201)
        self.send_header('Location', f'/{filename}')
        self.end_headers()
        self.wfile.write(b'Uploaded')

server = HTTPServer(('', 8001), UploadHandler)
print('Upload сервер на порту 8001')
server.serve_forever()

Для тестирования можно отправить POST-запрос через curl: curl -X POST --data "Hello" http://localhost:8001/. Файл сохранится в папку uploads.

Сервер с кастомной страницей ошибки 404

Модификация SimpleHTTPRequestHandler для замены стандартного сообщения об ошибке.

from http.server import SimpleHTTPRequestHandler, HTTPServer

class CustomErrorHandler(SimpleHTTPRequestHandler):
    def send_error(self, code, message=None):
        if code == 404:
            self.send_response(404)
            self.send_header('Content-Type', 'text/html')
            self.end_headers()
            self.wfile.write(b'<h1>Страница не найдена</h1><p>Попробуйте другую ссылку.</p>')
        else:
            super().send_error(code, message)

server = HTTPServer(('', 8002), CustomErrorHandler)
print('Сервер с кастомной 404 на порту 8002')
server.serve_forever()

При обращении к несуществующему пути отображается HTML-страница с сообщением на русском языке.

Сервер с фильтрацией файлов по расширению

Расширение SimpleHTTPRequestHandler, которое позволяет отдавать только файлы определенных типов.

from http.server import SimpleHTTPRequestHandler, HTTPServer
import os

class FilteredHandler(SimpleHTTPRequestHandler):
    def list_directory(self, path):
        # Переопределяем отображение директории, чтобы показывать только .html файлы
        files = [f for f in os.listdir(path) if f.endswith('.html') or os.path.isdir(os.path.join(path, f))]
        # Используем родительский метод для отображения списка
        # Но можно написать свою HTML-страницу
        return super().list_directory(path)  # упрощение

server = HTTPServer(('', 8003), FilteredHandler)
server.serve_forever()

Данный пример упрощен; полная реализация требует замены шаблона листинга директории.

Сервер с самоподписанным SSL-сертификатом

Использование модуля ssl для шифрования трафика. Потребуется сгенерировать сертификат (например, с помощью openssl).

import ssl
from http.server import HTTPServer, SimpleHTTPRequestHandler

httpd = HTTPServer(('localhost', 4443), SimpleHTTPRequestHandler)
httpd.socket = ssl.wrap_socket(httpd.socket,
                               keyfile='key.pem',
                               certfile='cert.pem',
                               server_side=True)
print('HTTPS сервер на порту 4443')
httpd.serve_forever()

Перед запуском необходимо создать файлы key.pem и cert.pem. Подключение происходит по протоколу HTTPS.