GalaxyCogs/exportchannels/exportchannels.py
SeaswimmerTheFsh c7dd2ab0fe
Some checks failed
Gitea Actions Demo / Explore-Gitea-Actions (push) Failing after 5s
fix(issues): hopefully fixed the cog just not working
2023-08-17 02:54:32 -04:00

104 lines
3.6 KiB
Python

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!")
data_path = data_manager.cog_data_path(self)
bundled_data_path = data_manager.bundled_data_path(self)
out = f'{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,
'--reuse-media',
'--media',
'--fuck_russia', 'true',
]
os.chdir(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',
'--markdown', 'false',
'--fuck_russia', 'true',
]
os.chdir(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)