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.
393 lines
14 KiB
Python
393 lines
14 KiB
Python
# Ensō~Chan - A Multi Purpose Discord Bot That Has Everything Your Server Needs!
|
|
# Copyright (C) 2020 Goudham Suresh
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import datetime
|
|
import random
|
|
|
|
import asyncpg
|
|
from discord import Embed, Member
|
|
from discord.ext.commands import cooldown, command, BucketType, bot_has_permissions, Cog
|
|
|
|
|
|
# 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
|
|
|
|
|
|
# Set up the Cog
|
|
class Interactive(Cog):
|
|
"""Interactive Commands! (E.G Kiss/Hug/Cuddle)"""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@Cog.listener()
|
|
async def on_ready(self):
|
|
"""Printing out that Cog is ready on startup"""
|
|
print(f"{self.__class__.__name__} Cog has been loaded\n-----")
|
|
|
|
# TODO: MAKE EVERYTHING IN HERE ASYNCHRONOUS !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
|
|
|
@command(name="kiss")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def kiss(self, ctx, member: Member):
|
|
"""Kiss your partner"""
|
|
|
|
# Get the guild
|
|
guild = ctx.guild
|
|
|
|
# Error handling to make sure that the user can kiss themselves
|
|
if member.id == ctx.author.id:
|
|
kiss = False
|
|
title = f":kissing_heart: :kissing_heart: | **{ctx.author.display_name}** kissed **themselves**"
|
|
else:
|
|
kiss = True
|
|
title = f":kissing_heart: :kissing_heart: | **{ctx.author.display_name}** kissed **{member.display_name}**"
|
|
|
|
# Setup pool connection
|
|
pool = self.bot.db
|
|
async with pool.acquire() as conn:
|
|
|
|
# Get the author's row from the members table
|
|
try:
|
|
select_query = """SELECT * FROM members WHERE member_id = $1 and guild_id = $2"""
|
|
result = await conn.fetchrow(select_query, ctx.author.id, guild.id)
|
|
|
|
# Catch errors
|
|
except asyncpg.PostgresError as e:
|
|
print("PostGres Error: Member Record Could Not Be Retrieved For Kiss Command", e)
|
|
|
|
# Checking conditions to make sure user is married/kissing their partner
|
|
else:
|
|
married_user = result["married"]
|
|
|
|
if married_user is None and kiss:
|
|
await ctx.send("Σ(‘◉⌓◉’) You need to be married in order to use this command! Baka!")
|
|
return
|
|
elif not member.id == married_user and kiss:
|
|
await ctx.send("Σ(‘◉⌓◉’) You can only kiss your partner! Baka!")
|
|
return
|
|
|
|
# Release connection back to pool
|
|
finally:
|
|
await pool.release(conn)
|
|
|
|
try:
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(kissing_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="cuddle")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def cuddle(self, ctx, member: Member):
|
|
"""Cuddle your partner"""
|
|
|
|
# Get the guild
|
|
guild = ctx.guild
|
|
|
|
# Error handling to make sure that the user can cuddle themselves
|
|
if member.id == ctx.author.id:
|
|
cuddle = False
|
|
title = f":blush: :blush: | **{ctx.author.display_name}** cuddled **themselves**"
|
|
else:
|
|
cuddle = True
|
|
title = f":blush: :blush: | **{ctx.author.display_name}** cuddled **{member.display_name}**"
|
|
|
|
# Setup pool connection
|
|
pool = self.bot.db
|
|
async with pool.acquire() as conn:
|
|
|
|
# Get the author's row from the members table
|
|
try:
|
|
select_query = """SELECT * FROM members WHERE member_id = $1 and guild_id = $2"""
|
|
result = await conn.fetchrow(select_query, ctx.author.id, guild.id)
|
|
|
|
# Catch errors
|
|
except asyncpg.PostgresError as e:
|
|
print("PostGres Error: Member Record Could Not Be Retrieved For Cuddle Command", e)
|
|
|
|
# Checking conditions to make sure user is married/cuddling their partner
|
|
else:
|
|
married_user = result["married"]
|
|
|
|
if married_user is None and cuddle:
|
|
await ctx.send("Σ(‘◉⌓◉’) You need to be married in order to use this command! Baka!")
|
|
return
|
|
elif not member.id == married_user and cuddle:
|
|
await ctx.send("Σ(‘◉⌓◉’) You can only cuddle your partner! Baka!")
|
|
return
|
|
|
|
try:
|
|
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(cuddling_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="kill")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def kill(self, ctx, member: Member):
|
|
"""Kill a member"""
|
|
|
|
if member is ctx.author:
|
|
title = f":scream: :scream: | **{ctx.author.display_name}** killed **themselves**"
|
|
else:
|
|
title = f":scream: :scream: | **{ctx.author.display_name}** killed **{member.display_name}**"
|
|
|
|
try:
|
|
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(killing_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="slap")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def slap(self, ctx, member: Member):
|
|
"""Slap a member"""
|
|
|
|
if member is ctx.author:
|
|
title = f":cold_sweat: :cold_sweat: | **{ctx.author.display_name}** slapped **themselves**"
|
|
else:
|
|
title = f":cold_sweat: :cold_sweat: | **{ctx.author.display_name}** slapped **{member.display_name}**"
|
|
|
|
try:
|
|
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(slapping_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="pat")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def pat(self, ctx, member: Member):
|
|
"""Pat a member"""
|
|
|
|
if member is ctx.author:
|
|
title = f"👉 👈 | **{ctx.author.display_name}** patted **themselves**"
|
|
else:
|
|
title = f"👉 👈 | **{ctx.author.display_name}** patted **{member.display_name}**"
|
|
|
|
try:
|
|
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(patting_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="lemon")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def lemon(self, ctx, member: Member):
|
|
"""Give Lemon to member"""
|
|
|
|
if member is ctx.author:
|
|
title = f":relaxed: :relaxed: | **{ctx.author.display_name}** gave a lemon to **themselves**"
|
|
else:
|
|
title = f":relaxed: :relaxed: | **{ctx.author.display_name}** gave a lemon to **{member.display_name}**"
|
|
|
|
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"]
|
|
|
|
try:
|
|
|
|
# Get the member and the userAvatar
|
|
member, userAvatar = getMember(ctx)
|
|
|
|
# Set up the embed to display a random lemon gif
|
|
embed = Embed(
|
|
title=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(lemon_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="choke")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def choke(self, ctx, member: Member):
|
|
"""Choke a member"""
|
|
|
|
if member is ctx.author:
|
|
title = f":confounded: :confounded: | **{ctx.author.display_name}** choked **themselves**"
|
|
else:
|
|
title = f":confounded: :confounded: | **{ctx.author.display_name}** choked **{member.display_name}**"
|
|
|
|
try:
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(choking_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
@command(name="hug")
|
|
@bot_has_permissions(embed_links=True)
|
|
@cooldown(1, 3, BucketType.user)
|
|
async def hug(self, ctx, member: Member):
|
|
"""Hug a member"""
|
|
|
|
if member is ctx.author:
|
|
title = f":smiling_face_with_3_hearts: :smiling_face_with_3_hearts: | **{ctx.author.display_name}** hugged **themselves**"
|
|
else:
|
|
title = f":smiling_face_with_3_hearts: :smiling_face_with_3_hearts: | **{ctx.author.display_name}** hugged **{member.display_name}**"
|
|
|
|
try:
|
|
|
|
# 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=title,
|
|
colour=self.bot.random_colour(),
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_image(url=random.choice(hugging_array))
|
|
embed.set_footer(text=f"Requested by {member}", icon_url=userAvatar)
|
|
|
|
# Send the embedded message to the user
|
|
await ctx.send(embed=embed)
|
|
|
|
except FileNotFoundError as e:
|
|
print(e)
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Interactive(bot))
|