mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-10 23:08:37 +02:00
41e290449e
* fix environment_vars timing bug for EnvString substitution FRIGATE_ENV_VARS is populated at module import time but was never updated when environment_vars config section set new vars into os.environ. This meant HA OS users setting FRIGATE_* vars via environment_vars could not use them in EnvString fields. * add EnvString support to MQTT and ONVIF host fields * add tests * docs * update reference config
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import os
|
|
from pathlib import Path
|
|
from typing import Annotated
|
|
|
|
from pydantic import AfterValidator, ValidationInfo
|
|
|
|
FRIGATE_ENV_VARS = {k: v for k, v in os.environ.items() if k.startswith("FRIGATE_")}
|
|
secrets_dir = os.environ.get("CREDENTIALS_DIRECTORY", "/run/secrets")
|
|
# read secret files as env vars too
|
|
if os.path.isdir(secrets_dir) and os.access(secrets_dir, os.R_OK):
|
|
for secret_file in os.listdir(secrets_dir):
|
|
if secret_file.startswith("FRIGATE_"):
|
|
FRIGATE_ENV_VARS[secret_file] = (
|
|
Path(os.path.join(secrets_dir, secret_file)).read_text().strip()
|
|
)
|
|
|
|
|
|
def validate_env_string(v: str) -> str:
|
|
return v.format(**FRIGATE_ENV_VARS)
|
|
|
|
|
|
EnvString = Annotated[str, AfterValidator(validate_env_string)]
|
|
|
|
|
|
def validate_env_vars(v: dict[str, str], info: ValidationInfo) -> dict[str, str]:
|
|
if isinstance(info.context, dict) and info.context.get("install", False):
|
|
for k, val in v.items():
|
|
os.environ[k] = val
|
|
if k.startswith("FRIGATE_"):
|
|
FRIGATE_ENV_VARS[k] = val
|
|
|
|
return v
|
|
|
|
|
|
EnvVars = Annotated[dict[str, str], AfterValidator(validate_env_vars)]
|