GalaxyCogs/exportchannels/exportchannels.py

import discord
import asyncio
import os
from redbot.core import Config, checks, commands, data_manager

class ExportChannels(commands.Cog):
    """Custom cog to export channels to JSON and HTML formats using Discord Chat Exporter.
    Developed by SeaswimmerTheFsh and yname."""

    def __init__(self, bot):
        self.bot = bot
        self.config = Config.get_conf(self, identifier=48258471944753312)
        self.config.register_global(
            bot_token = None
        )

    class ConfigException(Exception):
        pass

    async def export(self, channels: list):
        token = await self.config.bot_token()
        if token is None:
            raise(self.ConfigException, "Bot token not set!")
        self.data_path = data_manager.cog_data_path(self)
        self.bundled_data_path = data_manager.bundled_data_path(self)
        out = f'{self.data_path}{os.sep}Exported Channels'
        channel = channels[0]
        file_location_html = f'{os.sep}{out}{os.sep}%g{os.sep}%c{os.sep}Export.html'
        file_location_json =  f'{os.sep}{out}{os.sep}%g{os.sep}%c{os.sep}DCE-f.json'
        try:
            os.mkdir(out)
        except FileExistsError:
            pass
        args = [
            'dotnet',
            'DiscordChatExporter.Cli.dll',
            'export',
            '--format', 'HtmlDark',
            '--output', file_location_html,
            '--token', token,
            '--channel', channel,
            '--media',
            '--fuck_russia', 'true',
        ]
        os.chdir(self.bundled_data_path)
        process_1 = await asyncio.create_subprocess_exec(*args)
        while True:
            return_code_1 = process_1.poll()
            if return_code_1 is not None:
                break
        args = [
            'dotnet',
            'DiscordChatExporter.Cli.dll',
            'export',
            '--format', 'Json',
            '--output', file_location_json,
            '--token', token,
            '--channel', channels,
            '--reuse_media',
            '--media',
            '--fuck_russia', 'true',
        ]
        os.chdir(self.bundled_data_path)
        process_2 = await asyncio.create_subprocess_exec(*args)
        while True:
            return_code_2 = process_2.poll()
            if return_code_2 is not None:
                break

    @commands.group()
    @checks.is_owner()
    async def exportset(self, ctx):
        """Configuration options for the ExportChannels cog."""

    @exportset.command()
    @checks.is_owner()
    async def token(self, ctx, token: str):
        """Sets the bot token used for Discord Chat Exporter."""
        await self.config.bot_token.set({token})
        await ctx.send(content="Token set!")
        await ctx.delete()

    @exportset.command()
    @checks.is_owner()
    async def checkoutputpath(self, ctx):
        """Checks what file path DCE is outputting to."""
        self.data_path = data_manager.cog_data_path(self)
        await ctx.send(content=f"{self.data_path}")

    @commands.command()
    @commands.admin()
    async def exportchannel(self, ctx, channel: discord.TextChannel):
        """Exports a channel using Discord Chat Exporter."""
        token = await self.config.bot_token()
        dce_install = data_manager.bundled_data_path(self)
        if token is None:
            await ctx.send(content=f"Please set your token with the ``{ctx.prefix}exportset token`` command!")
            return
        else:
            id_list = [thread.id for thread in channel.threads]
            id_list.insert(0, channel.id)
            await self.export(id_list)