college-bot-v2/src/conf.py

import toml
import logging
import subprocess
from pathlib import Path

log = logging.getLogger("jimmy.autoconf")

if (Path.cwd() / ".git").exists():
    try:
        log.debug("Attempting to auto-detect running version using git.")
        VERSION = subprocess.run(
            ["git", "rev-parse", "--short", "HEAD"],
            stdout=subprocess.PIPE,
            stderr=subprocess.DEVNULL,
            text=True,
            check=True
        ).stdout.strip()
    except subprocess.CalledProcessError:
        log.debug("Unable to auto-detect running version using git.", exc_info=True)
        VERSION = "unknown"
else:
    log.debug("Unable to auto-detect running version using git, no .git directory exists.")
    VERSION = "unknown"

try:
    CONFIG = toml.load('config.toml')
    CONFIG.setdefault("logging", {})
    CONFIG.setdefault("jimmy", {})
    CONFIG.setdefault("ollama", {})
    CONFIG.setdefault("rss", {"meta": {"channel": None}})
    CONFIG.setdefault("screenshot", {})
    CONFIG.setdefault(
        "server",
        {
            "host": "0.0.0.0",
            "port": 8080,
            "channel": 1032974266527907901
        }
    )
    CONFIG.setdefault(
        "redis",
        {
            "host": "redis",
            "port": 6379,
            "decode_responses": True
        }
    )
except FileNotFoundError:
    cwd = Path.cwd()
    log.critical("Unable to locate config.toml in %s.", cwd, exc_info=True)
    raise