200 files

custom_components/maintenance_supporter/websocket/tasks_crud.py

@@ -105,6 +105,7 @@ TASK_UPDATE_FIELD_MAP = {
     "nfc_tag_id": "nfc_tag_id",
     "require_tag_scan": "require_tag_scan",
     "allow_skip": "allow_skip",
+    "notify_enabled": "notify_enabled",
     "reading_unit": "reading_unit",
     "readings": "readings",
     "consumes_parts": "consumes_parts",
@@ -200,6 +201,10 @@ _TASK_CREATE_SCHEMA: dict[Any, Any] =     {
         # #150: per-task skip lock — false hides Skip in the UIs and the
         # coordinator refuses (WS + voice), so automations cannot skip either.
         vol.Optional("allow_skip"): vol.Any(bool, None),
+        # #173: per-task notification mute — false = no reminders for this
+        # task (status changes, repeats, lead-time, bundles); the dashboard
+        # and entities still show it.
+        vol.Optional("notify_enabled"): vol.Any(bool, None),
         # v2.20 (#83): unit for `reading`-type tasks ("kWh", "m³", ...).
         vol.Optional("reading_unit"): vol.Any(vol.All(str, vol.Length(max=MAX_READING_UNIT_LENGTH)), None),
         # #161 phase 2: reading slots [{id?, name, unit?}] — shape-validated
@@ -383,6 +388,9 @@ async def ws_create_task(
     # #150: stored only when False — absence means skipping is allowed.
     if msg.get("allow_skip") is False:
         task_data["allow_skip"] = False
+    # #173: same shape — stored only when False (absence = notifications on).
+    if msg.get("notify_enabled") is False:
+        task_data["notify_enabled"] = False
     # v2.20 (#83): unit for `reading`-type tasks.
     if msg.get("reading_unit") is not None:
         task_data["reading_unit"] = (msg["reading_unit"] or "").strip() or None
@@ -492,6 +500,10 @@ _TASK_UPDATE_SCHEMA: dict[Any, Any] =     {
         # #150: per-task skip lock — false hides Skip in the UIs and the
         # coordinator refuses (WS + voice), so automations cannot skip either.
         vol.Optional("allow_skip"): vol.Any(bool, None),
+        # #173: per-task notification mute — false = no reminders for this
+        # task (status changes, repeats, lead-time, bundles); the dashboard
+        # and entities still show it.
+        vol.Optional("notify_enabled"): vol.Any(bool, None),
         # v2.20 (#83): unit for `reading`-type tasks ("kWh", "m³", ...).
         vol.Optional("reading_unit"): vol.Any(vol.All(str, vol.Length(max=MAX_READING_UNIT_LENGTH)), None),
         # #161 phase 2: reading slots [{id?, name, unit?}] — shape-validated
@@ -612,6 +624,12 @@ async def ws_update_task(
             task["allow_skip"] = False
         else:
             task.pop("allow_skip", None)
+    # #173: notify_enabled follows the same only-when-False rule.
+    if "notify_enabled" in msg:
+        if msg["notify_enabled"] is False:
+            task["notify_enabled"] = False
+        else:
+            task.pop("notify_enabled", None)
 
     # The loop above copies values verbatim, which is wrong for part links:
     # `task/create` validates them and `task/update` did not, so an edit could
@@ -925,3 +943,50 @@ async def ws_duplicate_task(
     await async_persist_task(hass, entry, new_task)
 
     connection.send_result(msg["id"], {"task_id": new_task["id"]})
+
+
+@websocket_api.websocket_command(
+    {
+        vol.Required("type"): f"{DOMAIN}/task/move",
+        vol.Required("entry_id"): str,
+        vol.Required("task_id"): str,
+        vol.Required("target_entry_id"): str,
+    }
+)
+@require_write
+@websocket_api.async_response
+async def ws_move_task(
+    hass: HomeAssistant,
+    connection: websocket_api.ActiveConnection,
+    msg: dict[str, Any],
+) -> None:
+    """Move a task to another object — config, history, readings and trigger
+    state travel with it; the task gets a new reference number and its
+    entities are recreated under the target object (forum #23)."""
+    entry = _load_object_entry(hass, connection, msg)
+    if entry is None:
+        return
+    if msg["task_id"] not in entry.data.get(CONF_TASKS, {}):
+        connection.send_error(msg["id"], "not_found", "Task not found")
+        return
+    target = _load_object_entry(
+        hass, connection, {**msg, "entry_id": msg["target_entry_id"]}, not_found_message="Target object not found"
+    )
+    if target is None:
+        return
+    if target.entry_id == entry.entry_id:
+        connection.send_error(msg["id"], "invalid_target", "The task already belongs to that object")
+        return
+    if (target.data.get(CONF_OBJECT) or {}).get("archived_at"):
+        connection.send_error(msg["id"], "invalid_target", "The target object is archived")
+        return
+    from .tasks_persist import async_move_task
+
+    try:
+        await async_move_task(hass, entry, target, msg["task_id"])
+    except ValueError as err:
+        connection.send_error(msg["id"], "limit_reached", str(err))
+        return
+    await hass.config_entries.async_reload(entry.entry_id)
+    await hass.config_entries.async_reload(target.entry_id)
+    connection.send_result(msg["id"], {"task_id": msg["task_id"], "entry_id": target.entry_id})