zorruno
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Refresh Flounder's local MotionEye preview images."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
import sys
|
|
import time
|
|
from urllib.error import URLError
|
|
from urllib.request import urlopen
|
|
|
|
|
|
OUTPUTS = (
|
|
"front-door.jpg",
|
|
"driveway.jpg",
|
|
"garage.jpg",
|
|
"bike-shed.jpg",
|
|
"caravan.jpg",
|
|
)
|
|
OUTPUT_DIR = Path(os.environ.get("HASS_CONFIG", "/config")) / "www/flounder"
|
|
HASS_URL = os.environ.get("HASS_URL", "http://127.0.0.1:8123")
|
|
|
|
|
|
def download(url_path: str, destination: Path) -> bool:
|
|
url = f"{HASS_URL}{url_path}"
|
|
for attempt in range(2):
|
|
try:
|
|
with urlopen(url, timeout=4) as response:
|
|
image = response.read()
|
|
if image.startswith(b"\xff\xd8") and image.endswith(b"\xff\xd9"):
|
|
temporary = destination.with_suffix(".tmp")
|
|
temporary.write_bytes(image)
|
|
os.replace(temporary, destination)
|
|
return True
|
|
except (OSError, TimeoutError, URLError):
|
|
pass
|
|
if attempt == 0:
|
|
time.sleep(2)
|
|
return False
|
|
|
|
|
|
def main() -> int:
|
|
arguments = sys.argv[1:]
|
|
outputs = OUTPUTS
|
|
if arguments and arguments[0] == "--primary":
|
|
arguments = arguments[1:]
|
|
outputs = OUTPUTS[:3]
|
|
|
|
if len(arguments) != len(outputs):
|
|
return 2
|
|
|
|
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
|
failures = 0
|
|
for url_path, filename in zip(arguments, outputs, strict=True):
|
|
if not download(url_path, OUTPUT_DIR / filename):
|
|
failures += 1
|
|
time.sleep(2)
|
|
return 1 if failures else 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|