Chris Cowan
bf4e850655
All checks were successful
Build and Push Docker Image / build (push) Successful in 49s
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
|
import cgi
|
|
import cgitb
|
|
import os
|
|
import json
|
|
from pathlib import Path
|
|
|
|
cgitb.enable()
|
|
|
|
ENV_FILE = Path("/config/backupbot.env")
|
|
print("Content-Type: application/json\n")
|
|
|
|
|
|
def read_env():
|
|
env = {}
|
|
if ENV_FILE.exists():
|
|
with ENV_FILE.open() as f:
|
|
for line in f:
|
|
line = line.strip()
|
|
if not line or line.startswith("#") or "=" not in line:
|
|
continue
|
|
key, val = line.split("=", 1)
|
|
env[key.strip()] = val.strip().split("#")[0].strip()
|
|
return env
|
|
|
|
|
|
def write_env(env):
|
|
# Rotate the old env file just in case
|
|
if ENV_FILE.exists():
|
|
ENV_FILE.replace(ENV_FILE.with_suffix(".env.bak"))
|
|
with ENV_FILE.open("w") as f:
|
|
for key, val in env.items():
|
|
f.write(f"{key}={val}\n")
|
|
|
|
|
|
form = cgi.FieldStorage()
|
|
action = form.getvalue("action")
|
|
|
|
try:
|
|
if action == "get":
|
|
print(json.dumps(read_env()))
|
|
elif action == "set":
|
|
length = int(os.environ.get("CONTENT_LENGTH", "0"))
|
|
data = json.loads(os.read(0, length))
|
|
write_env(data)
|
|
print(json.dumps({"status": "ok", "message": "Configuration saved."}))
|
|
else:
|
|
print(json.dumps({"status": "error", "message": "Invalid action"}))
|
|
except Exception as e:
|
|
print(json.dumps({"status": "error", "message": str(e)}))
|