fixed start after move
This commit is contained in:
@@ -1,5 +1,4 @@
|
||||
#!/usr/bin/env python3
|
||||
import asyncio
|
||||
from os import getenv
|
||||
from os.path import isfile
|
||||
|
||||
@@ -7,9 +6,9 @@ from disnake import OptionType, Option, Localized, ApplicationCommandInteraction
|
||||
from disnake.ext.commands import Bot, is_owner
|
||||
from dotenv import load_dotenv
|
||||
|
||||
from lib.CogsPrep import work_with_cogs, cog_list
|
||||
from lib.Comands import determine_prefix
|
||||
from lib.Logger import logger
|
||||
from integral_lib.CogsPrep import work_with_cogs, cog_list
|
||||
from integral_lib.Comands import determine_prefix
|
||||
from integral_lib.Logger import logger
|
||||
|
||||
load_dotenv()
|
||||
if not isfile('.env') or not getenv('CONF_FILE'):
|
||||
@@ -36,7 +35,7 @@ bot = Bot(command_prefix=determine_prefix,
|
||||
|
||||
bot.i18n.load("locale/")
|
||||
|
||||
asyncio.run(work_with_cogs('load', bot, cog_list()))
|
||||
work_with_cogs('load', bot, cog_list())
|
||||
|
||||
|
||||
@bot.event
|
||||
|
||||
@@ -4,9 +4,9 @@ import disnake
|
||||
from disnake import Option, OptionType, Localized
|
||||
from disnake.ext import tasks
|
||||
|
||||
from lib.Comands import *
|
||||
from lib.DB_Worker import fill_bd, prepare_db, work_with_db
|
||||
from lib.Logger import logger
|
||||
from integral_lib.Comands import *
|
||||
from integral_lib.DB_Worker import fill_bd, prepare_db, work_with_db
|
||||
from integral_lib.Logger import logger
|
||||
|
||||
|
||||
class Admin(commands.Cog, name='Admin'):
|
||||
|
||||
@@ -4,9 +4,9 @@ import disnake
|
||||
from disnake import OptionChoice, Option, OptionType, Member, VoiceState
|
||||
from disnake.ext import commands
|
||||
|
||||
from lib.ListGenerator import ListGenerator
|
||||
from lib.Logger import logger
|
||||
from lib.Player import play_audio
|
||||
from integral_lib.ListGenerator import ListGenerator
|
||||
from integral_lib.Logger import logger
|
||||
from integral_lib.Player import play_audio
|
||||
|
||||
|
||||
class Audio(commands.Cog, name='Audio'):
|
||||
@@ -27,7 +27,7 @@ class Audio(commands.Cog, name='Audio'):
|
||||
logger.info('Skip playing by Game')
|
||||
else:
|
||||
# Prepare list of audio
|
||||
from bot.lib.Comands import read_json
|
||||
from bot.integral_lib.Comands import read_json
|
||||
_role = await read_json(member.guild.id, 'tigger_role')
|
||||
audio: list = []
|
||||
for _a in ListGenerator('audio'):
|
||||
|
||||
@@ -4,8 +4,8 @@ import disnake
|
||||
from disnake import OptionChoice
|
||||
from disnake.ext import commands
|
||||
|
||||
from lib.Comands import write_json
|
||||
from lib.Logger import logger
|
||||
from integral_lib.Comands import write_json
|
||||
from integral_lib.Logger import logger
|
||||
|
||||
|
||||
class Fun(commands.Cog, name='Fun'):
|
||||
|
||||
@@ -2,8 +2,8 @@ import disnake
|
||||
from disnake import Option
|
||||
from disnake.ext import commands
|
||||
|
||||
from lib import YandexPlayer
|
||||
from lib.Logger import logger
|
||||
from integral_lib import YandexPlayer
|
||||
from integral_lib.Logger import logger
|
||||
|
||||
|
||||
class Testing(commands.Cog, name='Testing'):
|
||||
|
||||
@@ -2,7 +2,7 @@ import disnake
|
||||
from disnake import Option, OptionType, Colour
|
||||
from disnake.ext import commands
|
||||
|
||||
from lib.Logger import logger
|
||||
from integral_lib.Logger import logger
|
||||
|
||||
|
||||
class General(commands.Cog):
|
||||
|
||||
@@ -6,8 +6,8 @@ from disnake import ApplicationCommandInteraction
|
||||
from disnake.ext import commands
|
||||
|
||||
from __init__ import version_info as ver
|
||||
from lib.Comands import determine_prefix
|
||||
from lib.Logger import logger
|
||||
from integral_lib.Comands import determine_prefix
|
||||
from integral_lib.Logger import logger
|
||||
|
||||
|
||||
class BotInfo(commands.Cog, name='Bot Info'):
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
"""
|
||||
lib.CogsPrepare
|
||||
integral_lib.CogsPrepare
|
||||
~~~~~~~~~~~~~
|
||||
Loads, unloads Cogs files
|
||||
cog_list: return list of cog filenames
|
||||
@@ -22,9 +22,9 @@ def cog_list(fold: str = './cogs') -> List[str]:
|
||||
return cogs_list
|
||||
|
||||
|
||||
async def work_with_cogs(what_do: str,
|
||||
bot: InteractionBotBase,
|
||||
cog: str | list):
|
||||
def work_with_cogs(what_do: str,
|
||||
bot: InteractionBotBase,
|
||||
cog: str | list):
|
||||
if isinstance(cog, str):
|
||||
cog = cog.split()
|
||||
for _filename in cog:
|
||||
@@ -1,5 +1,5 @@
|
||||
"""
|
||||
lib.Commands
|
||||
integral_lib.Commands
|
||||
~~~~~~~~~~~~~~
|
||||
Some prepare for commands
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
"""
|
||||
lib
|
||||
integral_lib
|
||||
~~~~~~~~~~~~~
|
||||
Some libs for the bot which help him
|
||||
"""
|
||||
@@ -1,34 +1,24 @@
|
||||
import tracemalloc
|
||||
|
||||
tracemalloc.start()
|
||||
|
||||
import pytest
|
||||
from disnake.ext.commands.common_bot_base import CommonBotBase
|
||||
from mock import mock
|
||||
|
||||
import os
|
||||
import sys
|
||||
|
||||
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
|
||||
|
||||
from bot.lib.CogsPrep import cog_list, work_with_cogs
|
||||
from integral_lib.CogsPrep import cog_list, work_with_cogs
|
||||
|
||||
|
||||
def test_cog_list():
|
||||
with mock.patch('bot.lib.CogsPrep.listdir') as MockClass:
|
||||
with mock.patch('integral_lib.CogsPrep.listdir') as MockClass:
|
||||
MockClass.return_value = ['cog1.py', 'cog2.py']
|
||||
result = cog_list()
|
||||
assert result == ['cog1', 'cog2']
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@pytest.mark.parametrize("cog", ["cog1.py", "cog2"])
|
||||
@pytest.mark.parametrize("what_do", ['load', 'unload', 'reload', 'disable', 'enable'])
|
||||
async def test_work_with_cogs(what_do, cog):
|
||||
with mock.patch('bot.lib.CogsPrep.rename') as mock_rename:
|
||||
def test_work_with_cogs(what_do, cog):
|
||||
with mock.patch('integral_lib.CogsPrep.rename') as mock_rename:
|
||||
mock_rename.return_value = None
|
||||
mock_bot = mock.MagicMock(spec=CommonBotBase)
|
||||
result = await work_with_cogs(what_do, mock_bot, cog)
|
||||
result = work_with_cogs(what_do, mock_bot, cog)
|
||||
if what_do in ['load', 'enable']:
|
||||
assert mock_bot.load_extension.called
|
||||
elif what_do in ['unload', 'disable']:
|
||||
@@ -36,10 +26,3 @@ async def test_work_with_cogs(what_do, cog):
|
||||
elif what_do == 'reload':
|
||||
assert mock_bot.reload_extension.called
|
||||
|
||||
|
||||
snapshot = tracemalloc.take_snapshot()
|
||||
top_stats = snapshot.statistics('lineno')
|
||||
|
||||
print("[ Top 10 ]")
|
||||
for stat in top_stats[:10]:
|
||||
print(stat)
|
||||
|
||||
Reference in New Issue
Block a user