From 9b9736ab8bf2e7b8997fb594348fd7c0fa4b7334 Mon Sep 17 00:00:00 2001 From: "Dustin C. Hatch" Date: Mon, 1 Aug 2022 22:14:06 -0500 Subject: [PATCH] svc: api: Get URLs file path from env var The `HUDCTRL_URLS_FILE` environment variable can be used to specify the location to the `urls.json` file. --- svc/src/hudctrl/api.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/svc/src/hudctrl/api.py b/svc/src/hudctrl/api.py index cfa35ac..7275af0 100644 --- a/svc/src/hudctrl/api.py +++ b/svc/src/hudctrl/api.py @@ -1,9 +1,11 @@ import io import logging +import os +from pathlib import Path from typing import Optional -import fastapi from PIL import Image +import fastapi from .hud import HUDService, NoMonitorConfig from .xrandr import MonitorConfig @@ -15,11 +17,16 @@ log = logging.getLogger(__name__) logging.basicConfig(level=logging.DEBUG) +HUDCTRL_URLS_FILE = os.environ.get('HUDCTRL_URLS_FILE') + + app = fastapi.FastAPI( docs_url='/api-doc/', ) svc = HUDService() +if HUDCTRL_URLS_FILE: + svc.urls_file = Path(HUDCTRL_URLS_FILE) class PNGImageResponse(fastapi.Response):