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))
|