first commit

app/config.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+@dataclass(frozen=True)
+class Settings:
+    stream_url: str
+    api_base_url: str
+    app_id: str
+    app_secret: str
+    device_list_path: str
+    device_account: str
+    stream_method: str
+    detr_model: str
+    confidence: float
+    frame_skip: int
+    jpeg_quality: int
+    resize_width: int | None
+    vehicle_labels: set[str]
+
+
+def _optional_int(name: str) -> int | None:
+    value = os.getenv(name, "").strip()
+    if not value:
+        return None
+    parsed = int(value)
+    return parsed if parsed > 0 else None
+
+
+def load_settings() -> Settings:
+    stream_url = os.getenv("STREAM_URL", "").strip()
+
+    vehicle_labels = {
+        item.strip()
+        for item in os.getenv("VEHICLE_LABELS", "car,motorcycle,bus,truck,bicycle").split(",")
+        if item.strip()
+    }
+
+    return Settings(
+        stream_url=stream_url,
+        api_base_url=os.getenv("API_BASE_URL", "https://apicapacity.51iwifi.com"),
+        app_id=os.getenv("APP_ID", ""),
+        app_secret=os.getenv("APP_SECRET", ""),
+        device_list_path=os.getenv("DEVICE_LIST_PATH", "devicelist.env"),
+        device_account=os.getenv("DEVICE_ACCOUNT", "21cn"),
+        stream_method=os.getenv("STREAM_METHOD", "capacity.geye.device.devUrl.get"),
+        detr_model=os.getenv("DETR_MODEL", "facebook/detr-resnet-50"),
+        confidence=float(os.getenv("DETR_CONFIDENCE", "0.6")),
+        frame_skip=max(1, int(os.getenv("FRAME_SKIP", "3"))),
+        jpeg_quality=min(100, max(1, int(os.getenv("JPEG_QUALITY", "80")))),
+        resize_width=_optional_int("RESIZE_WIDTH"),
+        vehicle_labels=vehicle_labels,
+    )
+
+
+def mask_url(url: str) -> str:
+    if "token=" not in url:
+        return url
+    prefix, _ = url.split("token=", 1)
+    return f"{prefix}token=***"