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.
Enso-Bot/cogs/enso.py

497 lines
20 KiB
Python

import asyncio
import datetime
import random
import string
import discord
from discord import Embed, Colour
from discord.ext import commands
from discord.ext.commands import cooldown, BucketType, command, is_owner, bot_has_permissions
from settings import colour_list, enso_guild_ID, enso_ensochancommands_Mention, blank_space, enso_embedmod_colours, \
enso_verification_ID
"""events = {
"🎤": 722483603409469470, # Karaoke Night
"🎧": 696753950879383605, # Enso Bros Podcast
"🎥": 722482922518609990, # Movie Night
"🎮": 722493033882452078, # Game Night
":GameNight:": 722493033882452078,
":EnsoBros:": 696753950879383605,
":MovieNight:": 722482922518609990,
":Karaoke:": 722483603409469470
}"""
# 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}**"
# Send a message to the channel that Enso~Chan has dm'ed them!
def helpDm():
hamothyID = '<@&715412394968350756>'
# Returning F String to send to the User
return f"I've just pinged your dms UwU! <a:huh:676195228872474643> <a:huh:676195228872474643>" \
f"\nPlease ping my owner {hamothyID} for any issues/questions you have!"
# Method to retrieve information about the user and the guild
def get_user_info(self, ctx):
# Allowing the bot to dm the user
author = ctx.author
# Define guild icon, enso bot icon and enso bot name
guild_icon = ctx.guild.icon_url
enso_icon = self.bot.user.avatar_url
enso_name = self.bot.user.display_name
return author, guild_icon, enso_icon, enso_name
# 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
# Function to display all the images requested of the people
def displayServerImage(array, ctx, name):
# Get the member and the userAvatar
member, userAvatar = getMember(ctx)
# Set embed up for the person requested by the user
embed = Embed(
title=f"**Look At What A Cutie {name.capitalize()} is! <a:huh:676195228872474643> <a:huh:676195228872474643> **",
colour=Colour(random.choice(colour_list)),
timestamp=datetime.datetime.utcnow())
embed.set_image(url=random.choice(array))
embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))
return embed
def enso_people():
"""Return all the people in the txt files"""
return ['hammy', 'hussein', 'inna', 'kate', 'calvin',
'lukas', 'stitch', 'corona', 'ging', 'ash',
'gria', 'lilu', 'ifrah', 'skye', 'chloe',
'connor', 'taz', 'ryder', 'ange', 'rin',
'izzy', 'david', 'clarity', 'angel', "studentjon"]
class Enso(commands.Cog):
"""Commands for Ensō server"""
def __init__(self, bot):
self.bot = bot
@commands.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-----")
@commands.group(invoke_without_command=True)
@bot_has_permissions(embed_links=True)
async def enso(self, ctx, name=None):
"""Shows Random Person from Ensō"""
# Making sure this command only works in Enso
if not ctx.guild.id == enso_guild_ID:
await ctx.send("**Sorry! That command is only for a certain guild!**")
return
# If the channel that the command has been sent is in the list of accepted channels
if str(ctx.channel) in "enso-chan-commands":
if name:
# Get the lowercase
lcase_name = name.lower()
try:
# Retrieve image of the member specified
with open(f'images/ServerMembers/{lcase_name}.txt') as file:
images_array = file.readlines()
# Embed the image into a message and send it to the channel
embed = displayServerImage(images_array, ctx, lcase_name)
await ctx.send(embed=embed)
except Exception as e:
print(e)
# Send the list of available members to the channel
nice = string.capwords(', '.join(map(str, enso_people())))
# Send error message saying that the person isn't recognised
await ctx.send(f"Sorry! That person doesn't exist! Try the names listed below!"
f"\n{nice}")
else:
# Retrieve a random image of a member in the bot
with open(f'images/ServerMembers/{random.choice(enso_people())}.txt') as file:
array = file.readlines()
# Get the member and the userAvatar
member, userAvatar = getMember(ctx)
# Embed the image in a message and send it to the channel
embed = Embed(
title=f"Oh Look! A Cute Person <a:huh:676195228872474643> <a:huh:676195228872474643> ",
colour=Colour(random.choice(colour_list)),
timestamp=datetime.datetime.utcnow())
embed.set_image(url=random.choice(array))
embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))
await ctx.send(embed=embed)
else:
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()
@enso.command()
async def list(self, ctx):
"""Shows the List of People in the Bot"""
# Send the list of available members to the channel
nice = string.capwords(', '.join(map(str, enso_people())))
# Send error message saying that the person isn't recognised
await ctx.send(f"Try the names listed below!"
f"\n{nice}")
@command(name="rules", aliases=["Rules"])
@cooldown(1, 5, BucketType.user)
async def rules(self, ctx):
"""Ruleset for Ensō"""
# Making sure this command only works in Enso
if not ctx.guild.id == enso_guild_ID:
await ctx.send("**Sorry! That command is only for a certain guild!**")
return
# Define Izzy's roles ID
izzyID = '<@397944038440828928>'
# Get information about the user and the guild
author, guild_icon, enso_icon, enso_name = get_user_info(self, ctx)
# Set up embed to list all the rules within the server
embed = Embed(title="(っ◔◡◔)っ Ensō Rules",
colour=enso_embedmod_colours,
description="ヽ(͡◕ ͜ʖ ͡◕)ノ Please respect the following rules that are going to be listed below ヽ(͡◕ ͜ʖ ͡◕)ノ",
timestamp=datetime.datetime.utcnow())
embed.set_thumbnail(url=guild_icon)
embed.set_author(name=enso_name,
icon_url=enso_icon)
fields = [
(blank_space,
"**➳ Don't be overly toxic/purposely problematic** \n This one is pretty self explanatory, just treat others the way you want to be treated and you'll get along with everyone :)",
False),
(blank_space,
"**➳ Respect all admins and staff** \n They are enforcing these rules to help make and keep this server a fantastic place to hang out.",
False),
(blank_space,
"**➳ Keep content organized into their respective channels** \n For example. When connected to a voice channel, all messages relating to the discussion in voice-chat should be sent in #vc-chat",
False),
(blank_space,
"**➳ No advertising other servers** \nIt's disrespectful to do that and won't be tolerated in this server",
False),
(blank_space,
"**➳ No pornographic/adult/other NSFW material** \n This is a community server and not meant to share this kind of material. Try to stay around PG 13 as most of our users are between 13 - 16",
False),
(blank_space,
"**➳ Don't take insults too far** \n Poking fun at others is okay, just don't take it too far. Any disputes can be brought up to a staff member and they will handle it." +
"\nIf you end up causing a problem or taking things into your in hands, you will be punished",
False),
(blank_space,
"**➳ Explicit Language** \n Swearing is perfectly fine as long as it's not in excess, with some exceptions of course." +
"These exceptions being racial, sexual, and ethnic slurs",
False),
(blank_space,
"**➳ Discord ToS** \n As well as following the rules we have set forth, please make sure to follow Discord's ToS https://discordapp.com/terms ",
False),
(blank_space,
"```( ͡°ω ͡°) Disciplinary Actions ( ͡°ω ͡°)```", False),
(blank_space,
"**➳ First Offense** \n Warning",
True),
(blank_space,
"**➳ Second Offense** \n1 hour mute",
True),
(blank_space,
"**➳ Third Offense** \n12 hour mute",
True),
(blank_space,
"**➳ Fourth Offense** \n24 hour mute",
True),
(blank_space,
"**➳Fifth Offense** \n Kicked from the server",
True),
(blank_space,
"**➳ Sixth Offense** \n Banned from the server",
True),
(blank_space,
"**➳ There are, of course, exceptions to these rules based on the severity of the offense Minor offenses will play out as described but major offenses will be dealt with at the discretion of the staff member involved.**",
False),
(blank_space,
f"**➳ Any disputes about a staff members choices or actions can be brought to myself, {ctx.message.author.mention} or my co-owner, {izzyID}**",
False)]
# Add fields to the embed
for name, value, inline in fields:
embed.add_field(name=name, value=value, inline=inline)
# Dm the user the embedded message
await author.send(embed=embed)
# Send the helpDm() message to the channel that the user is in
message = await ctx.send(helpDm())
# Let the user read the message for 10 seconds
await asyncio.sleep(10)
# Delete the message
await message.delete()
@command(name="roles", aliases=["Roles"])
@cooldown(1, 5, BucketType.user)
async def roles(self, ctx):
"""Leveled role/xp system for Ensō"""
if not ctx.guild.id == enso_guild_ID:
await ctx.send("**Sorry! That command is only for a certain guild!**")
return
# Get the url of the leveled roles image
roles_image = "https://media.discordapp.net/attachments/669812887564320769/717149671771996180/unknown.png"
# Setting up embedded message about the leveled roles system within the server
embed = Embed(title="```So you wanna know how the leveled roles system works huh?```",
colour=enso_embedmod_colours,
description="------------------------------------------------",
timestamp=datetime.datetime.utcnow())
# Get information about the user and the guild
author, guild_icon, enso_icon, enso_name = get_user_info(self, ctx)
embed.set_image(url=roles_image)
embed.set_thumbnail(url=guild_icon)
embed.set_author(name=enso_name,
icon_url=enso_icon)
embed.add_field(name="Cooldown", value="**•XP is gained every time you talk with a 2 minute cooldown.**",
inline=False),
embed.add_field(name="Message Length",
value="**•XP is not determined by the size of the message. You will not get more XP just because "
"the message is bigger.**",
inline=False),
embed.add_field(name="Roles",
value="**•As seen below, those are the colours and roles that will be achieved upon gaining that "
"amount of experience**",
inline=False)
# Dm the user the embedded message
await author.send(embed=embed)
# Send the helpDm() message to the channel that the user is in
message = await ctx.send(helpDm())
# Let the user read the message for 10 seconds
await asyncio.sleep(10)
# Delete the message
await message.delete()
# Allowing people to get ping-able self roles
@command(name="verification", hidden=True)
@is_owner()
async def verification(self, ctx):
# Set up embed to let the user know that they have to react with ✅
embed = Embed(title="**Verification**",
colour=Colour(0xFF69B4),
timestamp=datetime.datetime.utcnow())
embed.set_thumbnail(url=ctx.guild.icon_url)
embed.set_author(name=ctx.author, icon_url=ctx.author.avatar_url)
embed.add_field(
name="Remember to read the rules!",
value="React with ✅ to gain access to the rest of the server!",
inline=False)
# Send embed to the channel it was called in and automatically add the reaction ✅
# verif = await ctx.send(embed=embed)
# await verif.add_reaction('✅')
# Edit the Embed And Update it
verif = await ctx.fetch_message(728424149692842115)
await verif.edit(embed=embed)
# Cog listener for enabling roles to be added to users when they react to the embedded message
@commands.Cog.listener()
async def on_raw_reaction_add(self, payload):
# Get the guild
guild = self.bot.get_guild(enso_guild_ID)
# Get the member
member = guild.get_member(payload.user_id)
# Getting the channel verification by setting it to #verification
channel = guild.get_channel(enso_verification_ID)
# If the channel is #verification
if payload.channel_id == channel.id:
# A check that makes sure that the reaction is done by the bot
def check(m):
return m == self.bot.user
# If the member is not a user, do nothing
if check(payload.member):
return
else:
# Get the 'Lucid' role and then give it to the user
role = discord.utils.get(guild.roles, name='Lucid')
# if the emoji that was reacted is the tick mark.
if payload.emoji.name == "":
await member.add_roles(role)
# Set hamothyID equal to my id in discord
hamothyID = '<@&715412394968350756>'
# Set the channel id to "general"
general = guild.get_channel(663651584399507481)
# String for welcoming people in the #general channel
general_welcome = f"Welcome to the server! {member.mention} I hope you enjoy your stay here <a:huh:676195228872474643> <a:huh:676195228872474643> " \
f"\nPlease go into <#722347423913213992> to choose some ping-able roles for events! " \
f"\nPlease ping {hamothyID} for any questions about the server and of course, the other staff members!"
# Send welcome message to #general
await general.send(general_welcome)
# If the message id equals the self roles message
if payload.message_id == 722514840559812649:
# Print out the emoji name
print(payload.emoji.name)
# Find a role corresponding to the Emoji name.
guild_id = payload.guild_id
# Find the guild Enso and find the role of the emoji that has been reacted to
guild = discord.utils.find(lambda g: g.id == guild_id, self.bot.guilds)
role = discord.utils.find(lambda r: r.name == payload.emoji.name, guild.roles)
# if the role does exist
if role is not None:
# Print to me that the role was found and display the id of the role
print(role.name + " was found!")
print(role.id)
# Find the member who had reacted to the emoji
member = discord.utils.find(lambda m: m.id == payload.user_id, guild.members)
# Add the role to the member
await member.add_roles(role)
# Print to me that the role has been added
print("done")
"""# Make sure the reaction event doesn't count other channels
if not payload.channel_id == 722347423913213992:
return
role = payload.member.guild.get_role(events.get(payload.emoji.name))
await payload.member.add_roles(role)
print(f"{payload.member.name} Was Given Role {role}")"""
# Cog listener for enabling roles to be removed from users when they unreact to the embedded messaged
@commands.Cog.listener()
async def on_raw_reaction_remove(self, payload):
# If the message id equals the self roles message
if payload.message_id == 722514840559812649:
# Print out the emoji name
print(payload.emoji.name)
# Get the server id
guild_id = payload.guild_id
# Find the guild Enso and find the role of the emoji that has been unreacted to
guild = discord.utils.find(lambda g: g.id == guild_id, self.bot.guilds)
role = discord.utils.find(lambda r: r.name == payload.emoji.name, guild.roles)
# if the role does exist
if role is not None:
# Find the member that has the role which the emoji is connected to
member = discord.utils.find(lambda m: m.id == payload.user_id, guild.members)
# Remove the role from the member
await member.remove_roles(role)
"""# Make sure the reaction event doesn't count other channels
if not payload.channel_id == 722347423913213992:
return
guild = self.bot.get_guild(payload.guild_id)
member = guild.get_member(payload.user_id)
role = guild.get_role(events.get(payload.emoji.name))
await member.remove_roles(role)
print(f"{member.name} Was Removed from Role {role}")"""
# Allowing people to get ping-able self roles
@command(name="rolemenu", hidden=True)
@is_owner()
async def role_menu(self, ctx):
# Setting the channel to "
channel = ctx.guild.get_channel(722347423913213992)
# Set up embed to let people know what ping-able roles can be chosen
embed = Embed(title="**Role Menu: Ping-Able Roles**",
colour=Colour.orange(),
timestamp=datetime.datetime.utcnow())
embed.set_thumbnail(url=ctx.guild.icon_url)
embed.set_author(name=ctx.author, icon_url=ctx.author.avatar_url)
embed.set_footer(text=f"{ctx.message.author}", icon_url=ctx.author.avatar_url)
embed.add_field(
name="\u200b",
value="React to give yourself roles to be pinged for these events!",
inline=False)
embed.add_field(
name="\u200b",
value="🎥 : `Movie Nights`",
inline=False)
embed.add_field(
name="\u200b",
value="🎤 : `Karaoke Nights`",
inline=False)
embed.add_field(
name="\u200b",
value="🎧 : `Enso Bros Podcasts`",
inline=False)
embed.add_field(
name="\u200b",
value="🎮 : `Game Nights`",
inline=False)
# Edit the Embed And Update it
message = await ctx.fetch_message(722514840559812649)
await message.edit(embed=embed)
# Send the embed to the channel "newpeople"
# await channel.send(embed=embed)
def setup(bot):
bot.add_cog(Enso(bot))