www_site/infra/scripts/update-tts-env.py

24 lines
610 B
Python

from pathlib import Path
import sys
path = Path("/etc/kaotings/api.env")
url = sys.stdin.readline().rstrip()
key = sys.stdin.readline().rstrip()
values = {"TTS_UPSTREAM_URL": url, "TTS_API_KEY": key}
lines = path.read_text().splitlines()
output = []
seen = set()
for line in lines:
name = line.split("=", 1)[0]
if name in values:
output.append(f"{name}={values[name]}")
seen.add(name)
else:
output.append(line)
for name, value in values.items():
if name not in seen:
output.append(f"{name}={value}")
path.write_text("\n".join(output) + "\n")
path.chmod(0o600)