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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
import json
import random
from contextlib import suppress
from io import BytesIO
from pathlib import Path
import discord
from PIL import Image
from discord.ext import commands
from pydis_core.utils.logging import get_logger
from bot.bot import Bot
from bot.utils import helpers
log = get_logger(__name__)
HTML_COLOURS = json.loads(Path("bot/resources/fun/html_colours.json").read_text("utf8"))
XKCD_COLOURS = json.loads(Path("bot/resources/fun/xkcd_colours.json").read_text("utf8"))
COLOURS = [
(255, 0, 0, 255), (255, 128, 0, 255), (255, 255, 0, 255), (0, 255, 0, 255),
(0, 255, 255, 255), (0, 0, 255, 255), (255, 0, 255, 255), (128, 0, 128, 255)
] # Colours to be replaced - Red, Orange, Yellow, Green, Light Blue, Dark Blue, Pink, Purple
IRREPLACEABLE = [
(0, 0, 0, 0), (0, 0, 0, 255)
] # Colours that are meant to stay the same - Transparent and Black
class EggDecorating(commands.Cog):
"""Decorate some easter eggs!"""
@staticmethod
def replace_invalid(colour: str) -> int | None:
"""Attempts to match with HTML or XKCD colour names, returning the int value."""
with suppress(KeyError):
return int(HTML_COLOURS[colour], 16)
with suppress(KeyError):
return int(XKCD_COLOURS[colour], 16)
return None
@commands.command(aliases=("decorateegg",))
async def eggdecorate(
self, ctx: commands.Context, *colours: discord.Colour | str
) -> Image.Image | None:
"""
Picks a random egg design and decorates it using the given colours.
Colours are split by spaces, unless you wrap the colour name in double quotes.
Discord colour names, HTML colour names, XKCD colour names and hex values are accepted.
"""
if len(colours) < 2:
await ctx.send("You must include at least 2 colours!")
return None
invalid = []
colours = list(colours)
for idx, colour in enumerate(colours):
if isinstance(colour, discord.Colour):
continue
value = self.replace_invalid(colour)
if value:
colours[idx] = discord.Colour(value)
else:
invalid.append(helpers.suppress_links(colour))
if len(invalid) > 1:
await ctx.send(f"Sorry, I don't know these colours: {' '.join(invalid)}")
return None
if len(invalid) == 1:
await ctx.send(f"Sorry, I don't know the colour {invalid[0]}!")
return None
async with ctx.typing():
# Expand list to 8 colours
colours_n = len(colours)
if colours_n < 8:
q, r = divmod(8, colours_n)
colours = colours * q + colours[:r]
num = random.randint(1, 6)
im = Image.open(Path(f"bot/resources/holidays/easter/easter_eggs/design{num}.png"))
data = list(im.getdata())
replaceable = {x for x in data if x not in IRREPLACEABLE}
replaceable = sorted(replaceable, key=COLOURS.index)
replacing_colours = {colour: colours[i] for i, colour in enumerate(replaceable)}
new_data = []
for x in data:
if x in replacing_colours:
new_data.append((*replacing_colours[x].to_rgb(), 255))
# Also ensures that the alpha channel has a value
else:
new_data.append(x)
new_im = Image.new(im.mode, im.size)
new_im.putdata(new_data)
bufferedio = BytesIO()
new_im.save(bufferedio, format="PNG")
bufferedio.seek(0)
file = discord.File(bufferedio, filename="egg.png") # Creates file to be used in embed
embed = discord.Embed(
title="Your Colourful Easter Egg",
description="Here is your pretty little egg. Hope you like it!"
)
embed.set_image(url="attachment://egg.png")
embed.set_footer(text=f"Made by {ctx.author.display_name}", icon_url=ctx.author.display_avatar.url)
await ctx.send(file=file, embed=embed)
return new_im
async def setup(bot: Bot) -> None:
"""Load the Egg decorating Cog."""
await bot.add_cog(EggDecorating())
|
