aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

average-stab

Browse Source
This commit is contained in:
Untone 2024-09-27 13:34:44 +03:00
parent 21591df427
commit 773a01cdb1
2 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
handlers/messages_routing.py
View File

@ -48,12 +48,10 @@ async def messages_routing(msg, state):
reply_to_msg_id = reply_msg.get("message_id")
if not reply_to_msg_id and latest_toxic_message_id:
reply_to_msg_id = int(latest_toxic_message_id)
try:
# count average between all of messages
toxic_pattern = f"toxic:{cid}:{uid}:*"
toxic_score = await get_average_pattern(toxic_pattern)
except Exception:
pass
# count average between all of messages
toxic_pattern = f"toxic:{cid}:{uid}:*"
toxic_score = await get_average_pattern(toxic_pattern)
# current mesasage toxicity
if reply_to_msg_id:

21
state/scan.py
View File

@ -24,11 +24,16 @@ async def get_all_pattern(uid):
async def get_average_pattern(pattern):
scores = []
async for key in redis.scan_iter(pattern):
scr = await redis.get(str(key))
scr = int(scr)
if isinstance(scr, int):
scores.append(scr)
logger.debug(f"found {len(scores)} messages")
toxic_score = math.floor(sum(scores) / len(scores)) if scores else 0
return toxic_score
toxic_score = 0
try:
async for key in redis.scan_iter(pattern):
scr = await redis.get(str(key))
scr = int(scr)
if isinstance(scr, int):
scores.append(scr)
logger.debug(f"found {len(scores)} messages")
toxic_score = math.floor(sum(scores) / len(scores)) if scores else 0
except Exception:
pass
finally:
return toxic_score