#!/usr/bin/python
from PyQt5.QtWidgets import QApplication, QWidget, QLabel, QDesktopWidget
from PyQt5.QtGui import QCursor, QFont
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QTimer
import psutil
import sys
  
class Speed(QWidget):
    def __init__(self):
        super(Speed, self).__init__()
        self.initUI()
        self.worker = MyWorker()
        self.worker.timeout.connect(self.update_speed)
        self.worker.start()
  
    def initUI(self):
        """初始化UI"""
        self.setWindowFlags(Qt.WindowStaysOnTopHint | Qt.FramelessWindowHint)
        self.setAttribute(Qt.WA_TranslucentBackground)
        self.resize(160, 70)
  
        # 上传和下载标签
        self.up = QLabel(self)
        self.down = QLabel(self)
        self.up.setGeometry(5, 5, 160, 20)
        self.down.setGeometry(5, 28, 160, 20)
  
        # 设置字体（使用 Segoe UI 系统字体）
        font = QFont("Segoe UI", 15, QFont.Bold)
        self.up.setFont(font)
        self.down.setFont(font)
  
        # 设置文字颜色和样式
        self.up.setStyleSheet("color:#e1b16a")
        self.down.setStyleSheet("color:#ce4a57")
        self.setStyleSheet("QLabel{font-size:15px;font-weight:bold}")
        # 设置窗口默认位置在屏幕右下角
        self.set_window_position()
  
        # 显示窗口
        self.show()
  
        # 拖动灵敏度设置：增加拖动区域的高度，使其更大
        self.drag_sensitivity = 35  # 扩大拖动区域，默认为35
  
    def set_window_position(self):
        """将窗口设置到屏幕右下角"""
        screen_geometry = QDesktopWidget().availableGeometry()
        screen_width = screen_geometry.width()
        screen_height = screen_geometry.height()
  
        window_width = self.width()
        window_height = self.height()
        x_position = screen_width - window_width - 10  # 右边距 10
        y_position = screen_height - window_height - 10  # 下边距 10
  
        self.move(x_position, y_position)
  
    def update_speed(self, speed_data):
        """更新上传和下载速度显示"""
        upload_speed, download_speed = speed_data
  
        # 上传速度显示
        upload_text = f"上传 {upload_speed:.2f} {'Kb/s' if upload_speed < 1024 else 'Mb/s'}"
        self.up.setText(upload_text)
  
        # 下载速度显示
        download_text = f"下载 {download_speed:.2f} {'Kb/s' if download_speed < 1024 else 'Mb/s'}"
        self.down.setText(download_text)
  
    def mousePressEvent(self, event):
        """鼠标按下事件，开始拖动窗口"""
        if event.button() == Qt.LeftButton and event.pos().y() < self.drag_sensitivity:
            self.m_flag = True
            self.m_Position = event.globalPos() - self.pos()
            event.accept()
            self.setCursor(QCursor(Qt.OpenHandCursor))
  
    def mouseMoveEvent(self, event):
        """鼠标移动事件，拖动窗口"""
        if Qt.LeftButton and self.m_flag:
            self.move(event.globalPos() - self.m_Position)
            event.accept()
  
    def mouseReleaseEvent(self, event):
        """鼠标释放事件，停止拖动窗口"""
        self.m_flag = False
        self.setCursor(QCursor(Qt.ArrowCursor))
  
  
class MyWorker(QThread):
    timeout = pyqtSignal(list)
  
    def __init__(self):
        super(MyWorker, self).__init__()
        self.previous_upload = 0
        self.previous_download = 0
  
    def run(self):
        """周期性获取网络流量数据并发射信号"""
        while True:
            try:
                net_io = psutil.net_io_counters()
                upload_speed = (net_io.bytes_sent - self.previous_upload) / 1024.0  # Kb/s
                download_speed = (net_io.bytes_recv - self.previous_download) / 1024.0  # Kb/s
  
                # 更新上次的发送和接收字节数
                self.previous_upload = net_io.bytes_sent
                self.previous_download = net_io.bytes_recv
  
                # 发射信号更新UI
                self.timeout.emit([upload_speed, download_speed])
            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess) as e:
                print(f"Error: {e}")
            self.sleep(1)  # 每秒更新一次
  
  
if __name__ == '__main__':
    app = QApplication(sys.argv)
    speed_monitor = Speed()
    sys.exit(app.exec_())