changed logger

This commit is contained in:
bacon
2024-04-13 20:03:57 +03:00
parent 78be10a88b
commit 413736b44d
13 changed files with 24 additions and 67 deletions

View File

@@ -64,5 +64,5 @@ push:
- ssh-add ~/.ssh/id_rsa - ssh-add ~/.ssh/id_rsa
script: script:
- ssh $BOT_HOST -oStrictHostKeyChecking=accept-new 'cd /opt/discord/tarkov && git pull' - ssh $BOT_HOST -oStrictHostKeyChecking=accept-new 'cd /opt/discord/tarkov && git pull && pip install -r requirements.txt'

2
bot.py
View File

@@ -6,11 +6,11 @@ from os.path import isfile
from disnake import OptionType, Option, Localized, ApplicationCommandInteraction, Intents, __version__ from disnake import OptionType, Option, Localized, ApplicationCommandInteraction, Intents, __version__
from disnake.ext.commands import Bot, is_owner from disnake.ext.commands import Bot, is_owner
from dotenv import load_dotenv from dotenv import load_dotenv
from loguru import logger
from __init__ import __version__ as ver from __init__ import __version__ as ver
from lib.CogsPrep import work_with_cogs, cog_list from lib.CogsPrep import work_with_cogs, cog_list
from lib.Comands import determine_prefix from lib.Comands import determine_prefix
from lib.Logger import logger
load_dotenv() load_dotenv()
if not isfile('.env') or not getenv('CONF_FILE'): if not isfile('.env') or not getenv('CONF_FILE'):

View File

@@ -3,10 +3,10 @@ from asyncio import sleep
import disnake import disnake
from disnake import Option, OptionType, Localized from disnake import Option, OptionType, Localized
from disnake.ext import commands, tasks from disnake.ext import commands, tasks
from loguru import logger
from lib.Comands import read_json, write_json from lib.Comands import read_json, write_json
from lib.DB_Worker import fill_bd, prepare_db, work_with_db from lib.DB_Worker import fill_bd, prepare_db, work_with_db
from lib.Logger import logger
class Admin(commands.Cog, name='Admin'): class Admin(commands.Cog, name='Admin'):

View File

@@ -4,10 +4,10 @@ import disnake
from disnake import OptionChoice, Option, OptionType, Member, VoiceState from disnake import OptionChoice, Option, OptionType, Member, VoiceState
from disnake.ext import commands from disnake.ext import commands
from disnake.utils import get from disnake.utils import get
from loguru import logger
from lib.Logger import logger
from lib.Player import play_audio
from lib.ListGenerator import ListGenerator from lib.ListGenerator import ListGenerator
from lib.Player import play_audio
class Audio(commands.Cog, name='Audio'): class Audio(commands.Cog, name='Audio'):

View File

@@ -3,9 +3,9 @@ from typing import List
import disnake import disnake
from disnake import OptionChoice from disnake import OptionChoice
from disnake.ext import commands from disnake.ext import commands
from loguru import logger
from lib.Comands import write_json from lib.Comands import write_json
from lib.Logger import logger
class Fun(commands.Cog, name='Fun'): class Fun(commands.Cog, name='Fun'):

View File

@@ -1,9 +1,9 @@
import disnake import disnake
from disnake import Option from disnake import Option
from disnake.ext import commands from disnake.ext import commands
from loguru import logger
from lib import YandexPlayer from lib import YandexPlayer
from lib.Logger import logger
class Testing(commands.Cog, name='Testing'): class Testing(commands.Cog, name='Testing'):

View File

@@ -2,7 +2,8 @@ import disnake
from disnake import Option, OptionType, Colour from disnake import Option, OptionType, Colour
from disnake.ext import commands from disnake.ext import commands
from lib.Logger import logger # from lib.Logger import logger
from loguru import logger
class General(commands.Cog): class General(commands.Cog):

View File

