flifloo/Administrator-py
Archived
1
0
Fork 0
Code Activity

Integrate backup & restore extensions

Browse source
This commit is contained in:
Ethanell 2020-07-30 11:56:31 +02:00
parent c59a89e5ac
commit 80c012438e
4 changed files with 154 additions and 125 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
extensions/__init__.py
View file

@ -8,3 +8,5 @@ bot.load_extension("extensions.reminder")
bot.load_extension("extensions.greetings")
bot.load_extension("extensions.presentation")
bot.load_extension("extensions.rorec")
bot.load_extension("extensions.backup")
bot.load_extension("extensions.restore")

36
extensions/backup.py
View file

@ -1,18 +1,30 @@
from discord.ext import commands
from backup_bot.logger import logger
from os.path import isdir
from os import mkdir
import shelve
from datetime import datetime
from discord import File, Embed
from collections import OrderedDict
from datetime import datetime
from os import mkdir
from os.path import isdir
from discord import File, Embed
from discord.ext import commands
from administrator.logger import logger
extension_name = "backup"
logger = logger.getChild(extension_name)
@commands.command("backup")
async def backup_cmd(ctx: commands.Context):
class Backup(commands.Cog):
def __init__(self, bot: commands.Bot):
self.bot = bot
def description(self):
return "Backup all message on the guild"
@commands.group("backup", pass_context=True)
@commands.guild_only()
@commands.has_permissions(manage_messages=True)
async def backup(self, ctx: commands.Context):
embed = Embed(title="Backup", description="In progress... \N{hourglass}")
msg = await ctx.send(embed=embed)
file_name = f"backup/{datetime.now().strftime('%d-%m-%Y %H:%M')}"
@ -62,23 +74,23 @@ async def backup_cmd(ctx: commands.Context):
await ctx.send(file=File(file_name + ".db", "backup.db"))
def setup(bot: commands.Bot):
def setup(bot):
logger.info(f"Loading...")
if not isdir("backup"):
logger.info(f"Create backup folder")
mkdir("backup")
try:
bot.add_command(backup_cmd)
bot.add_cog(Backup(bot))
except Exception as e:
logger.error(f"Error loading: {e}")
else:
logger.info(f"Load successful")
def teardown(bot: commands.Bot):
def teardown(bot):
logger.info(f"Unloading...")
try:
bot.remove_command("backup")
bot.remove_cog("Backup")
except Exception as e:
logger.error(f"Error unloading: {e}")
else:

34
extensions/restore.py
View file

@ -1,17 +1,29 @@
from discord.ext import commands
from backup_bot.logger import logger
from os.path import isdir
from os import mkdir, remove
import shelve
from requests import get
from os import mkdir, remove
from os.path import isdir
from discord import Embed
from discord.ext import commands
from requests import get
from administrator.logger import logger
extension_name = "restore"
logger = logger.getChild(extension_name)
@commands.command("restore")
async def restore_cmd(ctx: commands.Context):
class Restore(commands.Cog):
def __init__(self, bot: commands.Bot):
self.bot = bot
def description(self):
return "Restore a backup of guild's messages"
@commands.group("restore", pass_context=True)
@commands.guild_only()
@commands.has_permissions(manage_messages=True)
async def restore(self, ctx: commands.Context):
if len(ctx.message.attachments) != 1:
await ctx.send("No backup file given ! \N{cross mark}")
else:
@ -73,23 +85,23 @@ async def restore_cmd(ctx: commands.Context):
await msg.edit(embed=embed)
def setup(bot: commands.Bot):
def setup(bot):
logger.info(f"Loading...")
if not isdir("backup"):
logger.info(f"Create backup folder")
mkdir("backup")
try:
bot.add_command(restore_cmd)
bot.add_cog(Restore(bot))
except Exception as e:
logger.error(f"Error loading: {e}")
else:
logger.info(f"Load successful")
def teardown(bot: commands.Bot):
def teardown(bot):
logger.info(f"Unloading...")
try:
bot.remove_command("restore")
bot.remove_cog("Restore")
except Exception as e:
logger.error(f"Error unloading: {e}")
else:

3
requirements.txt
View file

@ -1,11 +1,14 @@
aiohttp==3.6.2
async-timeout==3.0.1
attrs==19.3.0
certifi==2020.6.20
chardet==3.0.4
discord==1.0.1
discord.py==1.3.4
idna==2.10
multidict==4.7.6
requests==2.24.0
SQLAlchemy==1.3.18
urllib3==1.25.10
websockets==8.1
yarl==1.4.2