import telebot # Telegram bot module
from re import match, compile # ANALYS WORD
from db import read_db, write_db # database functions
database = read_db()
bot = telebot.TeleBot(database['token'])
def is_bad(word, bad, excepts): # check bad words
# Replace spec.symbols
# Excepts
if word in excepts:
return False
# Agressive mode, causes false positives
# res = search(bad, word)
res = match(bad, word)
if res is not None:
return True
else:
return False
def rep(word): # TRANSLIT an OTHER
symbols = [['e', 'е'], ['a', 'а'], ['i', 'и'], ['t', 'т'], ['y', 'у'], ['u', 'у'], ['o', 'о'], ['d', 'д'], ['x', 'х'],
['t', 'т'], ['p', 'п'], ['r', 'р'], ['h', 'х'], ['b', 'б'], ['n', 'н'],
['🇽', 'х'], ['🇾', 'у'], ['🇹', 'т'], ['🇪', 'е'], ['❌', 'х'], ['✖', 'х'], ['❎', 'х']]
for symbol in symbols:
word = word.replace(symbol[0], symbol[1])
return word
def catch(message): # CATCH BAD WORDS
bad_list = ['ху.+й', 'х.+уй', 'xуи', 'xyи', 'хyи', 'xyй', 'xуй', 'ху.', '.+хуе', '.+хуё', 'xyu', 'xui', 'хyй',
'поху', '.уй', 'ах.ен', 'а.уе', '.+хуй', '.+хуй', 'хуя', '.+хуя',
'бл.+ть', 'бля', 'бл.+ть', '.+бл+.дь+', '.+бл.+дь',
'трах', 'еб.+ть', 'ебу', 'ебал', '..ебен', 'ёбан', 'ебть', 'eby', '..ебись', 'уеб', 'уёб', 'ебей', 'ебу', 'ебл', 'еба',
'.+ебн.+т.+', '.+еб.ть', 'ебо', '.+ебо', '.+еба', '.+ёбы', 'еби.+', 'ёба.+', 'ебля', 'ебё.+', 'заеб', 'заеб.+', 'заёб',
'.+заеб', '.+заёб', 'заеб.+', 'заёб.+', 'ёбск', '.+ебуч.+',
'еб.+утые', 'е.+б.+утые', 'ебан', 'еб.+н', 'ебн', 'ёбн', '.+ёбка', '.+ебка',
'пр..ба', '.б.л', 'у.б', '.блан',
'.+пизд', 'пизец', 'пздец', 'п.+здец', 'пизд', '.+пизж.+',
'пид.+р', 'пидр',
'д.лб.+б',
'f.+ck', 's.+ck', 'fck', 'sck']
excepts = ['хороший', 'хороший.', 'убил', 'убил.','telebot','telebot.']
words = message.text.split()
bad_found = False
for check in words:
# Mat -> Мат
check = rep(check)
# -М#ат$ => Мат
regex = compile('[^a-zA-Zа-яА-ЯЁё]')
check = regex.sub('', check)
if bad_found:
break
for bad in bad_list:
if is_bad(check.lower(), bad, excepts):
bad_found = True
break
if not bad_found:
for check in words:
check = rep(check)
regex = compile('[^a-zA-Zа-яА-ЯЁё]')
check = regex.sub('.', check)
if bad_found:
break
for bad in bad_list:
if is_bad(check.lower(), bad, excepts):
bad_found = True
break
if bad_found:
database = read_db()
chat_id = str(message.chat.id)
user_id = str(message.from_user.id)
if chat_id not in database:
database[chat_id] = {}
if user_id not in database[chat_id]:
database[chat_id][user_id] = {}
database[chat_id][user_id]["lunch"] = 0
if user_id in database[chat_id]:
database[chat_id][user_id]["lunch"] += 1
bot.send_message(message.chat.id,
f'Пользователь съел обед.\n'
f'Кол-во съеденных обедов: {database[chat_id][user_id]["lunch"]}',
parse_mode='HTML')
write_db(database)
@bot.message_handler(commands=['stats']) # stats
def send_stats(message):
database = read_db()
chat_id = str(message.chat.id)
user_id = str(message.from_user.id)
if user_id not in database[chat_id]:
database[chat_id] = {}
database[chat_id][user_id] = {}
database[chat_id][user_id]["lunch"] = 0
else:
lunch_count = database[chat_id][user_id]["lunch"]
if lunch_count > 0:
bot.send_message(chat_id, f'Вы съели {database[chat_id][user_id]["lunch"]} обедов! Ужас...', parse_mode='HTML')
else:
bot.send_message(chat_id, f'Вы не съели обед сегодня, поздравляю!', parse_mode='HTML')
@bot.message_handler()
def catch_all_messages(message):
catch(message)
@bot.edited_message_handler()
def catch_edited_messages(message):
catch(message)
def main():
mod = 1
if mod == 1:
while True:
try:
bot.polling()
except KeyboardInterrupt:
exit()
except:
pass
elif mod == 2:
bot.polling()
else:
exit()
if __name__ == "__main__":
main()