bacon/discord_bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ended test_lib_Cog.py

Browse Source
This commit is contained in:
bacon
2024-03-17 02:17:07 +03:00
parent 736621c516
commit fca483e4e6
4 changed files with 31 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lib/CogsPrep.py
View File

@@ -8,6 +8,8 @@ work_with_cogs: loads, reloads and unloads cogs files
"""
from os import listdir, rename
from disnake.ext.commands import Bot
from .Logger import logger
@@ -19,13 +21,15 @@ async def cog_list(fold: str = './cogs') -> list:
return cogs_list
async def work_with_cogs(what_do, bot, cog):
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 == "lad":
if what_do == "load":
bot.load_extension(f'cogs.{_filename}')
logger.info(f'Loaded cog {_filename}')
logger.info(f'Cog {_filename} loaded')
elif what_do == 'unload':
bot.unload_extension(f'cogs.{_filename}')
logger.info(f'Cog {_filename} unloaded')
@@ -34,8 +38,11 @@ async def work_with_cogs(what_do, bot, cog):
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')
rename(f'cogs/disabled/{_filename}.py',
f'cogs/{_filename}.py')
bot.load_extension(f'cogs.{_filename}')
logger.info(f'Cog {_filename} started and enabled')