Add item parameter to DynamicItem.from_custom_id

This commit is contained in:
Rapptz
2023-07-21 01:29:27 -04:00
parent 7c3868ef3b
commit f793af0456
4 changed files with 29 additions and 18 deletions

View File

@ -45,10 +45,10 @@ class DynamicCounter(
# This method actually extracts the information from the custom ID and creates the item.
@classmethod
async def from_custom_id(cls, interaction: discord.Interaction, match: re.Match[str], /):
async def from_custom_id(cls, interaction: discord.Interaction, item: discord.ui.Button, match: re.Match[str], /):
count = int(match['count'])
user_id = int(match['id'])
return cls(count, user_id)
return cls(user_id, count=count)
# We want to ensure that our button is only called by the user who created it.
async def interaction_check(self, interaction: discord.Interaction) -> bool:

View File

@ -51,7 +51,7 @@ class DynamicButton(discord.ui.DynamicItem[discord.ui.Button], template=r'button
# This is called when the button is clicked and the custom_id matches the template.
@classmethod
async def from_custom_id(cls, interaction: discord.Interaction, match: re.Match[str], /):
async def from_custom_id(cls, interaction: discord.Interaction, item: discord.ui.Button, match: re.Match[str], /):
user_id = int(match['id'])
return cls(user_id)