-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
executable file
·120 lines (98 loc) · 4.23 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#!/usr/bin/env python3
import sys
import datetime
import threading
import time
import os.path
from PyQt4.QtGui import QApplication, QMainWindow, QSystemTrayIcon, QIcon
from PyQt4 import QtCore
from sessiontester.ui.main_window import Ui_MainAppWindow
from sessiontester.settings import SettingsWindow
from sessiontester.config_utils import SessionConfig
import sessiontester.export_session as export_session
class SessionTesterWindow(QMainWindow, Ui_MainAppWindow):
progress_bar_change = QtCore.pyqtSignal()
session_end = QtCore.pyqtSignal()
def __init__(self):
QMainWindow.__init__(self)
self.isStarted = False
self.isPaused = True
self.isRIP = False
self.current_time = 0
self.tray_icon = QSystemTrayIcon(QIcon("sessiontester/resources/trayicon.xpm"))
self.tray_icon.activated.connect(self.maximize_window)
self.tray_icon.show()
self.session_end.connect(self.session_end_message)
self.setupUi(self)
self.sessionStartButton.clicked.connect(self.start_session)
self.newSessionButton.clicked.connect(self.new_session)
self.exportSessionButton.clicked.connect(self.export_session)
self.settingsButton.clicked.connect(self.open_settings)
self.progressBar.setTextVisible(True)
self.progressBar.setFormat("Session progress")
self.session_config = SessionConfig(os.path.join(os.path.expanduser('~'),'sessiontester','config.ini'))
def start_session(self):
self.session_time = self.__count_session_time(
self.sessionTimeField.time().toString("hh:mm:ss"))
if (self.isPaused is True) & (self.session_time > 0):
self.isPaused = False
self.sessionStartButton.setText("Pause session")
self.session_time = self.__count_session_time(
self.sessionTimeField.time().toString("hh:mm:ss"))
self.progressBar.setMaximum(self.session_time)
if self.isRIP is True:
self.progressBar.setValue(0)
self.current_time = 0
self.isRIP = False
self.progress_bar_change.connect(self.set_progress_bar)
self.t = threading.Thread(target=self.set_progress)
self.t.daemon = True
self.t.start()
else:
self.sessionStartButton.setText("Start session")
self.isPaused = True
def new_session(self):
self.progressBar.setValue(0)
self.progressBar.setFormat("Session progress")
self.sessionStartButton.setText("Start session")
self.current_time = 0
self.isPaused = True
def export_session(self):
export_session.export_session_to_html(self.textEdit.toPlainText(), self.session_config)
def set_progress_bar(self):
self.progressBar.setValue(self.current_time)
self.progressBar.setFormat("Time left:" + str(
datetime.timedelta(seconds=self.time_to_go)))
def session_end_message(self):
self.tray_icon.showMessage("Session is over!", "Session time is over")
self.sessionStartButton.setText("Start session")
self.progressBar.setFormat("Session is over!")
self.isPaused = True
self.isRIP = True
def set_progress(self):
while (self.current_time < self.session_time):
if self.isPaused is False:
time.sleep(1)
else:
break
if self.isPaused is False:
self.current_time = self.current_time + 1
self.progress_bar_change.emit()
self.time_to_go = self.session_time - self.current_time
if self.current_time == self.session_time:
self.session_end.emit()
def __count_session_time(self, time_string):
self.result_date = time_string.split(":")
return (int(self.result_date[0])*3600) + (int(self.result_date[1])*60) + int(self.result_date[2])
def maximize_window(self):
if self.isHidden() is False:
self.hide()
elif self.isHidden() is True:
self.show()
def open_settings(self):
self.settings = SettingsWindow(self)
self.settings.show()
app = QApplication(sys.argv)
window = SessionTesterWindow()
window.show()
sys.exit(app.exec_())