1 changed files with 124 additions and 0 deletions
@ -0,0 +1,124 @@ |
|||
from __future__ import annotations |
|||
|
|||
from discord.ext import commands |
|||
from discord import ui |
|||
import discord |
|||
import aiohttp |
|||
|
|||
|
|||
class Bot(commands.Bot): |
|||
# Suppress error on the User attribute being None since it fills up later |
|||
user: discord.ClientUser |
|||
|
|||
def __init__(self): |
|||
intents = discord.Intents.default() |
|||
super().__init__(command_prefix=commands.when_mentioned, intents=intents) |
|||
|
|||
async def setup_hook(self) -> None: |
|||
# Create a session for making HTTP requests. |
|||
self.session = aiohttp.ClientSession() |
|||
|
|||
async def close(self) -> None: |
|||
# Close the session when the bot is shutting down. |
|||
await self.session.close() |
|||
await super().close() |
|||
|
|||
async def on_ready(self): |
|||
print(f'Logged in as {self.user} (ID: {self.user.id})') |
|||
print('------') |
|||
|
|||
async def get_random_dog_image(self) -> str: |
|||
async with self.session.get('https://random.dog/woof.json') as resp: |
|||
js = await resp.json() |
|||
return js['url'] |
|||
|
|||
|
|||
# This is a row of buttons that will be used in our larger LayoutView later. |
|||
# An ActionRow is similar to a View but it can only contain up to 5 buttons or 1 select menu. |
|||
# Similar to a View it can be inherited to make it easier to manage. |
|||
class EmbedChangeButtons(ui.ActionRow): |
|||
def __init__(self, view: 'EmbedLikeView') -> None: |
|||
self.__view = view |
|||
super().__init__() |
|||
|
|||
@ui.button(label='New Image', style=discord.ButtonStyle.gray) |
|||
async def new_image(self, interaction: discord.Interaction[Bot], button: discord.ui.Button) -> None: |
|||
url = await interaction.client.get_random_dog_image() |
|||
self.__view.thumbnail.media.url = url |
|||
await interaction.response.edit_message(view=self.__view) |
|||
|
|||
@ui.button(label='Change Text', style=discord.ButtonStyle.primary) |
|||
async def change_text(self, interaction: discord.Interaction[Bot], button: discord.ui.Button) -> None: |
|||
await interaction.response.send_modal(ChangeTextModal(self.__view)) |
|||
|
|||
|
|||
# This is a simple modal to allow the content of the text portion of the "embed" to be changed by the user. |
|||
class ChangeTextModal(ui.Modal, title='Change Text'): |
|||
new_text = ui.TextInput(label='The new text', style=discord.TextStyle.long) |
|||
|
|||
def __init__(self, view: 'EmbedLikeView') -> None: |
|||
self.__view = view |
|||
self.new_text.default = view.random_text.content |
|||
super().__init__() |
|||
|
|||
async def on_submit(self, interaction: discord.Interaction, /) -> None: |
|||
self.__view.random_text.content = str(self.new_text.value) |
|||
await interaction.response.edit_message(view=self.__view) |
|||
self.stop() |
|||
|
|||
|
|||
# This defines a simple LayoutView that uses a Container to wrap its contents |
|||
# A Container is similar to an Embed, in that it has an accent colour and darkened background. |
|||
# It differs from an Embed in that it can contain other items, such as buttons, galleries, or sections, etc. |
|||
class EmbedLikeView(ui.LayoutView): |
|||
def __init__(self, *, url: str) -> None: |
|||
super().__init__() |
|||
|
|||
# When we want to use text somewhere, we can wrap it in a TextDisplay object so it becomes an Item. |
|||
self.random_text = ui.TextDisplay('This is a random dog image! Press the button to change it and this text!') |
|||
# A thumbnail is an Item that can be used to display an image as a thumbnail. |
|||
# It needs to be wrapped inside a Section object to be used. |
|||
# A Section is a container that can hold 3 TextDisplay and an accessory. |
|||
# The accessory can either be a Thumbnail or a Button. |
|||
# Since we're emulating an Embed, we will use a Thumbnail. |
|||
self.thumbnail = ui.Thumbnail(media=url) |
|||
self.section = ui.Section(self.random_text, accessory=self.thumbnail) |
|||
self.buttons = EmbedChangeButtons(self) |
|||
|
|||
# Wrap all of this inside a Container |
|||
# To visualize how this looks, you can think of it similar to this ASCII diagram: |
|||
# +----------------------Container--------------------+ |
|||
# | +--------------------Section--------------------+ | |
|||
# | | +----------------------------+ +-Thumbnail-+ | | |
|||
# | | | TextDisplay | | Accessory | | | |
|||
# | | | | | | | | |
|||
# | | | | | | | | |
|||
# | | | | | | | | |
|||
# | | +----------------------------+ +-----------+ | | |
|||
# | +-----------------------------------------------+ | |
|||
# | +------------------ActionRow--------------------+ | |
|||
# | |+-------------+ +-------------+ | | |
|||
# | || Button A | | Button B | | | |
|||
# | |+-------------+ +-------------+ | | |
|||
# | +-----------------------------------------------+ | |
|||
# +---------------------------------------------------+ |
|||
|
|||
# If you want the "embed" to have multiple images you can add a MediaGallery item |
|||
# to the container as well, which lets you have up to 10 images in a grid-like gallery. |
|||
|
|||
container = ui.Container(self.section, self.buttons, accent_color=discord.Color.blurple()) |
|||
self.add_item(container) |
|||
|
|||
|
|||
bot = Bot() |
|||
|
|||
|
|||
@bot.command() |
|||
async def embed(ctx: commands.Context[Bot]): |
|||
"""Shows the basic Embed-like LayoutView.""" |
|||
url = await ctx.bot.get_random_dog_image() |
|||
# Note that when sending LayoutViews, you cannot send any content, embeds, stickers, or polls. |
|||
await ctx.send(view=EmbedLikeView(url=url)) |
|||
|
|||
|
|||
bot.run('token') |
Loading…
Reference in new issue