forked from kemonomimi/nyabot
main: Added marriage checking command
This commit is contained in:
parent
89c1e5ba1e
commit
b30e857f9f
1 changed files with 26 additions and 0 deletions
26
main.py
26
main.py
|
@ -39,6 +39,7 @@ Slash commands:
|
|||
```
|
||||
/help - Shows this message
|
||||
/marry - Marries someone
|
||||
/marriage - Shows marriage information
|
||||
```
|
||||
"""
|
||||
|
||||
|
@ -60,6 +61,31 @@ def __find_mariage_for_member_id(member_id: int) -> list[int]:
|
|||
|
||||
# Slash commands
|
||||
|
||||
@bot.slash_command()
|
||||
@discord.guild_only()
|
||||
@discord.option(
|
||||
"target", type=discord.Member,
|
||||
description="User whos marraige you want to check",
|
||||
required=False,
|
||||
)
|
||||
async def marriage(
|
||||
context: discord.ApplicationContext, target: discord.Member | None
|
||||
):
|
||||
who_to_check = target if target is not None else context.user
|
||||
if who_to_check is None:
|
||||
await context.respond("failed to get target!", ephemeral=True)
|
||||
return
|
||||
|
||||
marriage = __find_mariage_for_member_id(who_to_check.id)
|
||||
if len(marriage) <= 0:
|
||||
await context.respond(f"{who_to_check.mention} is not married yet :(")
|
||||
return
|
||||
|
||||
marriage_formatted = '💞'.join(
|
||||
[f"<@{user}>" for user in marriage]
|
||||
)
|
||||
await context.respond(marriage_formatted)
|
||||
|
||||
|
||||
@bot.slash_command()
|
||||
@discord.guild_only()
|
||||
|
|
Loading…
Reference in a new issue