4f5a1b518a
- Gave the package a descriptive name.
- Passed over with black once more.
- Created setup.py to install dependencies.
- Updated author to reflect repo ownership to Dusty.
- Changed `git` command to use the __url__ attribute.
- Changed music to use ogg vorbis instead of mp3, purely for
performance.
- Tried to make sure nothing broke.
- Updated dockerfile. Pretty sure we don't need it though...
147 lines
4.6 KiB
Python
147 lines
4.6 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from discord.ext import commands
|
|
import discord
|
|
import traceback
|
|
import aiofiles
|
|
import os
|
|
|
|
|
|
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 load `{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")
|
|
|
|
@commands.command(hidden=True)
|
|
async def reboot(self, ctx):
|
|
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
|
|
)
|
|
await ctx.send("Sebi-Machine is restarting.")
|
|
with open(f"src/config/reboot", "w") as f:
|
|
f.write(f"1\n{ctx.channel.id}")
|
|
# noinspection PyProtectedMember
|
|
os._exit(1)
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Upload(bot))
|