Skip to content
Snippets Groups Projects
Commit 70e6eb1d authored by Iori Ichinose's avatar Iori Ichinose :speech_balloon:
Browse files

Merge branch 'revert-28a71678' into 'master'

Revert "fix for too many recent stored"

See merge request !1
parents 28a71678 a29608d2
No related branches found
No related tags found
1 merge request!1Revert "fix for too many recent stored"
......@@ -15,7 +15,7 @@ T_Recent = dict[str, list[Message]]
class Recorder:
def __init__(self):
self.HALFMAX_MESSAGES = 1000
self.MAX_MESSAGES = 1000
self.MAX_RECENT = 20
try:
with open(LOG_PATH, 'rb') as f:
......@@ -33,8 +33,6 @@ class Recorder:
pickle.dump(self.recent, f)
def insert(self, message: Message):
for key, value in self.recent.items():
print(key, len(value))
content = message.content
text = content.extract_plain_text().strip()
user_id = str(message.user_id)
......@@ -54,12 +52,10 @@ class Recorder:
self.messages[group_id][user_id] = []
self.messages[group_id][user_id].append(text)
if (len(self.messages[group_id][user_id]) > 2 * self.HALFMAX_MESSAGES):
self.messages[group_id][user_id] = \
self.messages[group_id][user_id][self.HALFMAX_MESSAGES:]
if (len(self.recent[group_id]) > 2 * self.HALFMAX_MESSAGES):
self.recent[group_id] = \
self.recent[group_id][self.HALFMAX_MESSAGES:]
if (len(self.messages[group_id][user_id]) == self.MAX_MESSAGES):
self.messages[group_id][user_id].pop(0)
if (len(self.recent[group_id]) == self.MAX_MESSAGES):
self.recent[group_id].pop(0)
def get_recent(self, context_id: int, count: int = 5) -> list[Message]:
group_id = str(context_id)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment