Refactoring: step 2
- независимые функции вынесены в хелперы - добавлены тесты на хелперы и простые команды бота - проверка урла Правил вынесена в хелпер - уровень логирования повышен для бота - добавлено пара кейсов и слов в спам - при проверке на спам идет сработка если кроме русского и препинания есть еще какие-то символы -
This commit is contained in:
@@ -1,7 +1,11 @@
|
||||
import html
|
||||
import logging
|
||||
import os
|
||||
import random
|
||||
import shelve
|
||||
from pathlib import Path
|
||||
|
||||
from telebot import types
|
||||
|
||||
from .config import ROLLBACK
|
||||
from .constants import DATA
|
||||
@@ -63,3 +67,16 @@ def reset_report_stats(chat_id):
|
||||
{shelve_db['Messages']=}
|
||||
{shelve_db['Banned']=}
|
||||
{shelve_db['Deleted']=}"""
|
||||
|
||||
|
||||
def update_stats(message: types.Message):
|
||||
if not Path(f"{DATA}{message.chat.id}").exists():
|
||||
reset_report_stats(message.chat.id)
|
||||
with shelve.open(f"{DATA}{message.chat.id}", writeback=True) as shelve_db:
|
||||
if "Messages" not in shelve_db:
|
||||
reset_report_stats(message.chat.id)
|
||||
if message.from_user.id not in shelve_db["Messages"]:
|
||||
shelve_db["Messages"][message.from_user.id] = {"User": message.from_user, "Count": 1}
|
||||
logging.info(f"[{message.chat.title[:10]}] [{message.from_user.id}] {message.from_user.first_name}")
|
||||
else:
|
||||
shelve_db["Messages"][message.from_user.id]["Count"] += 1
|
||||
|
||||
Reference in New Issue
Block a user