diff --git a/src/lib/logger.py b/src/lib/logger.py index 6a266b2..c7b7aeb 100644 --- a/src/lib/logger.py +++ b/src/lib/logger.py @@ -1,16 +1,26 @@ from loguru import logger from pathlib import Path import sys +from dotenv import load_dotenv +import os -LOG_PATH = Path("./data/logs/logs.log").absolute() -ERROR_PATH = Path("./data/logs/errors.log").absolute() +load_dotenv() + +LOG_PATH = Path(os.getenv("LOG_PATH", "./data/logs/logs.log")).absolute() +ERROR_PATH = Path(os.getenv("ERROR_PATH", "./data/logs/errors.log")).absolute() logger.remove() -logger.add(LOG_PATH, rotation="10 MB", compression="gz", level="INFO") -logger.add(sys.stdout, level="INFO") +logger.add( + LOG_PATH, rotation="10 MB", compression="gz", level=os.getenv("LOG_LEVEL", "INFO") +) +logger.add(sys.stdout, level=os.getenv("LOG_LEVEL", "INFO")) -logger.add(ERROR_PATH, rotation="10 MB", compression="gz", level="ERROR") -logger.add(sys.stderr, level="ERROR") -logger.add(sys.stdout, level="DEBUG") +logger.add( + ERROR_PATH, + rotation="10 MB", + compression="gz", + level=os.getenv("LOG_LEVEL_ERROR", "ERROR"), +) +logger.add(sys.stderr, level=os.getenv("LOG_LEVEL_ERROR", "ERROR")) logger = logger