update:增强iplimit

This commit is contained in:
lan
2023-01-16 22:58:50 +08:00
parent 06082feaaa
commit 343b28fb80
+1
View File
@@ -36,6 +36,7 @@ class IPRateLimit:
def add_ip(self, ip): def add_ip(self, ip):
ip_info = self.ips.get(ip, {'count': 0, 'time': datetime.now()}) ip_info = self.ips.get(ip, {'count': 0, 'time': datetime.now()})
ip_info['count'] += 1 ip_info['count'] += 1
ip_info['time'] = datetime.now()
self.ips[ip] = ip_info self.ips[ip] = ip_info
return ip_info['count'] return ip_info['count']