Added base cog to inherit from

This commit is contained in:
phixxy 2024-02-14 00:59:05 -08:00
parent d39a5c395b
commit 02fa32cb1c
4 changed files with 34 additions and 19 deletions

View file

View file

@ -0,0 +1,28 @@
import os
import aiohttp
from discord.ext import commands
import logging
class BotBaseCog(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.class_name = __name__
self.working_dir = f"tmp/{self.class_name}"
self.data_dir = f"data/{self.class_name}"
self.folder_setup()
self.http_session = self.create_aiohttp_session()
self.logger = logging.getLogger("bot")
def create_aiohttp_session(self):
return aiohttp.ClientSession()
def folder_setup(self) -> None:
try:
if not os.path.exists(self.working_dir):
os.mkdir(self.working_dir)
if not os.path.exists(self.data_dir):
os.mkdir(self.data_dir)
except:
self.logger.exception(f"{self.class_name} failed to make directories")

View file

@ -1,28 +1,14 @@
#plugin for sparkytron3000
import os
import random
import aiohttp
from discord.ext import commands
import logging
from cogs.base_cog.bot_base_cog import BotBaseCog
class Meme(commands.Cog):
class Meme(BotBaseCog):
def __init__(self, bot):
self.bot = bot
self.working_dir = "tmp/meme/"
self.folder_setup()
self.http_session = self.create_aiohttp_session()
self.logger = logging.getLogger("bot")
def create_aiohttp_session(self):
return aiohttp.ClientSession()
def folder_setup(self):
try:
if not os.path.exists(self.working_dir):
os.mkdir(self.working_dir)
except:
self.logger.exception("Meme failed to make directories")
super().__init__(bot)
async def answer_question(self, topic, model="gpt-3.5-turbo"):
headers = {