Making marry command store information

Storing Married Users
Storing the time of the Marriage
Making divorce command

cogs/anime/interactive.py

@@ -378,40 +378,34 @@ class Interactive(commands.Cog):
     async def marry(self, ctx, member: Member):
         """Allows the bot to wed two young lovers together"""
 
-        # Get the guild Enso
-        guild = self.bot.get_guild(enso_guild_ID)
+        # Getting the guild of the user
+        guild = ctx.author.guild
 
-        # Using connection to the database
+        # Make sure the guild is Enso
+        if guild.id != enso_guild_ID:
+            return
+
+        # Use database connection
         with db.connection() as conn:
-            select_query = """SELECT * FROM marriages"""
-            cursor = conn.cursor()
 
-            cursor.execute(select_query)
-            result = cursor.fetchall()
+            # Get the author's row from the Members Table
+            select_query = """SELECT * FROM members WHERE discordID = (?)"""
+            val = ctx.author.id,
+            cursor = conn.cursor()
 
-            for row in result:
-                if str(ctx.author.id) != row[1] or str(member.id) == row[2]:
-                    member = guild.get_member(int(row[2]))
-                    author = guild.get_member(int(row[1]))
-                    await ctx.send(f"Sorry! {member.mention} is already married to {author.mention}!")
+            # Execute the SQL Query
+            cursor.execute(select_query, val)
+            result = cursor.fetchone()
 
-                if str(ctx.author.id) == row[1]:
-                    member = guild.get_member(int(row[2]))
-                    await ctx.send(f"Sorry! You're currently married to {member.mention}")
-                    return
-                elif str(ctx.author.id) == row[2]:
-                    member = guild.get_member(int(row[1]))
-                    await ctx.send(f"Sorry! You're currently married to {member.mention}")
-                    return
-                elif str(member.id) == row[1]:
-                    member = guild.get_member(int(row[2]))
-                    await ctx.send(f"Sorry! You're currently married to {ctx.author.mention}")
-                    return
-                elif str(member.id) == row[2]:
-                    member = guild.get_member(int(row[1]))
-                    await ctx.send(f"Sorry! You're currently married to {ctx.author.mention}")
-                    return
-            conn.commit()
+            # Make sure that the user cannot marry themselves
+            if member.id == ctx.author.id:
+                await ctx.send("Senpaii! ˭̡̞(◞⁎˃ᆺ˂)◞*✰ You can't possibly marry yourself!")
+                return
+            # Make sure that the person is not already married to someone else within the server
+            elif result[2] is not None:
+                member = guild.get_member(int(result[2]))
+                await ctx.send(f"((╬◣﹏◢)) You're already married to {member.mention}!")
+                return
 
         # Send a message to the channel mentioning the author and the person they want to wed.
         await ctx.send(f"{ctx.author.mention} **Proposes To** {member.mention} **Do you accept??** "
@@ -422,6 +416,97 @@ class Interactive(commands.Cog):
         def check(m):
             return m.author == member and m.channel == ctx.channel
 
+        # Surround with try/except to catch any exceptions that may occur
+        try:
+            # Wait for the message from the mentioned user
+            msg = await self.bot.wait_for('message', check=check, timeout=30)
+
+            # if the person says yes
+            if msg.content.lower() in ['y', 'yes', 'yea']:
+                # Using connection to the database
+                with db.connection() as conn:
+                    message_time = msg.created_at.strftime("%a, %b %d, %Y")
+
+                    # Update the existing records in the database with the user that they are marrying along with the time of the accepted proposal
+                    update_query = """UPDATE members SET married = (?), marriedDate = (?) WHERE discordID = (?)"""
+                    proposer = member.id, message_time, ctx.author.id,
+                    proposee = ctx.author.id, message_time, member.id,
+                    cursor = conn.cursor()
+
+                    # Execute the SQL Query's
+                    cursor.execute(update_query, proposer)
+                    cursor.execute(update_query, proposee)
+                    conn.commit()
+                    print(cursor.rowcount, "Records Updated in Members!")
+
+                # Congratulate them!
+                await ctx.send(
+                    f"Congratulations! ｡ﾟ( ﾟ^∀^ﾟ)ﾟ｡ {ctx.author.mention} and {member.mention} are now married to each other!")
+
+            # if the person says no
+            elif msg.content.lower() in ['n', 'no', 'nah']:
+
+                # Try to console the person and wish them the best in their life
+                await ctx.send(f"{ctx.author.mention} It's okay King. Pick up your crown and move on (◕‿◕✿)")
+            else:
+                # Abort the process as the message sent did not make sense
+                await ctx.send("Senpaiiii! (｡╯︵╰｡) Speak English Please")
+
+        except asyncio.TimeoutError as ex:
+            print(ex)
+
+            # Send out an error message if the user waited too long
+            await ctx.send("(｡T ω T｡) They waited too long")
+
+    @command(name="divorce", aliases=["Divorce"])
+    @cooldown(1, 1, BucketType.user)
+    async def divorce(self, ctx, member: Member):
+        """Allows the bot to divorce users"""
+
+        # Getting the guild of the user
+        guild = ctx.author.guild
+
+        # Make sure the guild is Enso
+        if guild.id != enso_guild_ID:
+            return
+
+        # Use database connection
+        with db.connection() as conn:
+
+            # Get the author's row from the Members Table
+            select_query = """SELECT * FROM members WHERE discordID = (?)"""
+            val = ctx.author.id,
+            cursor = conn.cursor()
+
+            # Execute the SQL Query
+            cursor.execute(select_query, val)
+            result = cursor.fetchone()
+
+            # Make sure that the user cannot divorce themselves
+            if member.id == ctx.author.id:
+                await ctx.send("Senpaii! ˭̡̞(◞⁎˃ᆺ˂)◞*✰ You can't possibly divorce yourself!")
+                return
+            # Make sure that the person trying to divorce is actually married to the user
+            elif result[2] is None:
+                await ctx.send(f"((╬◣﹏◢)) You must be married in order to divorce someone! Baka!")
+                return
+            # Make sure the person is married to the person that they're trying to divorce
+            elif result[2] != str(member.id):
+                member = guild.get_member(int(result[2]))
+                await ctx.send(f"(ノ ゜口゜)ノ You can only divorce the person that you're married!"
+                               f"\n That person is {member.mention}")
+                return
+
+        # Send a message to the channel mentioning the author and the person they want to wed.
+        await ctx.send(
+            f"{ctx.author.mention} **Wishes to Divorce** {member.mention} **Are you willing to break this sacred bond??**"
+            f"\nRespond with [**Y**es/**N**o]")
+
+        # A check that makes sure that the reply is not from the author
+        # and that the reply is in the same channel as the proposal
+        def check(m):
+            return m.author == member and m.channel == ctx.channel
+
         # Surround with try/except to catch any exceptions that may occur
         try:
             # Wait for the message from the mentioned user
@@ -432,33 +517,71 @@ class Interactive(commands.Cog):
                 # Using connection to the database
                 with db.connection() as conn:
 
-                    # Define the Insert Into Statement inserting into the database
-                    insert_query = """INSERT INTO marriages (proposerID, proposeeID) VALUES (?, ?)"""
-                    val = ctx.author.id, member.id
+                    # Update the existing records in the database with the user that they are marrying along with the time of the accepted proposal
+                    update_query = """UPDATE members SET married = null, marriedDate = null WHERE discordID = (?)"""
+                    divorcer = ctx.author.id,
+                    divorcee = member.id,
                     cursor = conn.cursor()
 
-                    # Execute the SQL Query
-                    cursor.execute(insert_query, val)
+                    # Execute the SQL Query's
+                    cursor.execute(update_query, divorcer)
+                    cursor.execute(update_query, divorcee)
                     conn.commit()
-                    print(cursor.rowcount, "Record inserted successfully into Marriage Logs")
+                    print(cursor.rowcount, "Records Updated in Members!")
 
                 # Congratulate them!
                 await ctx.send(
-                    f"Congratulations! ヽ(・∀・)ﾉ {ctx.author.mention} and {member.mention} are now married to each other!")
+                    f" ૮( ´⁰▱๋⁰ )ა {ctx.author.mention} and {member.mention} are now divorced. I hope you two can find happiness in life with other people")
+
             # if the person says no
             elif msg.content.lower() in ['n', 'no', 'nah']:
