mirror of https://github.com/inexcode/ixizi
103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
|
from disco.bot import Bot, Plugin
|
|||
|
import rolldice
|
|||
|
from enum import Enum
|
|||
|
|
|||
|
class Drinks(Enum):
|
|||
|
espresso = (1, "Эспрессо")
|
|||
|
americano = (2, "Американо")
|
|||
|
cappuccino = (3, "Капучино")
|
|||
|
latte = (4, "Латте")
|
|||
|
raf = (5, "Раф-кофе")
|
|||
|
flat_white = (6, "Флэт Уайт")
|
|||
|
mocachino = (7, "Мокаччино")
|
|||
|
frappuchino = (8, "Фраппучино")
|
|||
|
hot_chocolate = (9, "Горячий шоколад")
|
|||
|
cacao = (10, "Какао")
|
|||
|
milk_coctail = (11, "Молочный коктейль")
|
|||
|
tea = (12, "Чай")
|
|||
|
|
|||
|
enjoy1 = (13, "Латте бельгийская вафля")
|
|||
|
enjoy2 = (14, "Арахисовый капучино")
|
|||
|
enjoy3 = (15, "Капучино кокосовый бисквит")
|
|||
|
enjoy4 = (16, "Лавандовый раф")
|
|||
|
enjoy5 = (17, "Грушевый раф с солёной карамелью")
|
|||
|
enjoy6 = (18, "Чай с вишней и пряностями")
|
|||
|
|
|||
|
latte_tea = (19, "Латте чай")
|
|||
|
|
|||
|
feel_winter1 = (20, "Хвойный раф")
|
|||
|
feel_winter2 = (21, "Раф грецкий орех в клиновом сиропе")
|
|||
|
feel_winter3 = (22, "Чай шиповник-липа")
|
|||
|
feel_winter4 = (23, "Чай японский лимон Юзу")
|
|||
|
|
|||
|
def __init__(self, id, title):
|
|||
|
self.id = id
|
|||
|
self.title = title
|
|||
|
|
|||
|
class Syrops(Enum):
|
|||
|
syroup1 = (1, "Карамель")
|
|||
|
syroup2 = (2, "Солёная карамель")
|
|||
|
syroup3 = (3, "Банан")
|
|||
|
syroup4 = (4, "Ваниль")
|
|||
|
syroup5 = (5, "Миндаль")
|
|||
|
syroup6 = (6, "Кокос")
|
|||
|
syroup7 = (7, "Кленовый")
|
|||
|
syroup8 = (8, "Клубника")
|
|||
|
syroup9 = (9, "Чёрная смородина")
|
|||
|
syroup10 = (10, "Имбирный пряник")
|
|||
|
syroup11 = (11, "Личи")
|
|||
|
syroup12 = (12, "Шоколад")
|
|||
|
syroup13 = (13, "Амаретто")
|
|||
|
syroup14 = (14, "Ирландский крем")
|
|||
|
sugar = (15, "Ванильный сахар")
|
|||
|
combo1 = (16, "Солёная карамель и имбирный пряник")
|
|||
|
|
|||
|
def __init__(self, id, title):
|
|||
|
self.id = id
|
|||
|
self.title = title
|
|||
|
|
|||
|
|
|||
|
class Teas(Enum):
|
|||
|
green = (1, "Зелёный")
|
|||
|
black = (2, "Чёрный")
|
|||
|
fruit = (3, "Фруктовый")
|
|||
|
|
|||
|
def __init__(self, id, title):
|
|||
|
self.id = id
|
|||
|
self.title = title
|
|||
|
|
|||
|
|
|||
|
class CoffeePlugin(Plugin):
|
|||
|
@Plugin.command('coffee', '[param:str]')
|
|||
|
def on_coffee(self, event, param=''):
|
|||
|
try:
|
|||
|
result = ""
|
|||
|
if param.lower() == 'classic':
|
|||
|
result1, expl1 = rolldice.roll_dice('1d12')
|
|||
|
elif param.lower() == 'raf':
|
|||
|
result1, expl1 = (5, 'User')
|
|||
|
else:
|
|||
|
result1, expl1 = rolldice.roll_dice('1d23')
|
|||
|
result += list(Drinks)[result1-1].title
|
|||
|
if result1 == 5:
|
|||
|
result2, expl2 = rolldice.roll_dice('1d18')
|
|||
|
if result2 > 16:
|
|||
|
result2, expl2 = rolldice.roll_dice('1d15')
|
|||
|
result21, expl21 = rolldice.roll_dice('1d15')
|
|||
|
result += "\n**Сиропы**: {} и {}".format(list(Syrops)[result2 - 1].title, list(Syrops)[result21 - 1].title)
|
|||
|
else:
|
|||
|
result += "\n**Сироп**: {}".format(list(Syrops)[result2 - 1].title)
|
|||
|
elif result1 == 9:
|
|||
|
result2, expl2 = rolldice.roll_dice('1d2')
|
|||
|
if result2 == 1:
|
|||
|
result += "\nТёмный"
|
|||
|
else:
|
|||
|
result += "\nМолочный"
|
|||
|
elif result1 == 12 or result1 == 19:
|
|||
|
result2, expl2 = rolldice.roll_dice('1d3')
|
|||
|
result += " {}".format(list(Teas)[result2 - 1].title)
|
|||
|
except Exception as e:
|
|||
|
event.msg.reply(str(e))
|
|||
|
else:
|
|||
|
event.msg.reply(result)
|