phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cleanup of new base cog

Browse source
This commit is contained in:
phixxy 2024-02-14 01:05:47 -08:00
parent 02fa32cb1c
commit 9dd6fc1e1b
2 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cogs/base_cog/bot_base_cog.py
View file

@ -6,15 +6,13 @@ import logging
class BotBaseCog(commands.Cog): class BotBaseCog(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
self.class_name = __name__ self.cog_name = __name__
self.working_dir = f"tmp/{self.class_name}" self.working_dir = f"tmp/{self.cog_name}"
self.data_dir = f"data/{self.class_name}" self.data_dir = f"data/{self.cog_name}"
self.folder_setup() self.folder_setup()
self.http_session = self.create_aiohttp_session() self.http_session = self.create_aiohttp_session()
self.logger = logging.getLogger("bot") self.logger = logging.getLogger("bot")
def create_aiohttp_session(self): def create_aiohttp_session(self):
return aiohttp.ClientSession() return aiohttp.ClientSession()

1
cogs/meme.py
View file

@ -37,6 +37,7 @@ class Meme(BotBaseCog):
brief="Generate a meme" brief="Generate a meme"
) )
async def meme(self, ctx): async def meme(self, ctx):
self.logger.info(f"Meme command called by {ctx.author.name}")
async def generate_random_meme(topic): async def generate_random_meme(topic):
async with self.http_session.get('https://api.imgflip.com/get_memes') as resp: async with self.http_session.get('https://api.imgflip.com/get_memes') as resp:
response_data = await resp.json() response_data = await resp.json()