mirror of
https://github.com/Matysh/houseplan-card
synced 2026-07-31 08:28:31 +00:00
84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
"""HTTP-эндпоинт загрузки файлов-инструкций House Plan.
|
|
|
|
Файлы (PDF и т.п.) грузятся не через WebSocket (у него лимит размера сообщения —
|
|
большой PDF рвёт соединение), а обычным multipart POST — как медиа в самом HA.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
from aiohttp import web
|
|
|
|
from homeassistant.components.http import HomeAssistantView
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .const import CONF_ADMIN_ONLY, DOMAIN, FILES_DIR, FILES_URL
|
|
from .validation import (
|
|
FILE_EXTENSIONS,
|
|
MAX_FILE_BYTES,
|
|
file_ext,
|
|
sanitize_filename,
|
|
sanitize_marker_id,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class HouseplanUploadView(HomeAssistantView):
|
|
"""POST /api/houseplan/upload — сохранить файл маркера, вернуть URL."""
|
|
|
|
url = "/api/houseplan/upload"
|
|
name = "api:houseplan:upload"
|
|
requires_auth = True
|
|
|
|
async def post(self, request: web.Request) -> web.Response:
|
|
hass: HomeAssistant = request.app["hass"]
|
|
entry = hass.data.get(DOMAIN, {}).get("entry")
|
|
admin_only = bool(entry and entry.options.get(CONF_ADMIN_ONLY, False))
|
|
if admin_only:
|
|
user = request.get("hass_user")
|
|
if user is None or not user.is_admin:
|
|
return web.json_response({"error": "unauthorized"}, status=403)
|
|
|
|
marker_id = "misc"
|
|
filename: str | None = None
|
|
blob: bytes | None = None
|
|
try:
|
|
reader = await request.multipart()
|
|
async for part in reader:
|
|
if part.name == "marker_id":
|
|
marker_id = sanitize_marker_id(await part.text())
|
|
elif part.name == "file":
|
|
filename = part.filename or "file"
|
|
blob = await part.read(decode=False)
|
|
except Exception as err: # noqa: BLE001
|
|
_LOGGER.warning("House Plan upload: ошибка чтения multipart: %s", err)
|
|
return web.json_response({"error": "bad_request"}, status=400)
|
|
|
|
if blob is None or not filename:
|
|
return web.json_response({"error": "no_file"}, status=400)
|
|
ext = file_ext(filename)
|
|
if ext not in FILE_EXTENSIONS:
|
|
return web.json_response(
|
|
{"error": "bad_ext", "allowed": sorted(FILE_EXTENSIONS)}, status=400
|
|
)
|
|
if len(blob) > MAX_FILE_BYTES:
|
|
return web.json_response(
|
|
{"error": "too_large", "max_mb": MAX_FILE_BYTES // 1024 // 1024}, status=413
|
|
)
|
|
|
|
safe_name = sanitize_filename(filename)
|
|
target_dir = Path(hass.config.path(FILES_DIR)) / marker_id
|
|
path = target_dir / safe_name
|
|
|
|
def _write() -> None:
|
|
target_dir.mkdir(parents=True, exist_ok=True)
|
|
path.write_bytes(blob)
|
|
|
|
await hass.async_add_executor_job(_write)
|
|
mtime = int(path.stat().st_mtime)
|
|
return web.json_response(
|
|
{"ok": True, "url": f"{FILES_URL}/{marker_id}/{safe_name}?v={mtime}", "name": filename}
|
|
)
|