Skip to content

Commit f2722af

Browse files
committed
feat: add defaults to hybrid ctx delete and edit
1 parent cb0ed98 commit f2722af

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

interactions/ext/hybrid_commands/context.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -367,21 +367,23 @@ async def send(
367367
**kwargs,
368368
)
369369

370-
async def delete(self, message: "Snowflake_Type") -> None:
370+
async def delete(self, message: "Snowflake_Type" = "@original") -> None:
371371
"""
372372
Delete a message sent in response to this context. Must be in the same channel as the context.
373373
374374
Args:
375-
message: The message to delete
375+
message: The message to delete. Defaults to @original which represents the initial response message.
376376
377377
"""
378378
if self._slash_ctx:
379379
return await self._slash_ctx.delete(message)
380-
await self.client.http.delete_message(self.channel_id, to_snowflake(message))
380+
await self.client.http.delete_message(
381+
self.channel_id, to_snowflake(message) if message != "@original" else self.message_id
382+
)
381383

382384
async def edit(
383385
self,
384-
message: "Snowflake_Type",
386+
message: "Snowflake_Type" = "@original",
385387
*,
386388
content: Optional[str] = None,
387389
embeds: Optional[
@@ -431,7 +433,10 @@ async def edit(
431433
files = [file, *files] if files else [file]
432434

433435
message_data = await self.client.http.edit_message(
434-
message_payload, self.channel_id, to_snowflake(message), files=files
436+
message_payload,
437+
self.channel_id,
438+
to_snowflake(message) if message != "@original" else self.message_id,
439+
files=files,
435440
)
436441
if message_data:
437442
return self.client.cache.place_message_data(message_data)

0 commit comments

Comments
 (0)