+
                 # Try to console the person and wish them the best in their life
-                await ctx.send(f"Unlucky (Ｔ▽Ｔ), maybe another time! {ctx.author.mention}")
+                await ctx.send(f"Sorry but you're gonna need {ctx.author.mention}'s consent to move forward with this!")
+
             else:
                 # Abort the process as the message sent did not make sense
-                await ctx.send("Senpaiiii! Speak English Please ⋋_⋌")
+                await ctx.send("Senpaiiii! (｡╯︵╰｡) Speak English Please")
 
         except asyncio.TimeoutError as ex:
             print(ex)
 
             # Send out an error message if the user waited too long
-            await ctx.send("Awww they waited too long (✖╭╮✖)")
+            await ctx.send("(｡T ω T｡) They waited too long")
 
 
 def setup(bot):
     bot.add_cog(Interactive(bot))
+
+
+"""
+                if str(ctx.author.id) == row[1] and str(member.id) == row[2]:
+                    member = guild.get_member(int(row[2]))
+                    await ctx.send(f"You and {member.mention} are married!")
+                    return
+                elif str(ctx.author.id) != row[1] and str(member.id) == row[2]:
+                    member = guild.get_member(int(row[2]))
+                    author = guild.get_member(int(row[1]))
+                    await ctx.send(f"Sorry! {member.mention} is already married to {author.mention}!")
+                elif str(ctx.author.id) != row[1] and str(member.id) == row[1]:
+                    member = guild.get_member(int(row[2]))
+                    author = guild.get_member(int(row[1]))
+                    await ctx.send(f"Sorry! {author.mention} is already married to {member.mention}!")
+                    return
+                elif str(ctx.author.id) == row[1]:
+                    member = guild.get_member(int(row[2]))
+                    await ctx.send(f"Sorry! You're currently married to {member.mention}")
+                    return
+                elif str(ctx.author.id) == row[2]:
+                    member = guild.get_member(int(row[1]))
+                    await ctx.send(f"Sorry! You're currently married to {member.mention}")
+                    return
+                elif str(member.id) == row[1]:
+                    member = guild.get_member(int(row[2]))
+                    await ctx.send(f"Sorry! You're currently married to {ctx.author.mention}")
+                    return
+                elif str(member.id) == row[2]:
+                    member = guild.get_member(int(row[1]))
+                    await ctx.send(f"Sorry! You're currently married to {ctx.author.mention}")
+                    return
+"""