aboutsummaryrefslogtreecommitdiff
path: root/handlers/gen.py
blob: 43301c54c0542884b7dc0e15548cbff6f88297b9 (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
import os

import mc
from aiogram import types as t

from shared import config
from shared.instances import bot, dp
from utils import filters as f


@dp.message_handler(commands=["gen"])
async def сгенерировать_хуету(msg: t.Message):
    await msg.answer(получить_говно(msg.chat.id))


@dp.message_handler(commands=["del"])
async def удалить_хуету(msg: t.Message):
    await msg.delete()

    if msg.reply_to_message:
        if msg.reply_to_message.from_user.id in [bot.id, msg.from_user.id]:
            await msg.reply_to_message.delete()
        else:
            await msg.answer("Ты умник, можно только свои или мои удалять")
    else:
        await msg.answer("Ты умник, ответь на сообщение")


@dp.message_handler(commands=["void"])
async def лоботомия(msg: t.Message):
    if msg.get_args() == "Я знаю что делаю":
        os.remove(f"data/{msg.chat.id}")
        await msg.answer("Лоботомия проведена успешно")
    else:
        await msg.answer(
            "Напишите <code>/void Я знаю что делаю</code>", parse_mode=t.ParseMode.HTML
        )


@dp.message_handler(commands=["chance"])
async def изменить_шанс_срания(msg: t.Message):
    if msg.get_args():
        try:
            chance = int(msg.get_args().split()[0])
            if 0 <= chance <= 100:
                config.chances[msg.chat.id] = chance
                config.save()
            else:
                raise RuntimeError()

            await msg.answer(f"Теперь я сру с шансом в: {chance}%")
        except Exception:
            await msg.answer(
                "Я хз что не так, но я знаю что ты дебил \n    /chance <ЧИСЛО ОТ 0 ДО 100>"
            )
    else:
        await msg.answer(f"Я сру с шансом в: {config.chances.get(msg.chat.id, 10)}%")


@dp.message_handler(f.message.chance, content_types=[t.ContentType.ANY])
async def срать_сообщение_с_шансом(msg: t.Message):
    await msg.answer(получить_говно(msg.chat.id))


def получить_говно(id: int) -> str:
    samples = mc.util.load_txt_samples(f"data/{id}", separator="§")
    return mc.StringGenerator(samples=samples).generate_string()