添加下载模板功能

ui/components/widget.py

@@ -1,8 +1,10 @@
 import random
+from typing import Union
 
 from PySide6.QtCore import QTimer, Qt, Signal
-from PySide6.QtWidgets import QWidget, QVBoxLayout, QFrame
-from qfluentwidgets import DisplayLabel, LargeTitleLabel
+from PySide6.QtGui import QIcon
+from PySide6.QtWidgets import QWidget, QVBoxLayout, QFrame, QLayout
+from qfluentwidgets import DisplayLabel, LargeTitleLabel, GroupHeaderCardWidget, FluentIconBase, CardGroupWidget
 
 from module.picker.schema import PickerStudent
 
@@ -68,3 +70,27 @@ class RollingTextWidget(QWidget):
     def clear_text(self):
         self.soLabel.clear()
         self.nameLabel.clear()
+
+
+class MyCardGroupWidget(CardGroupWidget):
+    def addLayout(self, layout: QLayout, stretch=0):
+        self.hBoxLayout.addLayout(layout, stretch=stretch)
+
+
+class MyGroupHeaderCardWidget(GroupHeaderCardWidget):
+    def addGroup(self, icon: Union[str, FluentIconBase, QIcon], title: str, content: str,
+                 object: Union[QWidget, QLayout],
+                 stretch=0) -> CardGroupWidget:
+        group = MyCardGroupWidget(icon, title, content, self)
+
+        if isinstance(object, QWidget):
+            group.addWidget(object, stretch=stretch)
+        elif isinstance(object, QLayout):
+            group.addLayout(object, stretch=stretch)
+
+        if self.groupWidgets:
+            self.groupWidgets[-1].setSeparatorVisible(True)
+
+        self.groupLayout.addWidget(group)
+        self.groupWidgets.append(group)
+        return group