364 lines
12 KiB
Python
364 lines
12 KiB
Python
# coding:utf-8
|
||
import sys
|
||
from typing import Union
|
||
|
||
from PyQt6.QtCore import QRect, QSize, Qt
|
||
from PyQt6.QtGui import QColor, QIcon, QPainter, QPixmap
|
||
from PyQt6.QtWidgets import QApplication, QHBoxLayout, QLabel, QVBoxLayout, QWidget
|
||
from qfluentwidgets.common.animation import BackgroundAnimationWidget
|
||
from qfluentwidgets.common.config import qconfig
|
||
from qfluentwidgets.common.icon import FluentIconBase
|
||
from qfluentwidgets.common.router import qrouter
|
||
from qfluentwidgets.common.style_sheet import (
|
||
FluentStyleSheet,
|
||
isDarkTheme,
|
||
)
|
||
from qfluentwidgets.components.navigation import (
|
||
NavigationInterface,
|
||
NavigationItemPosition,
|
||
NavigationTreeWidget,
|
||
)
|
||
from qfluentwidgets.components.widgets.frameless_window import FramelessWindow
|
||
from qframelesswindow import TitleBar, TitleBarBase
|
||
|
||
from app.view.widgets.stacked_widget import StackedWidget
|
||
|
||
|
||
class CustomFluentWindowBase(BackgroundAnimationWidget, FramelessWindow):
|
||
"""Fluent window base class"""
|
||
|
||
def __init__(self, parent=None):
|
||
self._isMicaEnabled = False
|
||
self._lightBackgroundColor = QColor(240, 244, 249)
|
||
self._darkBackgroundColor = QColor(32, 32, 32)
|
||
self._backgroundPixmap = None # 存储背景图片
|
||
self._backgroundOpacity = 1.0 # 背景图片不透明度
|
||
super().__init__(parent=parent)
|
||
|
||
self.hBoxLayout = QHBoxLayout(self)
|
||
self.stackedWidget = StackedWidget(self)
|
||
self.navigationInterface = None
|
||
|
||
# initialize layout
|
||
self.hBoxLayout.setSpacing(0)
|
||
self.hBoxLayout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
FluentStyleSheet.FLUENT_WINDOW.apply(self.stackedWidget)
|
||
|
||
# enable mica effect on win11
|
||
self.setMicaEffectEnabled(True)
|
||
|
||
# show system title bar buttons on macOS
|
||
if sys.platform == "darwin":
|
||
self.setSystemTitleBarButtonVisible(True)
|
||
|
||
qconfig.themeChangedFinished.connect(self._onThemeChangedFinished)
|
||
|
||
def addSubInterface(
|
||
self,
|
||
interface: QWidget,
|
||
icon: Union[FluentIconBase, QIcon, str],
|
||
text: str,
|
||
position=NavigationItemPosition.TOP,
|
||
):
|
||
"""add sub interface"""
|
||
raise NotImplementedError
|
||
|
||
def removeInterface(self, interface: QWidget, isDelete=False):
|
||
"""remove sub interface
|
||
|
||
Parameters
|
||
----------
|
||
interface: QWidget
|
||
sub interface to be removed
|
||
|
||
isDelete: bool
|
||
whether to delete the sub interface
|
||
"""
|
||
raise NotImplementedError
|
||
|
||
def switchTo(self, interface: QWidget):
|
||
self.stackedWidget.setCurrentWidget(interface, popOut=False)
|
||
|
||
def _onCurrentInterfaceChanged(self, index: int):
|
||
widget = self.stackedWidget.widget(index)
|
||
self.navigationInterface.setCurrentItem(widget.objectName())
|
||
qrouter.push(self.stackedWidget, widget.objectName())
|
||
|
||
self._updateStackedBackground()
|
||
|
||
def _updateStackedBackground(self):
|
||
isTransparent = self.stackedWidget.currentWidget().property(
|
||
"isStackedTransparent"
|
||
)
|
||
if bool(self.stackedWidget.property("isTransparent")) == isTransparent:
|
||
return
|
||
|
||
self.stackedWidget.setProperty("isTransparent", isTransparent)
|
||
self.stackedWidget.setStyle(QApplication.style())
|
||
|
||
def setCustomBackgroundColor(self, light, dark):
|
||
"""set custom background color
|
||
|
||
Parameters
|
||
----------
|
||
light, dark: QColor | Qt.GlobalColor | str
|
||
background color in light/dark theme mode
|
||
"""
|
||
self._lightBackgroundColor = QColor(light)
|
||
self._darkBackgroundColor = QColor(dark)
|
||
self._updateBackgroundColor()
|
||
|
||
def setBackgroundImage(self, imagePath: str, opacity: float = 1.0):
|
||
"""设置背景图片
|
||
|
||
Parameters
|
||
----------
|
||
imagePath: str
|
||
背景图片路径
|
||
opacity: float
|
||
背景图片不透明度，范围0.0-1.0
|
||
"""
|
||
self._backgroundPixmap = QPixmap(imagePath)
|
||
if self._backgroundPixmap.isNull():
|
||
print(f"无法加载背景图片: {imagePath}")
|
||
return
|
||
|
||
self._backgroundOpacity = max(0.0, min(1.0, opacity)) # 确保在0-1范围内
|
||
|
||
# 设置StackedWidget为透明，以便显示背景图片
|
||
self.stackedWidget.setProperty("isTransparent", True)
|
||
self.stackedWidget.setStyle(QApplication.style())
|
||
|
||
self.update() # 触发重绘
|
||
|
||
def removeBackgroundImage(self):
|
||
"""移除背景图片"""
|
||
self._backgroundPixmap = None
|
||
|
||
# 恢复StackedWidget的默认背景
|
||
self.stackedWidget.setProperty("isTransparent", False)
|
||
self.stackedWidget.setStyle(QApplication.style())
|
||
|
||
self.update()
|
||
|
||
def _normalBackgroundColor(self):
|
||
if not self.isMicaEffectEnabled():
|
||
return (
|
||
self._darkBackgroundColor
|
||
if isDarkTheme()
|
||
else self._lightBackgroundColor
|
||
)
|
||
|
||
return QColor(0, 0, 0, 0)
|
||
|
||
def _onThemeChangedFinished(self):
|
||
if self.isMicaEffectEnabled():
|
||
self.windowEffect.setMicaEffect(self.winId(), isDarkTheme())
|
||
|
||
def paintEvent(self, e):
|
||
# 创建绘制器
|
||
painter = QPainter(self)
|
||
|
||
# 如果有背景图片，先绘制背景图片
|
||
if self._backgroundPixmap and not self._backgroundPixmap.isNull():
|
||
# 设置不透明度
|
||
painter.setOpacity(self._backgroundOpacity)
|
||
|
||
# 缩放图片以适应窗口大小
|
||
scaled_pixmap = self._backgroundPixmap.scaled(
|
||
self.size(), Qt.AspectRatioMode.IgnoreAspectRatio, Qt.TransformationMode.SmoothTransformation
|
||
)
|
||
painter.drawPixmap(0, 0, scaled_pixmap)
|
||
|
||
# 然后调用父类的绘制方法
|
||
super().paintEvent(e)
|
||
|
||
def setMicaEffectEnabled(self, isEnabled: bool):
|
||
"""set whether the mica effect is enabled, only available on Win11"""
|
||
if sys.platform != "win32" or sys.getwindowsversion().build < 22000:
|
||
return
|
||
|
||
self._isMicaEnabled = isEnabled
|
||
|
||
if isEnabled:
|
||
self.windowEffect.setMicaEffect(self.winId(), isDarkTheme())
|
||
# 启用Mica效果时移除背景图片
|
||
self.removeBackgroundImage()
|
||
else:
|
||
self.windowEffect.removeBackgroundEffect(self.winId())
|
||
|
||
self.setBackgroundColor(self._normalBackgroundColor())
|
||
|
||
def isMicaEffectEnabled(self):
|
||
return self._isMicaEnabled
|
||
|
||
def systemTitleBarRect(self, size: QSize) -> QRect:
|
||
"""Returns the system title bar rect, only works for macOS
|
||
|
||
Parameters
|
||
----------
|
||
size: QSize
|
||
original system title bar rect
|
||
"""
|
||
return QRect(
|
||
size.width() - 75, 0 if self.isFullScreen() else 9, 75, size.height()
|
||
)
|
||
|
||
def setTitleBar(self, titleBar):
|
||
super().setTitleBar(titleBar)
|
||
|
||
# hide title bar buttons on macOS
|
||
if (
|
||
sys.platform == "darwin"
|
||
and self.isSystemButtonVisible()
|
||
and isinstance(titleBar, TitleBarBase)
|
||
):
|
||
titleBar.minBtn.hide()
|
||
titleBar.maxBtn.hide()
|
||
titleBar.closeBtn.hide()
|
||
|
||
|
||
class FluentTitleBar(TitleBar):
|
||
"""Fluent title bar"""
|
||
|
||
def __init__(self, parent):
|
||
super().__init__(parent)
|
||
self.setFixedHeight(48)
|
||
self.hBoxLayout.removeWidget(self.minBtn)
|
||
self.hBoxLayout.removeWidget(self.maxBtn)
|
||
self.hBoxLayout.removeWidget(self.closeBtn)
|
||
|
||
# add window icon
|
||
self.iconLabel = QLabel(self)
|
||
self.iconLabel.setFixedSize(18, 18)
|
||
self.hBoxLayout.insertWidget(
|
||
0, self.iconLabel, 0, Qt.AlignmentFlag.AlignLeft | Qt.AlignmentFlag.AlignVCenter
|
||
)
|
||
self.window().windowIconChanged.connect(self.setIcon)
|
||
|
||
# add title label
|
||
self.titleLabel = QLabel(self)
|
||
self.hBoxLayout.insertWidget(
|
||
1, self.titleLabel, 0, Qt.AlignmentFlag.AlignLeft | Qt.AlignmentFlag.AlignVCenter
|
||
)
|
||
self.titleLabel.setObjectName("titleLabel")
|
||
self.window().windowTitleChanged.connect(self.setTitle)
|
||
|
||
self.vBoxLayout = QVBoxLayout()
|
||
self.buttonLayout = QHBoxLayout()
|
||
self.buttonLayout.setSpacing(0)
|
||
self.buttonLayout.setContentsMargins(0, 0, 0, 0)
|
||
self.buttonLayout.setAlignment(Qt.AlignmentFlag.AlignTop)
|
||
self.buttonLayout.addWidget(self.minBtn)
|
||
self.buttonLayout.addWidget(self.maxBtn)
|
||
self.buttonLayout.addWidget(self.closeBtn)
|
||
self.vBoxLayout.addLayout(self.buttonLayout)
|
||
self.vBoxLayout.addStretch(1)
|
||
self.hBoxLayout.addLayout(self.vBoxLayout, 0)
|
||
|
||
FluentStyleSheet.FLUENT_WINDOW.apply(self)
|
||
|
||
def setTitle(self, title):
|
||
self.titleLabel.setText(title)
|
||
self.titleLabel.adjustSize()
|
||
|
||
def setIcon(self, icon):
|
||
self.iconLabel.setPixmap(QIcon(icon).pixmap(18, 18))
|
||
|
||
|
||
class CustomFluentWindow(CustomFluentWindowBase):
|
||
"""Fluent window"""
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.setTitleBar(FluentTitleBar(self))
|
||
|
||
self.navigationInterface = NavigationInterface(self, showReturnButton=True)
|
||
self.widgetLayout = QHBoxLayout()
|
||
|
||
# initialize layout
|
||
self.hBoxLayout.addWidget(self.navigationInterface)
|
||
self.hBoxLayout.addLayout(self.widgetLayout)
|
||
self.hBoxLayout.setStretchFactor(self.widgetLayout, 1)
|
||
|
||
self.widgetLayout.addWidget(self.stackedWidget)
|
||
self.widgetLayout.setContentsMargins(0, 48, 0, 0)
|
||
|
||
self.navigationInterface.displayModeChanged.connect(self.titleBar.raise_)
|
||
self.titleBar.raise_()
|
||
|
||
def addSubInterface(
|
||
self,
|
||
interface: QWidget,
|
||
icon: Union[FluentIconBase, QIcon, str],
|
||
text: str,
|
||
position=NavigationItemPosition.TOP,
|
||
parent=None,
|
||
isTransparent=False,
|
||
) -> NavigationTreeWidget:
|
||
"""add sub interface, the object name of `interface` should be set already
|
||
before calling this method
|
||
|
||
Parameters
|
||
----------
|
||
interface: QWidget
|
||
the subinterface to be added
|
||
|
||
icon: FluentIconBase | QIcon | str
|
||
the icon of navigation item
|
||
|
||
text: str
|
||
the text of navigation item
|
||
|
||
position: NavigationItemPosition
|
||
the position of navigation item
|
||
|
||
parent: QWidget
|
||
the parent of navigation item
|
||
|
||
isTransparent: bool
|
||
whether to use transparent background
|
||
"""
|
||
if not interface.objectName():
|
||
raise ValueError("The object name of `interface` can't be empty string.")
|
||
if parent and not parent.objectName():
|
||
raise ValueError("The object name of `parent` can't be empty string.")
|
||
|
||
interface.setProperty("isStackedTransparent", isTransparent)
|
||
self.stackedWidget.addWidget(interface)
|
||
|
||
# add navigation item
|
||
routeKey = interface.objectName()
|
||
item = self.navigationInterface.addItem(
|
||
routeKey=routeKey,
|
||
icon=icon,
|
||
text=text,
|
||
onClick=lambda: self.switchTo(interface),
|
||
position=position,
|
||
tooltip=text,
|
||
parentRouteKey=parent.objectName() if parent else None,
|
||
)
|
||
|
||
# initialize selected item
|
||
if self.stackedWidget.count() == 1:
|
||
self.stackedWidget.currentChanged.connect(self._onCurrentInterfaceChanged)
|
||
self.navigationInterface.setCurrentItem(routeKey)
|
||
qrouter.setDefaultRouteKey(self.stackedWidget, routeKey)
|
||
|
||
self._updateStackedBackground()
|
||
|
||
return item
|
||
|
||
def removeInterface(self, interface, isDelete=False):
|
||
self.navigationInterface.removeWidget(interface.objectName())
|
||
self.stackedWidget.removeWidget(interface)
|
||
interface.hide()
|
||
|
||
if isDelete:
|
||
interface.deleteLater()
|
||
|
||
def resizeEvent(self, e):
|
||
self.titleBar.move(46, 0)
|
||
self.titleBar.resize(self.width() - 46, self.titleBar.height())
|