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

separate-latest-toxic

Browse Source
This commit is contained in:
Untone 2024-09-26 15:33:10 +03:00
parent a78c752e4d
commit d71a6e00ec
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
handlers/messages_routing.py
View File

@ -9,7 +9,7 @@ from handlers.handle_private import handle_private
logger = logging.getLogger('handlers.messages_routing') logger = logging.getLogger('handlers.messages_routing')
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
latest_toxic = 0 latest_toxic = {}
async def messages_routing(msg, state): async def messages_routing(msg, state):
cid = msg["chat"]["id"] cid = msg["chat"]["id"]
@ -33,15 +33,17 @@ async def messages_routing(msg, state):
elif bool(text): elif bool(text):
mid = msg.get("message_id") mid = msg.get("message_id")
if text == '/score': if text.startswith('/score'):
await telegram_api( await telegram_api(
"sendMessage", "sendMessage",
chat_id=cid, chat_id=cid,
reply_to=mid, reply_to=mid,
text=f"{latest_toxic}%" text=f"{latest_toxic.get(cid, 0)}%"
) )
toxic_score = detector(text) toxic_score = detector(text)
logger.info(f'\ntext: {text}\ntoxic: {math.floor(toxic_score*100)}%') toxic_perc = math.floor(toxic_score*100)
latest_toxic[cid] = toxic_perc
logger.info(f'\ntext: {text}\ntoxic: {toxic_perc}%')
if toxic_score > 0.81: if toxic_score > 0.81:
if toxic_score > 0.90: if toxic_score > 0.90:
await redis.set(f"removed:{uid}:{cid}:{mid}", text) await redis.set(f"removed:{uid}:{cid}:{mid}", text)