blob: 72afb79613f6babfd50cf16b325666c122b26294 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
import random
import mc
from aiogram import types as t
from shared.database import Message
from shared.instances import bot, config, dp, session
from utils import filters as f
def get_text(chat_id: int) -> str:
with session() as s:
samples = [
m.tuple()[0]
for m in s.query(Message.message).filter(Message.chat_id == chat_id).all()
]
assert (
len(samples) != 0
), "Нету данных на основе которых можно сгенерировать сообщение"
generator = mc.PhraseGenerator(samples)
gen_config = config.get_config(chat_id).gen
validators = []
if gen_config.max_word_count is not None or gen_config.min_word_count is not None:
validators.append(
mc.builtin.validators.words_count(
minimal=gen_config.min_word_count,
maximal=gen_config.max_word_count,
)
)
while True:
message = generator.generate_phrase_or_none(1, validators=validators)
if message is not None:
return message
@dp.message_handler(commands=["gen"])
async def gen_command(msg: t.Message) -> None:
if config.get_config(msg.chat.id).gen.delete_command:
await msg.delete()
await msg.answer(get_text(msg.chat.id))
@dp.message_handler(commands=["del"])
async def del_command(msg: t.Message) -> None:
await msg.delete()
if msg.reply_to_message:
if msg.reply_to_message.from_user.id == bot.id:
await msg.reply_to_message.delete()
else:
await msg.reply("Можно удалять только сообщения бота")
else:
await msg.reply("Вы не ответили на сообщение")
@dp.message_handler(
f.message.is_chat,
f.message.chance,
content_types=[t.ContentType.ANY],
)
async def chance_message(msg: t.Message) -> None:
if config.get_config(msg.chat.id).gen.reply:
await msg.reply(get_text(msg.chat.id))
else:
await msg.answer(get_text(msg.chat.id))
|
