213 lines
10 KiB
Python
213 lines
10 KiB
Python
import json
|
||
import os
|
||
import re
|
||
from random import randint, choice
|
||
|
||
import Levenshtein
|
||
import discord
|
||
from discord.ext import commands
|
||
from discord import app_commands
|
||
from discord.app_commands import locale_str as _T
|
||
from cogs.checks import is_secret
|
||
|
||
from bot import db
|
||
|
||
meow = ['Мяу', '?', ':Р', ':3', '^___^', '(o･ω･o)', '(≧◡≦)']
|
||
|
||
vase_error = ['По техническим причинам падение вазы невозможно по причинам '
|
||
'отсутствия этих самых ваз\n'
|
||
'Обратитесь к специализированному сотруднику для пополнения '
|
||
'ресурсов. __Спасибо за понимание__',
|
||
|
||
'Сотрудники Лаборатории Aperture Science обеспокоены частым разрушением '
|
||
'важного и дорогого лабораторного оборудования, из-за чего на это '
|
||
'был установлен запрет\n'
|
||
'Обратитесь к специализированному сотруднику для снятия '
|
||
'запрета. __Спасибо за понимание__']
|
||
|
||
|
||
class TalkModule(commands.Cog, name="Общение"):
|
||
"""Работа с разговорным модулем кота"""
|
||
|
||
def __init__(self, bot):
|
||
self.bot = bot
|
||
self.unknown_phrases = []
|
||
if os.path.exists('speak_data.json'):
|
||
with open('speak_data.json', 'r') as f:
|
||
self.data = json.load(f)
|
||
else:
|
||
self.data = [[['привет', 'здаров', 'приветушки'], ['Привет', 'Дароу', 'Мяу', ':)']]]
|
||
|
||
@commands.Cog.listener()
|
||
async def on_message(self, message):
|
||
if db.guild_settings.find_one({'id': message.guild.id}) is None: return
|
||
if message.author.name != 'Котейка' and db.guild_settings.find_one({"id": message.guild.id})['aux']['talk']:
|
||
try:
|
||
pattern = re.compile(
|
||
r'(^|[^а-яА-ЯЁёa-zA-Z]|[,.])((ко(т((ейк)|(ик)|(ан)|([её]нок)|(я[рт]))?|(шак)))([иаыуэя]|(ов)|(ом))?)([^а-яА-ЯЁёa-zA-Z]|$)')
|
||
if pattern.search(message.content.lower()):
|
||
rand_meow = meow[randint(0, len(meow) - 1)]
|
||
if False: pass # rand_meow == '\*Роняет вазу\*':
|
||
# try:
|
||
# if self.data['vase'] <= 0:
|
||
# await message.channel.send(vase_error[randint(0, len(vase_error) - 1)])
|
||
# self.data['vase'] = 0
|
||
# else:
|
||
# await message.channel.send(rand_meow)
|
||
# self.data['vase'] -= 1
|
||
# except IndexError:
|
||
# self.data['vasa'] = 10
|
||
else:
|
||
sended = False
|
||
max_con = 0
|
||
right_answer = ''
|
||
i = pattern.search(message.content.lower()).group(0)
|
||
|
||
text = message.content.lower().replace(i, '')
|
||
if text.startswith(','):
|
||
text = text[1:]
|
||
if text.startswith(' '):
|
||
text = text[1:]
|
||
text = text.replace('<@>', '')
|
||
if len(text) >= 3:
|
||
for group in self.data:
|
||
# print(group)
|
||
for known_pharase in group[0]:
|
||
# print(known_pharase)
|
||
con = Levenshtein.ratio(known_pharase, text)
|
||
# print(con)
|
||
if con >= 0.65:
|
||
if max_con < con:
|
||
max_con = con
|
||
right_answer = choice(self.data[self.data.index(group)][1])
|
||
|
||
sended = True
|
||
|
||
if not sended:
|
||
await message.channel.send(choice(meow))
|
||
text = message.content.lower().replace(i, '')
|
||
if text.startswith(','):
|
||
text = text[1:]
|
||
if text.startswith(' '):
|
||
text = text[1:]
|
||
text = text.replace('<@>', '')
|
||
if len(text) >= 3:
|
||
self.unknown_phrases.append(text)
|
||
else:
|
||
await message.reply(right_answer)
|
||
except Exception as e:
|
||
print(repr(e))
|
||
|
||
@commands.command(brief="Выводит список неизвестных фраз",
|
||
help="Если кот найдет сообщение, в котором содержится слово \"кот\" (неважно какого "
|
||
"вида/числа/рода) он попытается на него ответить. Если в базе нет нужного ответа, то кот"
|
||
"положит сообщение в этот список")
|
||
@commands.check(is_secret)
|
||
async def up(self, ctx):
|
||
if len(self.unknown_phrases) <= 10:
|
||
e = discord.Embed(title="Неизвестные фразы:",
|
||
description='\n'.join([f"{self.unknown_phrases.index(el) + 1}. {el}"
|
||
for el in self.unknown_phrases]))
|
||
else:
|
||
e = discord.Embed(title="Неизвестные фразы:",
|
||
description='\n'.join(
|
||
[f"{len(self.unknown_phrases) - 10 + i}. {self.unknown_phrases[-10 + i]}"
|
||
for i in range(10)]))
|
||
await ctx.send(embed=e)
|
||
|
||
@commands.command(brief="Вносит фразу в список неизвестних для дальнейшего обучения")
|
||
@commands.check(is_secret)
|
||
async def down(self, ctx, phrase):
|
||
self.unknown_phrases.append(phrase)
|
||
|
||
@commands.command(brief="Обучение коты разговору с жалкими человеками",
|
||
help="На одну неизвестную фразу может приходиться от одного до нескольких ответов.\n"
|
||
"Кот будет по одному писать неизвестную фразу и ждать минуту\n"
|
||
"Для ввода нескольких ответов напиши `+`, а потом по одной пиши ответы. После ввода всех"
|
||
"ответов напиши `stop`\n"
|
||
"`cancel` для отмены обучения\n"
|
||
"`next` для пропуска и удаления вопроса из списка\n"
|
||
"`pass` для пропуска \"на потом\"\n")
|
||
@commands.check(is_secret)
|
||
async def teach(self, ctx):
|
||
t_author = ctx.message.author
|
||
t_channel = ctx.message.channel
|
||
pharases = tuple(self.unknown_phrases)
|
||
|
||
def check(m):
|
||
return (m.author == t_author) and (m.channel == t_channel)
|
||
|
||
for phrase in phrases:
|
||
await ctx.send(phrase)
|
||
|
||
try:
|
||
answer = await self.bot.wait_for('message', timeout=60.0, check=check)
|
||
except:
|
||
await ctx.send("Обучение прервано. Сработал таймаут на минуту")
|
||
|
||
if answer.content.lower() == 'cancel':
|
||
break
|
||
|
||
elif answer.content.lower() == 'next':
|
||
del self.unknown_phrases[self.unknown_phrases.index(phrase)]
|
||
continue
|
||
|
||
elif answer.content.lower() == 'pass':
|
||
continue
|
||
|
||
elif answer.content.lower() == '+':
|
||
await ctx.send('OK')
|
||
id = len(self.data)
|
||
self.data.append([[phrase], []])
|
||
while 1:
|
||
answerInCycle = await self.bot.wait_for('message', timeout=60.0, check=check)
|
||
if answerInCycle.content.lower() == 'cancel':
|
||
break
|
||
elif answerInCycle.content.lower() == 'stop':
|
||
del self.unknown_phrases[self.unknown_phrases.index(phrase)]
|
||
break
|
||
else:
|
||
self.data[id][1].append(answerInCycle.content.replace("\'", "\\'"))
|
||
|
||
with open('speak_data.json', 'w') as f:
|
||
json.dump(self.data, f)
|
||
await ctx.send('Сохранено')
|
||
continue
|
||
|
||
else:
|
||
self.data.append([[phrase], [answer.content]])
|
||
|
||
with open('speak_data.json', 'w') as f:
|
||
json.dump(self.data, f)
|
||
await ctx.send('Сохранено')
|
||
|
||
await ctx.send('Неизвестных фраз нет')
|
||
|
||
@commands.command(name='del', brief="Удаляет элемент разговорного-БД кота")
|
||
@commands.check(is_secret)
|
||
async def del_f(self, ctx, q, *, a: int = None):
|
||
if a is None:
|
||
del self.data[q]
|
||
else:
|
||
del self.data[q][a]
|
||
|
||
with open('speak_data.json', 'w') as f:
|
||
json.dump(self.data, f)
|
||
await ctx.send('Сохранено')
|
||
|
||
@commands.command(brief="Выводит сайт с базой данных кота")
|
||
@commands.check(is_secret)
|
||
async def knowledge(self, ctx):
|
||
await ctx.send('https://miyakobot.ru')
|
||
|
||
@app_commands.command()
|
||
async def meow(self, inter):
|
||
await inter.response.send_message("Meow")
|
||
|
||
@app_commands.command(name=_T("cats"))
|
||
async def cats(self, inter):
|
||
await inter.response.send_message(_T("are cutes"))
|
||
|
||
async def setup(bot):
|
||
await bot.add_cog(TalkModule(bot))
|