mirror of
https://github.com/jarulsamy/Plex-Bot.git
synced 2024-08-19 15:01:55 +02:00
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
import logging
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
|
|
import yaml
|
|
|
|
FORMAT = "%(asctime)s %(levelname)s: [%(filename)s:%(lineno)s - %(funcName)20s() ] %(message)s"
|
|
|
|
logging.basicConfig(format=FORMAT)
|
|
root_log = logging.getLogger()
|
|
plex_log = logging.getLogger("Plex")
|
|
bot_log = logging.getLogger("Bot")
|
|
|
|
|
|
def load_config(filename: str) -> Dict[str, str]:
|
|
|
|
# All config files should be in /config
|
|
# for docker deployment.
|
|
filename = Path("/config", filename)
|
|
try:
|
|
with open(filename, "r") as f:
|
|
config = yaml.safe_load(f)
|
|
except FileNotFoundError:
|
|
root_log.fatal("Configuration file not found.")
|
|
sys.exit(-1)
|
|
|
|
# Convert str level type to logging constant
|
|
levels = {
|
|
"DEBUG": logging.DEBUG,
|
|
"INFO": logging.INFO,
|
|
"WARNING": logging.WARNING,
|
|
"ERROR": logging.ERROR,
|
|
"CRITICAL": logging.CRITICAL,
|
|
}
|
|
|
|
config["root"]["log_level"] = levels[config["root"]["log_level"].upper()]
|
|
config["plex"]["log_level"] = levels[config["plex"]["log_level"].upper()]
|
|
config["discord"]["log_level"] = levels[config["discord"]["log_level"].upper()]
|
|
|
|
return config
|