106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import signal
|
|
from functools import partial
|
|
|
|
from PySide2 import QtCore, QtGui, QtWidgets
|
|
|
|
from data import Tasks, Log
|
|
from task_edit import TaskEdit
|
|
from report import Report
|
|
|
|
|
|
class App:
|
|
def __init__(self):
|
|
self.app = QtWidgets.QApplication(sys.argv)
|
|
|
|
self.tasks = Tasks()
|
|
self.log = Log()
|
|
self.active_task = self.log.last_log() or "Nothing"
|
|
|
|
icon = QtGui.QIcon.fromTheme("appointment-new")
|
|
|
|
self.menu = QtWidgets.QMenu()
|
|
|
|
self.tray = QtWidgets.QSystemTrayIcon()
|
|
self.tray.setIcon(icon)
|
|
self.tray.setContextMenu(self.menu)
|
|
self.tray.show()
|
|
self.tray.setToolTip("fimefracking")
|
|
self.update_tray_menu()
|
|
|
|
self.taskEdit = TaskEdit(None)
|
|
self.taskEdit.accepted.connect(self.tasks_edited)
|
|
|
|
self.reportDialog = Report(None)
|
|
self.reportDialog.accepted.connect(self.report_done)
|
|
|
|
@QtCore.Slot()
|
|
def tasks_edited(self):
|
|
self.tasks.tasks = self.taskEdit.tasks
|
|
self.update_tray_menu()
|
|
|
|
@QtCore.Slot()
|
|
def report_done(self):
|
|
self.active_task = self.log.last_log() or "Nothing"
|
|
self.update_tray_menu()
|
|
|
|
def change_task(self, task):
|
|
self.active_task = task
|
|
if task == "Nothing":
|
|
return
|
|
self.log.log(task)
|
|
self.update_tray_menu()
|
|
|
|
def update_tray_menu(self):
|
|
self.menu.clear()
|
|
tasks = list(self.tasks.tasks)
|
|
tasks.append("Nothing")
|
|
|
|
for t in tasks:
|
|
a = self.menu.addAction(t)
|
|
a.triggered.connect(partial(self.change_task, t))
|
|
if t == self.active_task:
|
|
a.setIcon(QtGui.QIcon.fromTheme("go-next"))
|
|
|
|
self.menu.addSeparator()
|
|
|
|
new_action = self.menu.addAction("Edit tasks")
|
|
new_action.triggered.connect(self.edit_tasks)
|
|
|
|
report_action = self.menu.addAction("Report")
|
|
report_action.triggered.connect(self.report)
|
|
|
|
self.menu.addSeparator()
|
|
|
|
exit_action = self.menu.addAction("Close")
|
|
exit_action.triggered.connect(self.app.quit)
|
|
|
|
def sigterm_handler(self, _signo, _frame):
|
|
self.app.quit()
|
|
|
|
def run(self):
|
|
timer = QtCore.QTimer()
|
|
# interrupt event loop regularly for signal handling
|
|
timer.timeout.connect(lambda: None)
|
|
timer.start(500)
|
|
signal.signal(signal.SIGTERM, self.sigterm_handler)
|
|
signal.signal(signal.SIGINT, self.sigterm_handler)
|
|
self.app.exec_()
|
|
|
|
@QtCore.Slot()
|
|
def report(self):
|
|
self.reportDialog.set_data(self.log.report())
|
|
self.reportDialog.show()
|
|
|
|
@QtCore.Slot()
|
|
def edit_tasks(self):
|
|
self.taskEdit.tasks = self.tasks.tasks
|
|
self.taskEdit.show()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = App()
|
|
app.run()
|