You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
5.4 KiB
124 lines
5.4 KiB
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')
|
|
|