#!/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 系統(tǒng)字體）
        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()

        # 拖動靈敏度設置：增加拖動區(qū)域的高度，使其更大
        self.drag_sensitivity = 35  # 擴大拖動區(qū)域，默認為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):
        """周期性獲取網(wǎng)絡流量數(shù)據(jù)并發(fā)射信號"""
        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

                # 更新上次的發(fā)送和接收字節(jié)數(shù)
                self.previous_upload = net_io.bytes_sent
                self.previous_download = net_io.bytes_recv

                # 發(fā)射信號更新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_())