import os from typing import Dict, List import discord from discord.ext import commands import discord.ext import discord.ext.commands import modules.keys as k from modules.permission import requires_admin from modules.sentenceprocessing import send_message from modules.settings import Settings as SettingsManager import requests settings_manager = SettingsManager() settings = settings_manager.settings class BaseCommands(commands.Cog): def __init__(self, bot): self.bot: discord.ext.commands.Bot = bot @commands.command() async def help(self, ctx: commands.Context) -> None: embed: discord.Embed = discord.Embed( title=f"{k.command_help_embed_title()}", description=f"{k.command_help_embed_desc()}", color=discord.Colour(0x000000), ) command_categories = { f"{k.command_help_categories_general()}": [ "mem", "talk", "about", "ping", "impact", "demotivator", "help", ], f"{k.command_help_categories_admin()}": ["stats", "retrain", "setlanguage"], } custom_commands: List[str] = [] for cog_name, cog in self.bot.cogs.items(): for command in cog.get_commands(): if ( command.name not in command_categories[f"{k.command_help_categories_general()}"] and command.name not in command_categories[f"{k.command_help_categories_admin()}"] ): custom_commands.append(command.name) if custom_commands: embed.add_field( name=f"{k.command_help_categories_custom()}", value="\n".join( [ f"{settings["bot"]["prefix"]}{command}" for command in custom_commands ] ), inline=False, ) for category, commands_list in command_categories.items(): commands_in_category: str = "\n".join( [f"{settings["bot"]["prefix"]}{command}" for command in commands_list] ) embed.add_field(name=category, value=commands_in_category, inline=False) await send_message(ctx, embed=embed) @requires_admin() @commands.command() async def setlanguage(self, ctx: commands.Context, locale: str) -> None: await ctx.defer() k.change_language(locale) await ctx.send(":thumbsup:") @commands.command() async def ping(self, ctx: commands.Context) -> None: await ctx.defer() latency: int = round(self.bot.latency * 1000) embed: discord.Embed = discord.Embed( title="Pong!!", description=( settings["bot"]["misc"]["ping_line"], f"`{k.command_ping_embed_desc()}: {latency}ms`\n", ), color=discord.Colour(0x000000), ) embed.set_footer( text=f"{k.command_ping_footer()} {ctx.author.name}", icon_url=ctx.author.display_avatar.url, ) await ctx.send(embed=embed) @commands.command() async def about(self, ctx: commands.Context) -> None: embed: discord.Embed = discord.Embed( title=f"{k.command_about_embed_title()}", description="", color=discord.Colour(0x000000), ) embed.add_field( name=k.command_about_embed_field1(), value=f"{settings["name"]}", inline=False, ) embed.add_field(name="Github", value=f"https://github.com/gooberinc/goober") await send_message(ctx, embed=embed) @commands.command() async def stats(self, ctx: commands.Context) -> None: memory_file: str = "memory.json" file_size: int = os.path.getsize(memory_file) with open(memory_file, "r") as file: line_count: int = sum(1 for _ in file) embed: discord.Embed = discord.Embed( title=f"{k.command_stats_embed_title()}", description=f"{k.command_stats_embed_desc()}", color=discord.Colour(0x000000), ) embed.add_field( name=f"{k.command_stats_embed_field1name()}", value=f"{k.command_stats_embed_field1value(file_size=file_size, line_count=line_count)}", inline=False, ) with open(settings["splash_text_loc"], "r") as f: splash_text = "".join(f.readlines()) embed.add_field( name=f"{k.command_stats_embed_field3name()}", value=f"{k.command_stats_embed_field3value( NAME=settings["name"], PREFIX=settings["bot"]["prefix"], ownerid=settings["bot"]["owner_ids"][0], PING_LINE=settings["bot"]["misc"]["ping_line"], showmemenabled=settings["bot"]["allow_show_mem_command"], USERTRAIN_ENABLED=settings["bot"]["user_training"], song=settings["bot"]["misc"]["active_song"], splashtext=splash_text )}", inline=False, ) await send_message(ctx, embed=embed) @commands.command() async def mem(self, ctx: commands.Context) -> None: if not settings["bot"]["allow_show_mem_command"]: return with open(settings["bot"]["active_memory"], "rb") as f: data: bytes = f.read() response = requests.post( "https://litterbox.catbox.moe/resources/internals/api.php", data={"reqtype": "fileupload", "time": "1h", "fileToUpload": data}, ) print(response.text) print(response.status_code) # await send_message(ctx, memorylitter.stdout.strip()) async def setup(bot: discord.ext.commands.Bot): print("Setting up base_commands") bot.remove_command("help") await bot.add_cog(BaseCommands(bot))