44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
|
# _____ _
|
||
|
|
# / ____| (_)
|
||
|
|
# | (___ ___ __ _ _____ ___ _ __ ___ _ __ ___ ___ _ __
|
||
|
|
# \___ \ / _ \/ _` / __\ \ /\ / / | '_ ` _ \| '_ ` _ \ / _ \ '__|
|
||
|
|
# ____) | __/ (_| \__ \\ V V /| | | | | | | | | | | | __/ |
|
||
|
|
# |_____/ \___|\__,_|___/ \_/\_/ |_|_| |_| |_|_| |_| |_|\___|_|
|
||
|
|
|
||
|
|
import json
|
||
|
|
import itertools
|
||
|
|
|
||
|
|
from redbot.core import commands
|
||
|
|
from redbot.core.bot import Red
|
||
|
|
|
||
|
|
class Backup(commands.Cog):
|
||
|
|
"""A utility to make reinstalling repositories and cogs after migrating the bot far easier."""
|
||
|
|
|
||
|
|
__author__ = "SeaswimmerTheFsh"
|
||
|
|
__version__ = "1.0.0"
|
||
|
|
|
||
|
|
def __init__(self, bot: Red):
|
||
|
|
super().__init__()
|
||
|
|
self.bot = bot
|
||
|
|
|
||
|
|
@commands.group(autohelp=True)
|
||
|
|
@commands.is_owner()
|
||
|
|
async def backup(self, ctx: commands.Context):
|
||
|
|
"""Backup your installed cogs."""
|
||
|
|
|
||
|
|
@backup.command(name='export')
|
||
|
|
@commands.is_owner()
|
||
|
|
async def backup_export(self, ctx: commands.Context):
|
||
|
|
"""Export your installed repositories and cogs to a file."""
|
||
|
|
downloader = ctx.bot.get_cog("Downloader")
|
||
|
|
if downloader is None:
|
||
|
|
await ctx.send(f"You do not have the `Downloader` cog loaded.\nPlease run `{ctx.prefix}load downloader` and try again.")
|
||
|
|
return
|
||
|
|
all_repos = list(downloader._repo_manager.repos)
|
||
|
|
repos_list = [[f"{i.name}", i.url] for i in all_repos]
|
||
|
|
cogs = await downloader.installed_cogs()
|
||
|
|
for cog, r in itertools.product(cogs, repos_list):
|
||
|
|
if cog.repo_name == list(r)[0]:
|
||
|
|
r.append(cog.name)
|
||
|
|
await ctx.send(json.dumps(repos_list, indent=4))
|