5dfe9cbd8e
* Update contributors.py * Update contributors.py * Added myself as a contributor * Rename cogs.txt to extensions.txt * Update contributors.py
120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from discord.ext import commands
|
|
import discord
|
|
import traceback
|
|
import aiofiles
|
|
|
|
class Upload:
|
|
"""
|
|
CogName should be the name of the cog
|
|
"""
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
print('upload loaded')
|
|
|
|
@commands.command()
|
|
async def reload(self, ctx, *, extension: str):
|
|
"""Reload an extension."""
|
|
await ctx.trigger_typing()
|
|
if ctx.author.id not in self.bot.ownerlist:
|
|
return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)
|
|
|
|
extension = extension.lower()
|
|
try:
|
|
self.bot.unload_extension("src.cogs.{}".format(extension))
|
|
self.bot.load_extension("src.cogs.{}".format(extension))
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
await ctx.send(f'Could not reload `{extension}` -> `{e}`')
|
|
else:
|
|
await ctx.send(f'Reloaded `{extension}`.')
|
|
|
|
@commands.command()
|
|
async def reloadall(self, ctx):
|
|
"""Reload all extensions."""
|
|
await ctx.trigger_typing()
|
|
if ctx.author.id not in self.bot.ownerlist:
|
|
return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)
|
|
|
|
try:
|
|
for extension in self.bot.extensions:
|
|
self.bot.unload_extension(extension)
|
|
self.bot.load_extension(extension)
|
|
await ctx.send(f"Reload success! :thumbsup:\n")
|
|
except Exception as e:
|
|
await ctx.send(f"Could not reload `{extension}` -> `{e}`.\n")
|
|
|
|
@commands.command()
|
|
async def unload(self, ctx, *, extension: str):
|
|
"""Unload an extension."""
|
|
await ctx.trigger_typing()
|
|
if ctx.author.id not in self.bot.ownerlist:
|
|
return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)
|
|
|
|
extension = extension.lower()
|
|
try:
|
|
self.bot.unload_extension("src.cogs.{}".format(extension))
|
|
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
if ctx.message.author.id not in self.bot.owner_list:
|
|
await ctx.send(f'Could not unload `{extension}` -> `{e}`')
|
|
|
|
else:
|
|
await ctx.send(f'Unloaded `{extension}`.')
|
|
|
|
@commands.command()
|
|
async def load(self, ctx, *, extension: str):
|
|
"""Load an extension."""
|
|
await ctx.trigger_typing()
|
|
if ctx.author.id not in self.bot.ownerlist:
|
|
return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)
|
|
|
|
extension = extension.lower()
|
|
try:
|
|
self.bot.load_extension("src.cogs.{}".format(extension))
|
|
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
await ctx.send(f'Could not unload `{extension}` -> `{e}`')
|
|
else:
|
|
await ctx.send(f'Loaded `{extension}`.')
|
|
|
|
@commands.command()
|
|
async def permunload(self, ctx, extension=None):
|
|
"""Disables permanently a cog."""
|
|
await ctx.trigger_typing()
|
|
if ctx.author.id not in self.bot.ownerlist:
|
|
return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)
|
|
|
|
if cog is None:
|
|
return await ctx.send("Please provide a extension. Do `help permunload` for more info")
|
|
|
|
extension = extension.lower()
|
|
|
|
async with aiofiles.open("extension.txt") as fp:
|
|
lines=fp.readlines()
|
|
|
|
removed = False
|
|
async with aiofiles.open("extension.txt", "w") as fp:
|
|
for i in lines:
|
|
if i.replace("\n", "") != extension:
|
|
fp.write(i)
|
|
else:
|
|
removed = True
|
|
break
|
|
|
|
if removed is True:
|
|
try:
|
|
self.bot.unload_extension(extension)
|
|
except:
|
|
pass
|
|
return await ctx.send("Extension removed successfully")
|
|
|
|
await ctx.send("Extension not found")
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Upload(bot))
|