# -*- coding: utf-8 -*-
"""
自定义中间件。
"""


class CorsMiddleware:
    """简易 CORS 中间件，允许所有跨域请求。"""

    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        response = self.get_response(request)
        response["Access-Control-Allow-Origin"] = "*"
        response["Access-Control-Allow-Methods"] = "GET, POST, PUT, DELETE, OPTIONS"
        response["Access-Control-Allow-Headers"] = "*"
        response["Access-Control-Allow-Credentials"] = "true"
        if request.method == "OPTIONS":
            response.status_code = 200
            response.content = b""
        return response