mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-09-12 17:06:19 +02:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
import time
|
||
|
|
||
|
|
||
|
class UserManager:
|
||
|
def __init__(self):
|
||
|
self.last_widget = {}
|
||
|
self.search_times = {}
|
||
|
self.search_ban_times = {}
|
||
|
|
||
|
def add_search_time(self, user_id: int, search_time: float):
|
||
|
current_time = time.time()
|
||
|
search_times = self.search_times.get(user_id, [])
|
||
|
search_times.append(search_time)
|
||
|
counter = 0
|
||
|
|
||
|
for i in reversed(search_times):
|
||
|
if i > current_time - 10:
|
||
|
counter = counter + 1
|
||
|
if counter > 5:
|
||
|
self.search_ban_times[user_id] = current_time + int(60)
|
||
|
del self.search_times[user_id]
|
||
|
return
|
||
|
else:
|
||
|
if counter == 1:
|
||
|
del self.search_times[user_id]
|
||
|
return
|
||
|
|
||
|
if len(search_times) > 20:
|
||
|
self.search_ban_times[user_id] = current_time + int(120)
|
||
|
del self.search_times[user_id]
|
||
|
return
|
||
|
|
||
|
self.search_times[user_id] = search_times
|
||
|
|
||
|
def check_search_ban_timeout(self, user_id: int):
|
||
|
ban_time = self.search_ban_times.get(user_id)
|
||
|
if ban_time:
|
||
|
timeout = int(ban_time - time.time())
|
||
|
if timeout > 0:
|
||
|
return timeout
|
||
|
del self.search_ban_times[user_id]
|