22 lines
800 B
Python
22 lines
800 B
Python
|
from enum import Enum
|
||
|
|
|
||
|
|
import discord
|
||
|
|
|
||
|
|
class GenerationState(Enum):
|
||
|
|
GENERATING = "generating"
|
||
|
|
FINISHED = "finished"
|
||
|
|
|
||
|
|
class GenerationStateButton(discord.ui.Button):
|
||
|
|
def __init__(self, label: str, style: discord.ButtonStyle, emoji: str | discord.Emoji | discord.PartialEmoji | None = None, disabled: bool=False):
|
||
|
|
super().__init__(label=label, style=style, emoji=emoji, disabled=disabled)
|
||
|
|
|
||
|
|
class GenerationStateView(discord.ui.View):
|
||
|
|
def __init__(self, state: GenerationState):
|
||
|
|
super().__init__()
|
||
|
|
self.state = state
|
||
|
|
|
||
|
|
if state == GenerationState.GENERATING:
|
||
|
|
self.add_item(GenerationStateButton(label="Generating...", style=discord.ButtonStyle.grey, emoji="✨", disabled=True))
|
||
|
|
elif state == GenerationState.FINISHED:
|
||
|
|
pass
|