python redis 实现IP校验黑名单

最新推荐文章于 2024-10-08 17:22:56 发布

土掉渣的二傻子

最新推荐文章于 2024-10-08 17:22:56 发布

阅读量1.7k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python 文章标签： ip校验 python redis 黑名单 mongodb

本文链接：https://blog.youkuaiyun.com/suzimuyu99/article/details/81076883

该博客介绍了如何使用Python进行IP校验，当某个IP在10秒内访问次数超过200次，系统会将该IP加入到Redis黑名单中。项目中采用了MongoDB作为数据库，并提供了相关的测试代码确保功能正常。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

python实现IP 检验，并且将10s内访问超过200次的ip加入黑名单列表。

本项目使用的是mongodb数据库

from datetime import datetime
import logging

# from redis import Redis
import redis
from mongoengine.context_managers import switch_collection

from ego_model.py_auth_models import BlackNameList

pool = redis.ConnectionPool(host='ip', port=6379, password="88141")
R = redis.Redis(connection_pool=pool)


class IPViewSet(object):
    @classmethod
    def check_save_ip(cls, ip, sn, logger_group):
        # 主方法， 检查ip是否在黑名单，如果不在，将ip的计数加1，并且设置超时时间10s，判断ip的次数是否超过200，如果超过，将该ip保存至黑名单
        logger = logging.getLogger(logger_group)
        if cls.check_black_ip(ip):
            ip_key = cls.get_redis_string_key(ip)
            current_count = R.incr(ip_key, 1)
            if current_count == 1:
                R.expire(ip_key, 10)

            if current_count > 200:
                # 10s超过200次，将ip加入黑名单
                cls.add_black_namelist(ip)
                logg