244 lines
10 KiB
Python
244 lines
10 KiB
Python
import discord
|
||
import asyncio
|
||
import os
|
||
from datetime import datetime
|
||
from discord.ext import commands
|
||
|
||
from .utils.dataIO import dataIO
|
||
from .utils import checks
|
||
|
||
|
||
class Immortal:
|
||
"""Creates a goodbye message when people leave"""
|
||
|
||
def __init__(self, bot):
|
||
self.bot = bot
|
||
self.path = "data/Fox-Cogs/immortal"
|
||
self.file_path = "data/Fox-Cogs/immortal/immortal.json"
|
||
self.the_data = dataIO.load_json(self.file_path)
|
||
|
||
def save_data(self):
|
||
"""Saves the json"""
|
||
dataIO.save_json(self.file_path, self.the_data)
|
||
|
||
async def adj_roles(self, server, author, member: discord.Member=None, rrole_names=[], arole_names=[]):
|
||
# Thank you SML for the addrole code
|
||
# https://github.com/smlbiobot/SML-Cogs/tree/master/mm
|
||
|
||
rroles = [r for r in server.roles if r.name in rrole_names]
|
||
aroles = [r for r in server.roles if r.name in arole_names]
|
||
try:
|
||
await self.bot.add_roles(member, *aroles)
|
||
await asyncio.sleep(0.5)
|
||
await self.bot.remove_roles(member, *rroles)
|
||
await asyncio.sleep(0.5)
|
||
|
||
except discord.Forbidden:
|
||
await self.bot.say(
|
||
"{} does not have permission to edit {}’s roles.".format(
|
||
author.display_name, member.display_name))
|
||
|
||
except discord.HTTPException:
|
||
await self.bot.say(
|
||
"Failed to adjust roles.")
|
||
except:
|
||
await self.bot.say("Unknown Exception")
|
||
|
||
|
||
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def iresort(self, ctx, member: discord.Member=None):
|
||
"""Sends someone on vacation!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Member", "Immortal", "Eternal", "Phantom", "Ghost", "Undead", "Revenant", "Crypt", "Relocate", "Guest"]
|
||
arole_names = ["Resort"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Resort" in [r.name for r in member.roles]:
|
||
await self.bot.say("You are being sent on Vacation! :tada:" +
|
||
"Please relocate to Immortal Resort (#889L92UQ) when you find the time.")
|
||
await self.bot.send_message(member, "You are being sent on Vacation! :tada: Please relocate " +
|
||
"to Immortal Resort (#889L92UQ) when you find the time.\n" +
|
||
"You'll have limited access to the server until you rejoin a main clan")
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def icrypt(self, ctx, member: discord.Member=None):
|
||
"""Sends someone to Crypt!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Immortal", "Eternal", "Ghost", "Phantom", "Revenant", "Undead", "Relocate", "Guest", "Resort"]
|
||
arole_names = ["Member", "Crypt"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Crypt" in [r.name for r in member.roles]:
|
||
await self.bot.say("Success")
|
||
await self.send_welcome(member)
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def irevenant(self, ctx, member: discord.Member=None):
|
||
"""Sends someone to Revenant!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Immortal", "Eternal", "Ghost", "Phantom", "Undead", "Crypt", "Relocate", "Guest", "Resort"]
|
||
arole_names = ["Member", "Revenant"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Revenant" in [r.name for r in member.roles]:
|
||
await self.bot.say("Success")
|
||
await self.send_welcome(member)
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def iundead(self, ctx, member: discord.Member=None):
|
||
"""Sends someone to Undead!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Immortal", "Eternal", "Ghost", "Phantom", "Revenant", "Crypt", "Relocate", "Guest", "Resort"]
|
||
arole_names = ["Member", "Undead"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Undead" in [r.name for r in member.roles]:
|
||
await self.bot.say("Success")
|
||
await self.send_welcome(member)
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def iphantom(self, ctx, member: discord.Member=None):
|
||
"""Sends someone to Phantom!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Immortal", "Eternal", "Ghost", "Undead", "Revenant", "Crypt", "Relocate", "Guest", "Resort"]
|
||
arole_names = ["Member", "Phantom"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Phantom" in [r.name for r in member.roles]:
|
||
await self.bot.say("Success")
|
||
await self.send_welcome(member)
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def ieternal(self, ctx, member: discord.Member=None):
|
||
"""Sends someone to Eternal!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Immortal", "Phantom", "Ghost", "Undead", "Revenant", "Crypt", "Relocate", "Guest", "Resort"]
|
||
arole_names = ["Member", "Eternal"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Eternal" in [r.name for r in member.roles]:
|
||
await self.bot.say("Success")
|
||
await self.send_welcome(member)
|
||
|
||
@commands.command(pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(manage_roles=True)
|
||
async def iimmortal(self, ctx, member: discord.Member=None):
|
||
"""Sends someone to Immortal!"""
|
||
|
||
if member is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
else:
|
||
server = ctx.message.server
|
||
author = ctx.message.author
|
||
role_names = ["Eternal", "Phantom", "Ghost", "Undead", "Revenant", "Crypt", "Relocate", "Guest", "Resort"]
|
||
arole_names = ["Member", "Immortal"]
|
||
await self.adj_roles(server, author, member, role_names, arole_names)
|
||
if "Immortal" in [r.name for r in member.roles]:
|
||
await self.bot.say("Success")
|
||
await self.send_welcome(member)
|
||
|
||
@commands.group(aliases=['setimmortal'], pass_context=True, no_pm=True)
|
||
@checks.mod_or_permissions(administrator=True)
|
||
async def immortalset(self, ctx):
|
||
"""Adjust immortal settings"""
|
||
|
||
server = ctx.message.server
|
||
if server.id not in self.the_data:
|
||
self.the_data[server.id] = {}
|
||
self.save_data()
|
||
|
||
if ctx.invoked_subcommand is None:
|
||
await self.bot.send_cmd_help(ctx)
|
||
|
||
@immortalset.command(pass_context=True, no_pm=True)
|
||
async def welcomechannel(self, ctx):
|
||
server = ctx.message.server
|
||
if 'WELCOMECHANNEL' not in self.the_data[server.id]:
|
||
self.the_data[server.id]['WELCOMECHANNEL'] = ''
|
||
|
||
self.the_data[server.id]['WELCOMECHANNEL'] = ctx.message.channel.id
|
||
self.save_data()
|
||
await self.bot.say("Welcome Channel set to "+ctx.message.channel.name)
|
||
|
||
async def send_welcome(self, member):
|
||
server = member.server
|
||
if server.id in self.the_data:
|
||
await self.bot.send_message(server.get_channel(self.the_data[server.id]['WELCOMECHANNEL']),
|
||
"You now have access to the server, " + member.mention + "\n" +
|
||
"Check " + server.get_channel("257557008662790145").mention + " & " +
|
||
server.get_channel("257560603093106688").mention+" for clan rules etc.\n" +
|
||
"We recommend turning all message notifications on for " + server.get_channel("257560603093106688").mention +
|
||
" if you want to know when tourneys are posted and other important info.\n" +
|
||
"You can also type `!help` for a list of bot commands/features.")
|
||
|
||
# @immortalset.command(pass_context=True)
|
||
# async def channel(self, ctx):
|
||
# server = ctx.message.server
|
||
# if 'channel' not in self.the_data[server.id]:
|
||
# self.the_data[server.id]['channel'] = ''
|
||
|
||
# self.the_data[server.id]['channel'] = ctx.message.channel.id
|
||
# self.save_data()
|
||
|
||
# async def _when_leave(self, member):
|
||
# server = member.server
|
||
# if server.id not in self.the_data:
|
||
# return
|
||
|
||
# await self.bot.say("YOU LEFT ME "+member.mention)
|
||
# self.the_data[server.id]
|
||
|
||
|
||
def check_folders():
|
||
if not os.path.exists("data/Fox-Cogs"):
|
||
print("Creating data/Fox-Cogs folder...")
|
||
os.makedirs("data/Fox-Cogs")
|
||
|
||
if not os.path.exists("data/Fox-Cogs/immortal"):
|
||
print("Creating data/Fox-Cogs/immortal folder...")
|
||
os.makedirs("data/Fox-Cogs/immortal")
|
||
|
||
|
||
def check_files():
|
||
if not dataIO.is_valid_json("data/Fox-Cogs/immortal/immortal.json"):
|
||
dataIO.save_json("data/Fox-Cogs/immortal/immortal.json", {})
|
||
|
||
|
||
def setup(bot):
|
||
check_folders()
|
||
check_files()
|
||
q = Immortal(bot)
|
||
bot.add_cog(q)
|