PySide6 実践講座 #5 トレイ常駐とリマインダー — 閉じても死なないアプリ

# src/daily/ui/tray.py
from PySide6.QtGui import QAction, QIcon
from PySide6.QtWidgets import QApplication, QMenu, QSystemTrayIcon


def create_tray(window) -> QSystemTrayIcon:
    tray = QSystemTrayIcon(QIcon(":/icons/daily.png"), parent=window)

    menu = QMenu()
    show_action = QAction("開く", menu)
    show_action.triggered.connect(window.show_and_raise)
    quit_action = QAction("終了", menu)
    quit_action.triggered.connect(QApplication.instance().quit)
    menu.addAction(show_action)
    menu.addSeparator()
    menu.addAction(quit_action)

    tray.setContextMenu(menu)
    tray.activated.connect(
        lambda reason: window.show_and_raise()
        if reason == QSystemTrayIcon.ActivationReason.Trigger
        else None
    )
    tray.show()
    return tray

# MainWindow に追加
from PySide6.QtGui import QCloseEvent


class MainWindow(QMainWindow):
    def closeEvent(self, event: QCloseEvent) -> None:
        if self._quit_requested:          # トレイメニューの「終了」でだけ本当に終了
            event.accept()
            return
        event.ignore()                    # 終了を止めて
        self.hide()                       # 非表示に置き換え

# src/daily/ui/reminder.py
from datetime import date, datetime

from PySide6.QtCore import QObject, QTimer
from PySide6.QtWidgets import QSystemTrayIcon


class Reminder(QObject):
    def __init__(self, repo, tray: QSystemTrayIcon, hour: int = 21) -> None:
        super().__init__()
        self._repo = repo
        self._tray = tray
        self._hour = hour
        self._fired_on: str | None = None    # 今日すでに鳴らしたか

        self._timer = QTimer(self)
        self._timer.setInterval(60 * 1000)   # 1 分
        self._timer.timeout.connect(self._tick)
        self._timer.start()

    def _tick(self) -> None:
        now = datetime.now()
        today = date.today().isoformat()
        if now.hour < self._hour or self._fired_on == today:
            return
        undone = [
            h.name for h in self._repo.active_habits()
            if today not in self._repo.checked_days(h.id)
        ]
        if undone:
            self._tray.showMessage(
                "Daily", f"まだの習慣: {', '.join(undone)}",
                QSystemTrayIcon.MessageIcon.Information,
            )
        self._fired_on = today

# src/daily/single_instance.py
from PySide6.QtNetwork import QLocalServer, QLocalSocket

_NAME = "daily-app-instance"


def acquire_or_notify() -> QLocalServer | None:
    """最初のインスタンスならサーバーを返し、すでに起動中なら合図だけ送って None。"""
    probe = QLocalSocket()
    probe.connectToServer(_NAME)
    if probe.waitForConnected(200):
        probe.write(b"show")             # 既存インスタンスに「ウィンドウを見せて」
        probe.waitForBytesWritten(200)
        return None

    server = QLocalServer()
    QLocalServer.removeServer(_NAME)     # 異常終了の残骸を掃除
    server.listen(_NAME)
    return server