Overload Fragment.repaintEvent to just calculate size

lib/definition.py

@@ -1,6 +1,5 @@
 import re
-import copy
-from typing import Any, Optional, cast
+from typing import Any, Optional, cast, overload
 import re
 from PyQt6.QtCore import QMargins, QPoint, QRect, QSize, QUrl, Qt, pyqtSignal
 from PyQt6.QtGui import QColor, QFont, QFontMetrics, QMouseEvent, QPaintEvent, QPainter, QResizeEvent, QTextOption, QTransform, QBrush
@@ -45,21 +44,7 @@ class Fragment:
         return self.__repr__()
 
     def size(self, width: int) -> QSize:
-        pos = self._position
-        pos.setX(self._indent * 30)
-        rect = QRect(pos, QSize(width - self._position.x(), 2000))
-        flags = (
-            Qt.AlignmentFlag.AlignLeft
-            | Qt.AlignmentFlag.AlignBaseline
-            | Qt.TextFlag.TextWordWrap
-        )
-        fm = QFontMetrics(self._font)
-        bounding = fm.boundingRect(rect, flags, self._text)
-        size = bounding.size()
-        size = size.grownBy(self._padding)
-        size = size.grownBy(self._border)
-        size = size.grownBy(self._margin)
-        return size
+        return self.repaintEvent(width)
 
     def height(self, width: int) -> int:
         return self.size(width).height()
@@ -70,16 +55,28 @@ class Fragment:
     def __repr__(self) -> str:
         return f"({self._position.x()}, {self._position.y()}): {self._text}"
 
-    def repaintEvent(self, painter: QPainter) -> int:
-        painter.save()
-        painter.setFont(self._font)
+    @overload
+    def repaintEvent(self, widthSrc:int) -> QSize:
+        ...
+    @overload
+    def repaintEvent(self, widthSrc: QPainter) -> int:
+        ...
+
+    def repaintEvent(self, widthSrc) -> int|QSize:
+        if isinstance(widthSrc, QPainter):
+            viewportWidth = widthSrc.viewport().width()
+            painter = widthSrc
+        else:
+            viewportWidth = widthSrc
+            painter = None
+        fm = QFontMetrics(self._font)
         top = (
             self._position.y()
-            + painter.fontMetrics().descent()
-            - painter.fontMetrics().height()
+            + fm.descent()
+            - fm.height()
         )
         left = self._position.x()
-        width = painter.viewport().width() - left
+        width = viewportWidth - left
         height = 2000
         rect = QRect(left, top, width, height)
         indent = self._indent * 30
@@ -87,7 +84,7 @@ class Fragment:
             Qt.AlignmentFlag.AlignLeft
             | Qt.AlignmentFlag.AlignBaseline
         )
-        boundingNoWrap = painter.fontMetrics().boundingRect(
+        boundingNoWrap = fm.boundingRect(
             rect, flags|Qt.TextFlag.TextSingleLine, self._text
         )
         flags = (
@@ -95,7 +92,7 @@ class Fragment:
             | Qt.AlignmentFlag.AlignBaseline
             | Qt.TextFlag.TextWordWrap
         )
-        bounding = painter.fontMetrics().boundingRect(
+        bounding = fm.boundingRect(
             rect, flags|Qt.TextFlag.TextWordWrap, self._text
         )
         text = self._text
@@ -110,7 +107,7 @@ class Fragment:
             while pos < rect.right():
                 if text[char] == ' ':
                     lastSpace = char
-                pos += painter.fontMetrics().horizontalAdvance(
+                pos += fm.horizontalAdvance(
                     text[char]
                 )
                 char += 1
@@ -119,6 +116,27 @@ class Fragment:
 
         size = boundingNoWrap.size()
 
+        if remainingText != '':
+            top += size.height()
+            remainingRect = QRect(
+                indent, top,
+                viewportWidth - indent, height
+            )
+            size = size.grownBy(
+                QMargins(
+                    0,0,0,
+                    fm.boundingRect(
+                        remainingRect, flags | Qt.TextFlag.TextWordWrap, remainingText
+                    ).height()
+                )
+            )
+        size = size.grownBy(self._margin)
+        size = size.grownBy(self._border)
+        size = size.grownBy(self._padding)
+        if painter is None:
+            return size
+        painter.save()
+        painter.setFont(self._font)
         painter.setPen(QColor("#f00"))
         if self._audio.isValid():
             radius = self._borderRect.height() / 2
@@ -129,25 +147,9 @@ class Fragment:
             painter.drawLine(start, end)
         painter.setPen(self._color)
         painter.drawText(rect, flags, text)
-        if remainingText != '':
-            top += size.height()
-            rect = QRect(
-                indent, top,
-                painter.viewport().width() - indent, height
-            )
-            painter.drawText(rect, flags|Qt.TextFlag.TextWordWrap, remainingText)
-            size = size.grownBy(
-                QMargins(
-                    0,0,0,
-                    painter.fontMetrics().boundingRect(
-                        rect, flags | Qt.TextFlag.TextWordWrap, remainingText
-                    ).height()
-                )
-            )
+        if remainingText:
+            painter.drawText(remainingRect, flags|Qt.TextFlag.TextWordWrap, remainingText)
         painter.restore()
-        size = size.grownBy(self._margin)
-        size = size.grownBy(self._border)
-        size = size.grownBy(self._padding)
         return size.height()
 
     #