mirror of
https://github.com/jarulsamy/Plex-Bot.git
synced 2024-08-19 15:01:55 +02:00
37 lines
978 B
Python
37 lines
978 B
Python
import yaml
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
import logging
|
|
import sys
|
|
|
|
FORMAT = "%(asctime)s %(levelname)s: [%(filename)s:%(lineno)s - %(funcName)20s() ] %(message)s"
|
|
|
|
logging.basicConfig(format=FORMAT)
|
|
logger = logging.getLogger("PlexBot")
|
|
|
|
|
|
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:
|
|
logging.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,
|
|
}
|
|
level = config["general"]["log_level"]
|
|
config["general"]["log_level"] = levels[level.upper()]
|
|
|
|
return config
|