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_c