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

1
.gitignore vendored
View file

@ -4,3 +4,4 @@ data/
logs/ logs/
src/__pycache__/ src/__pycache__/
tmp/ tmp/
cogs/base_cog/__pycache__/

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 #plugin for sparkytron3000
import os import os
import random import random
import aiohttp
from discord.ext import commands 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): def __init__(self, bot):
self.bot = bot super().__init__(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")
async def answer_question(self, topic, model="gpt-3.5-turbo"): async def answer_question(self, topic, model="gpt-3.5-turbo"):
headers = { headers = {