From 6a07ef45f27e31d04fb3d749319a7c1ffdbe8645 Mon Sep 17 00:00:00 2001 From: Slava Date: Sun, 17 Mar 2024 11:48:47 +0000 Subject: [PATCH] Update CogsPrep.py --- lib/CogsPrep.py | 96 ++++++++++++++++++++++++------------------------- 1 file changed, 48 insertions(+), 48 deletions(-) diff --git a/lib/CogsPrep.py b/lib/CogsPrep.py index 3c6e9aa..d82fb73 100644 --- a/lib/CogsPrep.py +++ b/lib/CogsPrep.py @@ -1,48 +1,48 @@ -""" -lib.CogsPrepare -~~~~~~~~~~~~~ -Loads, unloads Cogs files -cog_list: return list of cog filenames -work_with_cogs: loads, reloads and unloads cogs files - -""" -from os import listdir, rename - -from disnake.ext.commands import Bot - -from .Logger import logger - - -async def cog_list(fold: str = './cogs') -> list: - cogs_list = [] - for _filename in listdir(fold): - if _filename.endswith('.py'): - cogs_list.append(_filename[:-3]) - return cogs_list - - -async def work_with_cogs(what_do: str, - bot: Bot, - cog: str | list): - if isinstance(cog, str): - cog = cog.split() - for _filename in cog: - if what_do == "load": - bot.load_extension(f'cogs.{_filename}') - logger.info(f'Cog {_filename} loaded') - elif what_do == 'unload': - bot.unload_extension(f'cogs.{_filename}') - logger.info(f'Cog {_filename} unloaded') - elif what_do == 'reload': - bot.reload_extension(f'cogs.{_filename}') - logger.info(f'Cog {_filename} reloaded') - elif what_do == 'disable': - bot.unload_extension(f'cogs.{_filename}') - rename(f'cogs/{_filename}.py', - f'cogs/disabled/{_filename}.py') - logger.info(f'Cog {_filename} stopped and disabled') - elif what_do == 'enable': - rename(f'cogs/disabled/{_filename}.py', - f'cogs/{_filename}.py') - bot.load_extension(f'cogs.{_filename}') - logger.info(f'Cog {_filename} started and enabled') +""" +lib.CogsPrepare +~~~~~~~~~~~~~ +Loads, unloads Cogs files +cog_list: return list of cog filenames +work_with_cogs: loads, reloads and unloads cogs files + +""" +from os import listdir, rename + +from disnake.ext.commands import Bot + +from .Logger import logger + + +def cog_list(fold: str = './cogs') -> list: + cogs_list = [] + for _filename in listdir(fold): + if _filename.endswith('.py'): + cogs_list.append(_filename[:-3]) + return cogs_list + + +async def work_with_cogs(what_do: str, + bot: Bot, + cog: str | list): + if isinstance(cog, str): + cog = cog.split() + for _filename in cog: + if what_do == "load": + bot.load_extension(f'cogs.{_filename}') + logger.info(f'Cog {_filename} loaded') + elif what_do == 'unload': + bot.unload_extension(f'cogs.{_filename}') + logger.info(f'Cog {_filename} unloaded') + elif what_do == 'reload': + bot.reload_extension(f'cogs.{_filename}') + logger.info(f'Cog {_filename} reloaded') + elif what_do == 'disable': + bot.unload_extension(f'cogs.{_filename}') + rename(f'cogs/{_filename}.py', + f'cogs/disabled/{_filename}.py') + logger.info(f'Cog {_filename} stopped and disabled') + elif what_do == 'enable': + rename(f'cogs/disabled/{_filename}.py', + f'cogs/{_filename}.py') + bot.load_extension(f'cogs.{_filename}') + logger.info(f'Cog {_filename} started and enabled')