main: Added support for poly marriage

This commit also adds a PolyculeMemberJoinConfirmationView for
consenting with other polycule members
This commit is contained in:
Yuki Joou 2023-02-26 16:13:14 +01:00
parent dd576415ae
commit 360df21092
2 changed files with 87 additions and 5 deletions

View file

@ -70,3 +70,27 @@ class MariageConfirmationView(ConfirmationView):
await interaction.response.send_message( await interaction.response.send_message(
f"{replyer.mention} isn't ready yet..." f"{replyer.mention} isn't ready yet..."
) )
class PolyculeMemberJoinConfirmationView(ConfirmationView):
def __init__(self, target: discord.User):
super().__init__(timeout=15*60)
self.target = target
def _user_can_respond(self, user: discord.User | discord.Member):
return user == self.target
async def _on_wrong_user(self, interaction: discord.Interaction):
await interaction.response.send_message(
"ur not the one getting married, silly :3", ephemeral=True, delete_after=10
)
async def _on_accept(self, interaction: discord.Interaction, replyer: discord.User | discord.Member):
await interaction.response.send_message(
f"{replyer.mention} is okay for extending the polycule :3 kawaii desune~"
)
async def _on_deny(self, interaction: discord.Interaction, replyer: discord.User | discord.Member):
await interaction.response.send_message(
f"{replyer.mention} isn't ready yet..."
)

64
main.py
View file

@ -71,6 +71,9 @@ def __find_mariage_for_member_id(member_id: int) -> list[int]:
async def marry( async def marry(
context: discord.ApplicationContext, member_to_marry: discord.Member context: discord.ApplicationContext, member_to_marry: discord.Member
): ):
# TODO: Refactor marriage system
# This function is massive, it should be split up, and probably in another file!
# Check if the person asked in mariage is interested # Check if the person asked in mariage is interested
marriage_asker = context.author marriage_asker = context.author
if marriage_asker is None: if marriage_asker is None:
@ -116,12 +119,67 @@ async def marry(
return return
users_who_confirmed_the_marriage = [marriage_asker.id, member_to_marry.id]
users_who_need_to_allow_marriage = askees_marriage
users_who_need_to_allow_marriage += askers_marriage
for user_to_ask in users_who_need_to_allow_marriage:
if user_to_ask in users_who_confirmed_the_marriage:
continue
user_to_ask_as_member = await bot.fetch_user(user_to_ask)
polycule_confirmation_view = interaction_views.PolyculeMemberJoinConfirmationView(
user_to_ask_as_member)
member_to_add_to_polycule = member_to_marry if member_to_marry.id not in __find_mariage_for_member_id(
user_to_ask) else marriage_asker
await context.respond( await context.respond(
"ooh looks like you're trying to do a ploly marriage..." + f"{user_to_ask_as_member.mention}, do you agree to adding "
" unfortunately, that's still a work in progress," + + f"{member_to_add_to_polycule.mention} to the polycule?",
" come back later to register it :3" view=polycule_confirmation_view
) )
await polycule_confirmation_view.wait()
if polycule_confirmation_view.user_accepted is None:
await context.respond("you hit a bug owo. nothing i can do, try again!")
return
if polycule_confirmation_view.user_accepted == False:
await context.respond(f"seems like {user_to_ask_as_member} doesn't want " +
"to take new people in the polycule yet... can't proceed with mariage")
return
users_who_confirmed_the_marriage.append(user_to_ask)
# Everyone agrees, let's register the marriage
data = data_manager.get_data()
# Disband existing marriages with members of the new polycule to avoid issues
id_of_marriages_to_disband = []
for index, marriage in enumerate(data["marriages"]):
for user in users_who_confirmed_the_marriage:
if user in marriage:
id_of_marriages_to_disband.append(index)
for i in id_of_marriages_to_disband:
if len(data["marriages"]) <= 0:
# No marriages left, so nothing to disband!
break
data["marriages"].pop(i)
# Register the new marriage
data["marriages"].append(users_who_confirmed_the_marriage)
with data_manager.DataWriter() as writer:
writer.set_data(data)
formated_polycule_list = '\n'.join(
[f"- <@{user_id}>" for user_id in users_who_confirmed_the_marriage])
await context.respond(f"woo! a new polycule was formed! members:\n{formated_polycule_list}")
@bot.slash_command() @bot.slash_command()
async def help(ctx: discord.ApplicationContext): async def help(ctx: discord.ApplicationContext):