66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Shared /watch configuration helpers."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
|
||
|
|
CONFIG_DIR = Path.home() / ".config" / "watch"
|
||
|
|
CONFIG_FILE = CONFIG_DIR / ".env"
|
||
|
|
|
||
|
|
DEFAULT_DETAIL = "balanced"
|
||
|
|
|
||
|
|
DETAILS = {"transcript", "efficient", "balanced", "token-burner"}
|
||
|
|
|
||
|
|
|
||
|
|
def read_env_file(path: Path | None = None) -> dict[str, str]:
|
||
|
|
if path is None:
|
||
|
|
path = CONFIG_FILE
|
||
|
|
values: dict[str, str] = {}
|
||
|
|
if not path.exists():
|
||
|
|
return values
|
||
|
|
try:
|
||
|
|
lines = path.read_text(encoding="utf-8").splitlines()
|
||
|
|
except OSError:
|
||
|
|
return values
|
||
|
|
for line in lines:
|
||
|
|
raw = line.strip()
|
||
|
|
if not raw or raw.startswith("#") or "=" not in raw:
|
||
|
|
continue
|
||
|
|
key, _, value = raw.partition("=")
|
||
|
|
value = value.strip()
|
||
|
|
if len(value) >= 2 and value[0] in ('"', "'") and value[-1] == value[0]:
|
||
|
|
value = value[1:-1]
|
||
|
|
values[key.strip()] = value
|
||
|
|
return values
|
||
|
|
|
||
|
|
|
||
|
|
def get_config() -> dict[str, object]:
|
||
|
|
file_values = read_env_file()
|
||
|
|
|
||
|
|
detail = (
|
||
|
|
os.environ.get("WATCH_DETAIL")
|
||
|
|
or file_values.get("WATCH_DETAIL")
|
||
|
|
or DEFAULT_DETAIL
|
||
|
|
)
|
||
|
|
if detail not in DETAILS:
|
||
|
|
detail = DEFAULT_DETAIL
|
||
|
|
|
||
|
|
return {
|
||
|
|
"detail": detail,
|
||
|
|
"config_file": str(CONFIG_FILE),
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def frame_cap(detail: str) -> int | None:
|
||
|
|
if detail == "efficient":
|
||
|
|
return 50
|
||
|
|
if detail == "balanced":
|
||
|
|
return 100
|
||
|
|
if detail == "token-burner":
|
||
|
|
return None
|
||
|
|
if detail == "transcript":
|
||
|
|
return None
|
||
|
|
return 100
|