Enso-Bot/cogs/anime/interactive.py

import asyncio
import random

from discord import Colour, Embed, Member
from discord.ext import commands
from discord.ext.commands import cooldown, command, BucketType

import db
import settings
from settings import time, colour_list, enso_ensochancommands_Mention, enso_guild_ID


# Gets the member and user avatar
def getMember(ctx):
    # Set member as the author
    member = ctx.message.author
    # Get the member avatar
    userAvatar = member.avatar_url

    return member, userAvatar


# Error handling function to make sure that the commands only work in "enso-chan-commands"
def error_function():
    return f"**Sorry! I only work in {enso_ensochancommands_Mention}**"


# Set up the Cog
class Interactive(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @command(name="kiss", aliases=["Kiss"])
    @cooldown(1, 1, BucketType.user)
    async def kiss(self, ctx, target: Member):
        """Allows users to kiss a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the kissing gifs
                with open('images/FunCommands/kissing.txt') as file:
                    # Store content of the file in kissing_array
                    kissing_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random kissing gif
                embed = Embed(
                    title=f"<a:huh:676195228872474643> <a:huh:676195228872474643> | **{member.display_name}** kissed **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(kissing_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="kill", aliases=["Kill"])
    @cooldown(1, 1, BucketType.user)
    async def kill(self, ctx, target: Member):
        """Allows users to kill a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the killing gifs
                with open('images/FunCommands/killing.txt') as file:
                    # Store content of the file in killing_array
                    killing_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random killing gif
                embed = Embed(
                    title=f"<:monkaW:718960264896184380> <:monkaW:718960264896184380> | **{member.display_name}** killed **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(killing_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="cuddle", aliases=["Cuddle"])
    @cooldown(1, 1, BucketType.user)
    async def cuddle(self, ctx, target: Member):
        """Allows users to cuddle a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the cuddling gifs
                with open('images/FunCommands/cuddling.txt') as file:
                    # Store content of the file in cuddling_array
                    cuddling_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random cuddling gif
                embed = Embed(
                    title=f"<:blushlook1:677310734123663363> <:blushlook2:679524467248201769> | **{member.display_name}** cuddled **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(cuddling_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="slap", aliases=["Slap"])
    @cooldown(1, 1, BucketType.user)
    async def slap(self, ctx, target: Member):
        """Allows users to slap a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the cuddling gifs
                with open('images/FunCommands/slapping.txt') as file:
                    # Store content of the file in cuddling_array
                    slapping_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random slapping gif
                embed = Embed(
                    title=f"<:baka:718942872061083678> <:baka:718942872061083678> | **{member.display_name}** slapped **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(slapping_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="pat", aliases=["Pat"])
    @cooldown(1, 1, BucketType.user)
    async def pat(self, ctx, target: Member):
        """Allows users to cuddle a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the patting gifs
                with open('images/FunCommands/patting.txt') as file:
                    # Store content of the file in patting_array
                    patting_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random patting gif
                embed = Embed(
                    title=f"<:xoxo:679893117482303564> <:xoxo:679893117482303564> | **{member.display_name}** patted **{target.display_name} on the head**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(patting_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="lemon", aliases=["Lemon"])
    @cooldown(1, 1, BucketType.user)
    async def lemon(self, ctx, target: Member):
        """Allows users to hand people lemons to members in the user"""

        lemon_array = ["https://media.discordapp.net/attachments/669812887564320769/720093589056520202/lemon.gif",
                       "https://media.discordapp.net/attachments/669812887564320769/720093575492272208/lemon2.gif",
                       "https://media.discordapp.net/attachments/718484280925224981/719629805263257630/lemon.gif"]

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random lemon gif
                embed = Embed(
                    title=f"<a:huh:676195228872474643> <a:huh:676195228872474643> | **{member.display_name}** Gives A Lemon To **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(lemon_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

                # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="choke", aliases=["Choke"])
    @cooldown(1, 1, BucketType.user)
    async def choke(self, ctx, target: Member):
        """Allows users to choke a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the choking gifs
                with open('images/FunCommands/choking.txt') as file:
                    # Store content of the file in choking_array
                    choking_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random choking gif
                embed = Embed(
                    title=f"<:qmq:676203031506976768> <:qmq:676203031506976768> | **{member.display_name}** choked **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(choking_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="hug", aliases=["Hug"])
    @cooldown(1, 1, BucketType.user)
    async def hug(self, ctx, target: Member):
        """Allows users to hug a person in the server"""

        # Surround with try/except to catch any exceptions that may occur
        try:

            # If the channel that the command has been sent is in the list of accepted channels
            if str(ctx.channel) in settings.channels:

                # Open the file containing the hug gifs
                with open('images/FunCommands/hugging.txt') as file:
                    # Store content of the file in hugging_array
                    hugging_array = file.readlines()

                # Get the member and the userAvatar
                member, userAvatar = getMember(ctx)

                # Set up the embed to display a random hugging gif
                embed = Embed(
                    title=f"<a:huh:676195228872474643> <a:huh:676195228872474643> | **{member.display_name}** hugged **{target.display_name}**",
                    colour=Colour(int(random.choice(colour_list))),
                    timestamp=time)
                embed.set_image(url=random.choice(hugging_array))
                embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))

                # Send the embedded message to the user
                await ctx.send(embed=embed)

            # else the command is sent in an invalid channel
            else:
                # Call error_function() and display it to the user
                message = await ctx.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

        except FileNotFoundError as e:
            print(e)

    @command(name="marry", aliases=["Marry"])
    @cooldown(1, 1, BucketType.user)
    async def marry(self, ctx, member: Member):
        """Allows the bot to wed two young lovers together"""

        # 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 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??** "
                       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
            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
            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:

                    # 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's
                    cursor.execute(update_query, divorcer)
                    cursor.execute(update_query, divorcee)
                    conn.commit()
                    print(cursor.rowcount, "Records Updated in Members!")

                # Congratulate them!
                await ctx.send(
                    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"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")

        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")


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
"""