PyQt/QSystemTrayIcon/TrayNotify.py
2021-12-09 18:46:17 +08:00

71 lines
2.3 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on 2021年12月09日
@author: Irony
@site: https://pyqt.site , https://github.com/PyQt5
@email: 892768447@qq.com
@file: TrayNotify
@description: 托盘闪烁
"""
try:
from PyQt5.QtCore import QTimer
from PyQt5.QtWidgets import (QApplication, QHBoxLayout, QPushButton,
QStyle, QSystemTrayIcon, QWidget)
except ImportError:
from PySide2.QtCore import QTimer
from PySide2.QtWidgets import (QApplication, QHBoxLayout, QPushButton,
QStyle, QSystemTrayIcon, QWidget)
class Window(QWidget):
def __init__(self, *args, **kwargs):
super(Window, self).__init__(*args, **kwargs)
layout = QHBoxLayout(self)
layout.addWidget(QPushButton('开始闪烁', self, clicked=self.start_flash))
layout.addWidget(QPushButton('停止闪烁', self, clicked=self.stop_flash))
# 创建托盘图标
self.tray_icon = QSystemTrayIcon(self)
self.tray_icon.setIcon(self.style().standardIcon(
QStyle.SP_ComputerIcon))
self.tray_icon.show()
# 图标闪烁定时器
self.tray_visible = True
self.flash_timer = QTimer(self, timeout=self.flash_icon)
def closeEvent(self, event):
self.stop_flash()
self.tray_icon.hide()
super(Window, self).closeEvent(event)
def start_flash(self):
"""开始闪烁"""
if not self.flash_timer.isActive():
self.flash_timer.start(500)
def stop_flash(self):
"""停止闪烁后需要显示图标"""
if self.flash_timer.isActive():
self.flash_timer.stop()
self.tray_icon.setIcon(self.style().standardIcon(
QStyle.SP_ComputerIcon))
def flash_icon(self):
"""根据当前图标是否可见切换图标"""
if self.tray_visible:
self.tray_icon.setIcon(self.style().standardIcon(
QStyle.SP_TrashIcon))
else:
self.tray_icon.setIcon(self.style().standardIcon(
QStyle.SP_ComputerIcon))
self.tray_visible = not self.tray_visible
if __name__ == '__main__':
import sys
app = QApplication(sys.argv)
w = Window()
w.show()
sys.exit(app.exec_())