Build Pigway Frame Flow with bilingual UI, complete KoKonna driver and safe automation

pigway_frame_flow/drivers/kokonna/mock.py

@@ -0,0 +1,108 @@
+"""Persistent mock implements the documented upload-immediately-activates behavior."""
+
+import json
+
+from PIL import Image
+
+from pigway_frame_flow.errors import ErrorCode as E
+from pigway_frame_flow.errors import FlowError
+from pigway_frame_flow.storage.database import now_iso
+
+
+class MockClient:
+    def __init__(self, root):
+        self.root = root
+        self.path = root / "mock-device.json"
+        if self.path.exists():
+            self.state = json.loads(self.path.read_text())
+        else:
+            self.state = {
+                "current": 1,
+                "counter": 1,
+                "images": [
+                    {"id": 1, "name": "Original Art.jpg", "createdAt": now_iso()}
+                ],
+            }
+            img = Image.new("RGB", (1200, 1600), "#efe2ca")
+            img.save(root / "mock-1.jpg")
+            self.save()
+        self.online = True
+        self.synced = True
+        self.uploads = 0
+        self.displays = []
+
+    def save(self):
+        temporary = self.path.with_suffix(".tmp")
+        temporary.write_text(json.dumps(self.state))
+        temporary.replace(self.path)
+
+    def get_device(self):
+        return {
+            "nickname": "客厅相框",
+            "model": "13.3-inch",
+            "modelCode": "B101",
+            "firmware": "MOCK-B101",
+            "online": self.online,
+            "lastHeartbeat": now_iso(),
+            "batteryLevel": 40,
+            "isCharging": False,
+            "screenWidth": 1200,
+            "screenHeight": 1600,
+            "screenRotate": 270,
+            "timezone": "Asia/Shanghai",
+            "switchType": "queue",
+            "switchMinute": 60,
+            "sdUsedSize": 123456,
+            "imageId": self.state["current"],
+            "coverId": None,
+            "synced": self.synced,
+        }
+
+    def list_images(self):
+        return {
+            "total": len(self.state["images"]),
+            "list": [
+                dict(x, current=x["id"] == self.state["current"])
+                for x in reversed(self.state["images"])
+            ],
+        }
+
+    def upload(self, path, name):
+        self.uploads += 1
+        self.state["counter"] += 1
+        image_id = self.state["counter"]
+        (self.root / f"mock-{image_id}.jpg").write_bytes(path.read_bytes())
+        self.state["images"].append(
+            {"id": image_id, "name": name, "createdAt": now_iso()}
+        )
+        self.state["current"] = image_id
+        self.save()
+        return image_id
+
+    def display(self, image_id):
+        if image_id not in [x["id"] for x in self.state["images"]]:
+            raise FlowError(E.NOT_FOUND, "Image not found", 404)
+        self.displays.append(image_id)
+        self.state["current"] = image_id
+        self.save()
+
+    def delete(self, image_id):
+        self.state["images"] = [x for x in self.state["images"] if x["id"] != image_id]
+        if self.state["current"] == image_id:
+            self.state["current"] = (
+                self.state["images"][-1]["id"] if self.state["images"] else None
+            )
+        (self.root / f"mock-{image_id}.jpg").unlink(missing_ok=True)
+        self.save()
+
+    def download(self, image_id):
+        path = self.root / f"mock-{image_id}.jpg"
+        if not path.exists():
+            raise FlowError(E.NOT_FOUND, "Image not found", 404)
+        return path.read_bytes()
+
+    def display_by_name(self, name):
+        matches = [x for x in self.state["images"] if x["name"] == name]
+        if not matches:
+            raise FlowError(E.NOT_FOUND, "Image not found", 404)
+        return self.display(matches[-1]["id"])