@@ -1,13 +1,16 @@
import os import os
import disnake import disnake
import psutil import psutil
from disnake.ext import commands from disnake.ext import commands
from loguru import logger
from __init__ import version_info as ver from __init__ import version_info as ver
from lib.Comands import determine_prefix from lib.Comands import determine_prefix
from lib.Logger import logger
# from lib.Logger import logger
class BotInfo(commands.Cog, name='Bot Info'): class BotInfo(commands.Cog, name='Bot Info'):
def __init__(self, bot: commands.Bot): def __init__(self, bot: commands.Bot):
self.bot = bot # defining bot as global var in class self.bot = bot # defining bot as global var in class

View File

@@ -10,8 +10,8 @@ from os import listdir, rename
from typing import List from typing import List
from disnake.ext import commands from disnake.ext import commands
# from .Logger import logger
from .Logger import logger from loguru import logger
async def cog_list(fold: str = './cogs') -> List[str]: async def cog_list(fold: str = './cogs') -> List[str]:

View File

@@ -1,6 +1,7 @@
from sqlite3 import connect, Error from sqlite3 import connect, Error
from lib.Logger import logger # from lib.Logger import logger
from loguru import logger
class DBReader: class DBReader:

View File

@@ -1,52 +0,0 @@
import logging
class _CustomFormatter(logging.Formatter):
grey = "\x1b[38;20m"
yellow = "\x1b[33;20m"
red = "\x1b[31;20m"
bold_red = "\x1b[31;1m"
reset = "\x1b[0m"
fmt = "%(asctime)s - [%(levelname)s] -%(module)s- %(message)s"
FORMATS = {
logging.DEBUG: grey + fmt + reset,
logging.INFO: grey + fmt + reset,
logging.WARNING: yellow + fmt + reset,
logging.ERROR: red + fmt + reset,
logging.CRITICAL: bold_red + fmt + reset
}
old_factory = logging.getLogRecordFactory()
def _record_factory(*args, **kwargs):
record = _CustomFormatter.old_factory(*args, **kwargs)
_module = record.module
_levelname = record.levelname
if len(_module) % 2 == 0 and len(_module) < 12:
_module = ' ' * ((12 - len(_module)) // 2) + _module + ' ' * ((12 - len(_module)) // 2)
elif len(_module) % 2 == 1 and len(_module) < 12:
_module = ' ' * ((12 - len(_module)) // 2) + record.module + ' ' * ((12 - len(_module)) // 2 + 1)
if len(_levelname) % 2 == 0 and len(_levelname) < 8:
_levelname = ' ' * ((8 - len(_levelname)) // 2) + _levelname + ' ' * ((8 - len(_levelname)) // 2)
elif len(record.levelname) % 2 == 1 and len(record.module) < 8:
_levelname = ' ' * ((8 - len(_levelname)) // 2) + _levelname + ' ' * ((8 - len(_levelname)) // 2 + 1)
record.module = _module
record.levelname = _levelname
return record
logging.setLogRecordFactory(_record_factory)
def format(self, record):
log_fmt = self.FORMATS.get(record.levelno)
formatter = logging.Formatter(log_fmt, "%d-%m-%Y %H:%M:%S")
return formatter.format(record)
logger = logging.getLogger('Pisya_Bot')
logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
ch.setFormatter(_CustomFormatter())
logger.addHandler(ch)

View File

@@ -1,10 +1,13 @@
from asyncio import sleep from asyncio import sleep
from disnake import FFmpegOpusAudio from disnake import FFmpegOpusAudio
from loguru import logger
from .Logger import logger
# from .Logger import logger
@logger.catch
async def play_audio(audio, bot, vc): async def play_audio(audio, bot, vc):
if not bot.voice_clients: if not bot.voice_clients:
logger.error(f'Playing: {audio}') logger.error(f'Playing: {audio}')

View File

@@ -2,3 +2,4 @@ disnake[audio]
psutil psutil
python-dotenv python-dotenv
yandex-music yandex-music
loguru==0.7.2