python - 网站加载时如何检查django项目中的用户IP？

Question

我正在 django 网站上工作。我想让这个网站受到国家限制。所以我想在访问网站时检查用户ip。 url 是什么或加载哪个应用程序并不重要。因此，如果用户不是来自特定国家/地区，我想向他们显示错误消息，表明他们无法访问该网站。

我正在使用GeoLitCity数据库来获取用户的国家/地区。我知道如何获取用户的ip。但我不知道在加载任何内容之前如何在初始化步骤中执行此操作。谁能帮我解决这个问题吗？

Answer 1

您可以创建 custom middleware class ValidateUserCountryMiddleware 将检查发出请求的国家/地区是否来自特定国家/地区。如果请求来自有效的国家/地区，则 Django 将处理该请求。否则，它将返回 HttpResponseForbidden 响应。

在我们的自定义中间件类中，我们将定义一个 process_request() 方法。它应该返回 None 或 HttpResponse 对象。

来自 process_request(): 上的文档

If it returns None, Django will continue processing this request, executing any other process_request() middleware, then, process_view() middleware, and finally, the appropriate view. If it returns an HttpResponse object, Django won’t bother calling any other request, view or exception middleware, or the appropriate view; it’ll apply response middleware to that HttpResponse, and return the result.

在这里，如果请求不是来自指定国家/地区，我们将返回 HttpResponseForbidden 响应。

from django.http import HttpResponseForbidden    

class ValidateUserCountryMiddleware(object):  
    """
    This custom middleware class will check the country from where the request
    was made using the IP address. If the country is valid, then Django will process
    the request. Otherwise, it will return a 403 forbidden response
    """

    def process_request(self, request):  
        # here write the logic to obtain the country from the IP address. 
        # Then check if its a valid country

        if not valid_country: # check if its a valid country 
            return HttpResponseForbidden('You cannot access the website from this location.') # dont't process the request and return 403 forbidden response

        return # return None in case of a valid request

然后将中间件添加到 settings.py 文件中的 MIDDLEWARE_CLASSES 中。

MIDDLEWARE_CLASSES = (
    ...
    # add your custom middleware here
    'my_project.middlewares.ValidateUserCountryMiddleware',

)