discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

logger-3000
Some checks failed
Deploy to core / deploy (push) Failing after 1m9s
Details

Browse Source
This commit is contained in:
Untone 2024-02-20 19:33:24 +03:00
parent b59a8ef323
commit 3c03688544
2 changed files with 37 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@ -1,8 +1,9 @@
from granian.constants import Interfaces from granian.constants import Interfaces
from granian.server import Granian from granian.server import Granian
from services.logger import root_logger as logger
if __name__ == '__main__': if __name__ == '__main__':
print('[server] started') logger.info('started')
granian_instance = Granian( granian_instance = Granian(
'main:app', 'main:app',
@ -10,6 +11,6 @@ if __name__ == '__main__':
port=8000, port=8000,
threads=4, threads=4,
websockets=False, websockets=False,
interface=Interfaces.ASGI, interface=Interfaces.ASGI
) )
granian_instance.serve() granian_instance.serve()

48
services/logger.py
View File

@ -1,7 +1,6 @@
import logging import logging
import colorlog import colorlog
# Define the color scheme # Define the color scheme
color_scheme = { color_scheme = {
'DEBUG': 'light_black', 'DEBUG': 'light_black',
@ -19,20 +18,42 @@ secondary_colors = {
'module': {'DEBUG': 'light_black,bg_blue'}, 'module': {'DEBUG': 'light_black,bg_blue'},
} }
# Define the log format string
fmt_string = "%(log_color)s%(levelname)s: %(log_color)s[%(module)s]%(reset)s %(white)s%(message)s"
# Define formatting configuration
fmt_config = {
'log_colors': color_scheme,
'secondary_log_colors': secondary_colors,
'style': '%',
'reset': True
}
# Create a ColoredFormatter object for colorized logging # Create a ColoredFormatter object for colorized logging
formatter = colorlog.ColoredFormatter( formatter = colorlog.ColoredFormatter(fmt_string, **fmt_config)
"%(log_color)s%(levelname)s: %(log_color)s[%(module)s]%(reset)s %(white)s%(message)s",
log_colors=color_scheme,
secondary_log_colors=secondary_colors, # Set secondary log colors class MultilineColoredFormatter(colorlog.ColoredFormatter):
style='%', def format(self, record):
reset=True # Check if the message is multiline
) if record.getMessage() and '\n' in record.getMessage():
# Split the message into lines
lines = record.getMessage().split('\n')
formatted_lines = []
for line in lines:
# Format each line with the provided format
formatted_lines.append(super().format(record, line))
# Join the formatted lines
return '\n'.join(formatted_lines)
else:
# If not multiline or no message, use the default formatting
return super().format(record)
def get_colorful_logger(name='main'): def get_colorful_logger(name='main'):
# Create a stream handler for logging output # Create a stream handler for logging output
stream = logging.StreamHandler() stream = logging.StreamHandler()
stream.setFormatter(formatter) stream.setFormatter(MultilineColoredFormatter(fmt_string, **fmt_config))
# Create and configure the logger # Create and configure the logger
logger = logging.getLogger(name) logger = logging.getLogger(name)
@ -41,8 +62,7 @@ def get_colorful_logger(name='main'):
return logger return logger
# Set up the root logger with the same formatting
root_logger = logging.getLogger() root_logger = logging.getLogger()
root_logger.setLevel(logging.DEBUG) # Set the log level root_logger.setLevel(logging.DEBUG)
stream_handler = logging.StreamHandler() root_logger.addHandler(logging.StreamHandler())
stream_handler.setFormatter(formatter)
root_logger.addHandler(stream_handler)