mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
465 lines
19 KiB
Python
465 lines
19 KiB
Python
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"""
|
|
|
|
# Get the guild Enso
|
|
guild = self.bot.get_guild(enso_guild_ID)
|
|
|
|
# Using connection to the database
|
|
with db.connection() as conn:
|
|
select_query = """SELECT * FROM marriages"""
|
|
cursor = conn.cursor()
|
|
|
|
cursor.execute(select_query)
|
|
result = cursor.fetchall()
|
|
|
|
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}!")
|
|
|
|
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()
|
|
|
|
# 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:
|
|
|
|
# Define the Insert Into Statement inserting into the database
|
|
insert_query = """INSERT INTO marriages (proposerID, proposeeID) VALUES (?, ?)"""
|
|
val = ctx.author.id, member.id
|
|
cursor = conn.cursor()
|
|
|
|
# Execute the SQL Query
|
|
cursor.execute(insert_query, val)
|
|
conn.commit()
|
|
print(cursor.rowcount, "Record inserted successfully into Marriage Logs")
|
|
|
|
# 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"Unlucky (Ｔ▽Ｔ), maybe another time! {ctx.author.mention}")
|
|
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("Awww they waited too long (✖╭╮✖)")
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Interactive(bot))
|