from pathlib import Path
from datetime import datetime
import sys
p = Path(sys.argv[1])
text = p.read_text()
backup = p.with_name(p.name + ".bak_open_img_test_" + datetime.now().strftime("%Y%m%d_%H%M%S"))
backup.write_text(text)
updates = {
    "SUBLB_IMAGE_STUDIO_OPEN_IMG_TEST_API_BASE_URL": "http://127.0.0.1:48090",
    "SUBLB_IMAGE_STUDIO_OPEN_IMG_TEST_API_KEY": "sk-30a5fa8319f2cdc792b04b389d5b6126dcec9dbeb20c94e456226c2c15c234bf",
}
lines = text.splitlines()
seen = set()
out = []
for line in lines:
    if "=" in line and not line.lstrip().startswith("#"):
        k = line.split("=",1)[0]
        if k in updates:
            out.append(f"{k}={updates[k]}")
            seen.add(k)
            continue
    out.append(line)
for k,v in updates.items():
    if k not in seen:
        out.append(f"{k}={v}")
p.write_text("\n".join(out) + "\n")
print(str(backup))