337 lines
10 KiB
Python
337 lines
10 KiB
Python
|
"""
|
|||
|
|
Dashboard 组件布局模型
|
|||
|
|
|
|||
|
|
- 管理一组可视化组件（widget）的实例：类型、位置、尺寸、层级、配置
|
|||
|
|
- 支持：新增/删除/选中/移动/缩放/修改配置/保存加载(JSON)
|
|||
|
|
|
|||
|
|
说明：
|
|||
|
|
该模型面向 QML，使用 QAbstractListModel 暴露 role。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import uuid
|
|||
|
|
from dataclasses import dataclass, asdict, field
|
|||
|
|
from typing import Any, Dict, List, Optional
|
|||
|
|
|
|||
|
|
from PyQt6.QtCore import (
|
|||
|
|
QAbstractListModel,
|
|||
|
|
QModelIndex,
|
|||
|
|
Qt,
|
|||
|
|
pyqtSignal,
|
|||
|
|
pyqtSlot,
|
|||
|
|
pyqtProperty,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class WidgetItem:
|
|||
|
|
id: str
|
|||
|
|
type: str # e.g. "image_overlay", "web_chart"
|
|||
|
|
x: float = 0.0
|
|||
|
|
y: float = 0.0
|
|||
|
|
w: float = 400.0
|
|||
|
|
h: float = 300.0
|
|||
|
|
z: int = 0
|
|||
|
|
locked: bool = False
|
|||
|
|
props: Dict[str, Any] = field(default_factory=dict)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DashboardModel(QAbstractListModel):
|
|||
|
|
"""
|
|||
|
|
QML 用的组件列表模型。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
selectedIndexChanged = pyqtSignal()
|
|||
|
|
dirtyChanged = pyqtSignal()
|
|||
|
|
countChanged = pyqtSignal()
|
|||
|
|
|
|||
|
|
# 重要：role 名避免和 QML Item 的内置属性冲突（x/y/width/height 等）
|
|||
|
|
ROLE_UID = Qt.ItemDataRole.UserRole + 1
|
|||
|
|
ROLE_WIDGET_TYPE = Qt.ItemDataRole.UserRole + 2
|
|||
|
|
ROLE_POS_X = Qt.ItemDataRole.UserRole + 3
|
|||
|
|
ROLE_POS_Y = Qt.ItemDataRole.UserRole + 4
|
|||
|
|
ROLE_SIZE_W = Qt.ItemDataRole.UserRole + 5
|
|||
|
|
ROLE_SIZE_H = Qt.ItemDataRole.UserRole + 6
|
|||
|
|
ROLE_Z = Qt.ItemDataRole.UserRole + 7
|
|||
|
|
ROLE_LOCKED = Qt.ItemDataRole.UserRole + 8
|
|||
|
|
ROLE_PROPS = Qt.ItemDataRole.UserRole + 9
|
|||
|
|
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self._items: List[WidgetItem] = []
|
|||
|
|
self._selected_index: int = -1
|
|||
|
|
self._dirty: bool = False
|
|||
|
|
self._default_file: str = "dashboard.json"
|
|||
|
|
|
|||
|
|
# ---------- Qt model ----------
|
|||
|
|
def rowCount(self, parent=QModelIndex()) -> int:
|
|||
|
|
if parent.isValid():
|
|||
|
|
return 0
|
|||
|
|
return len(self._items)
|
|||
|
|
|
|||
|
|
@pyqtProperty(int, notify=countChanged)
|
|||
|
|
def count(self) -> int:
|
|||
|
|
return len(self._items)
|
|||
|
|
|
|||
|
|
def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):
|
|||
|
|
if not index.isValid():
|
|||
|
|
return None
|
|||
|
|
i = index.row()
|
|||
|
|
if i < 0 or i >= len(self._items):
|
|||
|
|
return None
|
|||
|
|
item = self._items[i]
|
|||
|
|
if role == self.ROLE_UID:
|
|||
|
|
return item.id
|
|||
|
|
if role == self.ROLE_WIDGET_TYPE:
|
|||
|
|
return item.type
|
|||
|
|
if role == self.ROLE_POS_X:
|
|||
|
|
return float(item.x)
|
|||
|
|
if role == self.ROLE_POS_Y:
|
|||
|
|
return float(item.y)
|
|||
|
|
if role == self.ROLE_SIZE_W:
|
|||
|
|
return float(item.w)
|
|||
|
|
if role == self.ROLE_SIZE_H:
|
|||
|
|
return float(item.h)
|
|||
|
|
if role == self.ROLE_Z:
|
|||
|
|
return int(item.z)
|
|||
|
|
if role == self.ROLE_LOCKED:
|
|||
|
|
return bool(item.locked)
|
|||
|
|
if role == self.ROLE_PROPS:
|
|||
|
|
return item.props
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def roleNames(self):
|
|||
|
|
return {
|
|||
|
|
self.ROLE_UID: b"uid",
|
|||
|
|
self.ROLE_WIDGET_TYPE: b"widgetType",
|
|||
|
|
self.ROLE_POS_X: b"posX",
|
|||
|
|
self.ROLE_POS_Y: b"posY",
|
|||
|
|
self.ROLE_SIZE_W: b"sizeW",
|
|||
|
|
self.ROLE_SIZE_H: b"sizeH",
|
|||
|
|
# 避免和 QML Item.z 冲突（该属性在部分版本上是 FINAL，不能在 delegate 里声明同名 required）
|
|||
|
|
self.ROLE_Z: b"layerZ",
|
|||
|
|
self.ROLE_LOCKED: b"locked",
|
|||
|
|
self.ROLE_PROPS: b"props",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# ---------- properties ----------
|
|||
|
|
@pyqtProperty(int, notify=selectedIndexChanged)
|
|||
|
|
def selectedIndex(self) -> int:
|
|||
|
|
return self._selected_index
|
|||
|
|
|
|||
|
|
def _set_dirty(self, v: bool):
|
|||
|
|
if self._dirty != v:
|
|||
|
|
self._dirty = v
|
|||
|
|
self.dirtyChanged.emit()
|
|||
|
|
|
|||
|
|
@pyqtProperty(bool, notify=dirtyChanged)
|
|||
|
|
def dirty(self) -> bool:
|
|||
|
|
return self._dirty
|
|||
|
|
|
|||
|
|
# ---------- helpers ----------
|
|||
|
|
def _touch_item(self, row: int):
|
|||
|
|
if 0 <= row < len(self._items):
|
|||
|
|
model_index = self.index(row, 0)
|
|||
|
|
self.dataChanged.emit(model_index, model_index, [])
|
|||
|
|
self._set_dirty(True)
|
|||
|
|
|
|||
|
|
def _new_id(self) -> str:
|
|||
|
|
return uuid.uuid4().hex
|
|||
|
|
|
|||
|
|
def _max_z(self) -> int:
|
|||
|
|
if not self._items:
|
|||
|
|
return 0
|
|||
|
|
return max(x.z for x in self._items)
|
|||
|
|
|
|||
|
|
# ---------- QML slots ----------
|
|||
|
|
@pyqtSlot(str, result=int)
|
|||
|
|
def addWidget(self, widget_type: str) -> int:
|
|||
|
|
"""
|
|||
|
|
添加一个 widget，返回 index。
|
|||
|
|
widget_type: "image_overlay" | "web_chart" | ...
|
|||
|
|
"""
|
|||
|
|
defaults: Dict[str, Any] = {}
|
|||
|
|
if widget_type == "image_overlay":
|
|||
|
|
defaults = {
|
|||
|
|
"imagePath": "",
|
|||
|
|
# 存 DataModel 的 json（仅 textItems）
|
|||
|
|
"textItemsJson": "",
|
|||
|
|
# Influx 查询可按组件覆盖（可选）
|
|||
|
|
"query": "",
|
|||
|
|
}
|
|||
|
|
elif widget_type == "web_chart":
|
|||
|
|
defaults = {
|
|||
|
|
"url": "",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
item = WidgetItem(
|
|||
|
|
id=self._new_id(),
|
|||
|
|
type=widget_type,
|
|||
|
|
x=20.0,
|
|||
|
|
y=20.0,
|
|||
|
|
w=500.0,
|
|||
|
|
h=350.0,
|
|||
|
|
z=len(self._items),
|
|||
|
|
locked=False,
|
|||
|
|
props=defaults,
|
|||
|
|
)
|
|||
|
|
self.beginInsertRows(QModelIndex(), len(self._items), len(self._items))
|
|||
|
|
self._items.append(item)
|
|||
|
|
self.endInsertRows()
|
|||
|
|
self.countChanged.emit()
|
|||
|
|
self.select(len(self._items) - 1)
|
|||
|
|
self._set_dirty(True)
|
|||
|
|
return len(self._items) - 1
|
|||
|
|
|
|||
|
|
@pyqtSlot(int)
|
|||
|
|
def remove(self, index: int):
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
self.beginRemoveRows(QModelIndex(), index, index)
|
|||
|
|
self._items.pop(index)
|
|||
|
|
self.endRemoveRows()
|
|||
|
|
self.countChanged.emit()
|
|||
|
|
if self._selected_index == index:
|
|||
|
|
self._selected_index = -1
|
|||
|
|
self.selectedIndexChanged.emit()
|
|||
|
|
elif self._selected_index > index:
|
|||
|
|
self._selected_index -= 1
|
|||
|
|
self.selectedIndexChanged.emit()
|
|||
|
|
self._set_dirty(True)
|
|||
|
|
|
|||
|
|
@pyqtSlot(int)
|
|||
|
|
def select(self, index: int):
|
|||
|
|
if index != self._selected_index:
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
self._selected_index = index
|
|||
|
|
else:
|
|||
|
|
self._selected_index = -1
|
|||
|
|
self.selectedIndexChanged.emit()
|
|||
|
|
|
|||
|
|
@pyqtSlot(int)
|
|||
|
|
def bringToFront(self, index: int):
|
|||
|
|
"""将指定组件置顶，避免拖拽/新建后层级混乱。"""
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
max_z = self._max_z()
|
|||
|
|
it = self._items[index]
|
|||
|
|
if it.z < max_z:
|
|||
|
|
it.z = max_z + 1
|
|||
|
|
self._touch_item(index)
|
|||
|
|
|
|||
|
|
@pyqtSlot(int, float, float, float, float)
|
|||
|
|
def setGeometry(self, index: int, x: float, y: float, w: float, h: float):
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
it = self._items[index]
|
|||
|
|
if it.locked:
|
|||
|
|
return
|
|||
|
|
it.x = float(x)
|
|||
|
|
it.y = float(y)
|
|||
|
|
it.w = max(20.0, float(w))
|
|||
|
|
it.h = max(20.0, float(h))
|
|||
|
|
self._touch_item(index)
|
|||
|
|
|
|||
|
|
@pyqtSlot(int, str, "QVariant")
|
|||
|
|
def setProp(self, index: int, key: str, value):
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
it = self._items[index]
|
|||
|
|
it.props[key] = value
|
|||
|
|
self._touch_item(index)
|
|||
|
|
|
|||
|
|
@pyqtSlot(int, str, result="QVariant")
|
|||
|
|
def getProp(self, index: int, key: str):
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
return self._items[index].props.get(key)
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
@pyqtSlot(result=str)
|
|||
|
|
def toJson(self) -> str:
|
|||
|
|
data = {"widgets": [asdict(x) for x in self._items]}
|
|||
|
|
return json.dumps(data, ensure_ascii=False, indent=2)
|
|||
|
|
|
|||
|
|
@pyqtSlot(str)
|
|||
|
|
def fromJson(self, json_str: str):
|
|||
|
|
try:
|
|||
|
|
data = json.loads(json_str)
|
|||
|
|
widgets = data.get("widgets", [])
|
|||
|
|
parsed: List[WidgetItem] = []
|
|||
|
|
for w in widgets:
|
|||
|
|
parsed.append(
|
|||
|
|
WidgetItem(
|
|||
|
|
id=str(w.get("id") or self._new_id()),
|
|||
|
|
type=str(w.get("type") or "image_overlay"),
|
|||
|
|
x=float(w.get("x", 0.0)),
|
|||
|
|
y=float(w.get("y", 0.0)),
|
|||
|
|
w=float(w.get("w", 400.0)),
|
|||
|
|
h=float(w.get("h", 300.0)),
|
|||
|
|
z=int(w.get("z", 0)),
|
|||
|
|
locked=bool(w.get("locked", False)),
|
|||
|
|
props=dict(w.get("props") or {}),
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
self.beginResetModel()
|
|||
|
|
self._items = parsed
|
|||
|
|
self.endResetModel()
|
|||
|
|
self.countChanged.emit()
|
|||
|
|
self.select(-1)
|
|||
|
|
self._set_dirty(False)
|
|||
|
|
except Exception:
|
|||
|
|
# ignore invalid json
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
@pyqtSlot(int, result="QVariant")
|
|||
|
|
def getItem(self, index: int):
|
|||
|
|
"""
|
|||
|
|
便于 QML 属性面板读取当前条目（避免 role 名冲突/role id 取值问题）。
|
|||
|
|
"""
|
|||
|
|
if 0 <= index < len(self._items):
|
|||
|
|
it = self._items[index]
|
|||
|
|
return {
|
|||
|
|
"uid": it.id,
|
|||
|
|
"widgetType": it.type,
|
|||
|
|
"posX": float(it.x),
|
|||
|
|
"posY": float(it.y),
|
|||
|
|
"sizeW": float(it.w),
|
|||
|
|
"sizeH": float(it.h),
|
|||
|
|
"z": int(it.z),
|
|||
|
|
"locked": bool(it.locked),
|
|||
|
|
"props": dict(it.props),
|
|||
|
|
}
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
@pyqtSlot(result=str)
|
|||
|
|
def debugSummary(self) -> str:
|
|||
|
|
return f"count={len(self._items)} selectedIndex={self._selected_index}"
|
|||
|
|
|
|||
|
|
@pyqtSlot(result=bool)
|
|||
|
|
def load(self) -> bool:
|
|||
|
|
return self.loadFromFile(self._default_file)
|
|||
|
|
|
|||
|
|
@pyqtSlot(result=bool)
|
|||
|
|
def save(self) -> bool:
|
|||
|
|
return self.saveToFile(self._default_file)
|
|||
|
|
|
|||
|
|
@pyqtSlot(str, result=bool)
|
|||
|
|
def loadFromFile(self, path: str) -> bool:
|
|||
|
|
try:
|
|||
|
|
if not os.path.exists(path):
|
|||
|
|
return False
|
|||
|
|
with open(path, "r", encoding="utf-8") as f:
|
|||
|
|
self.fromJson(f.read())
|
|||
|
|
return True
|
|||
|
|
except Exception:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@pyqtSlot(str, result=bool)
|
|||
|
|
def saveToFile(self, path: str) -> bool:
|
|||
|
|
try:
|
|||
|
|
with open(path, "w", encoding="utf-8") as f:
|
|||
|
|
f.write(self.toJson())
|
|||
|
|
self._set_dirty(False)
|
|||
|
|
return True
|
|||
|
|
except Exception:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|