def __init__(self, backend, parent=None):
super().__init__(parent)
self.backend = backend
self.ui = Ui_mainwindow()
self.ui.setupUi(self)
self.setWindowTitle('qTimer v%s' % VERSION)
session = self.backend.session
# Setup initial projects
self.backend.syncConditionally()
for project in session.query(Project):
parent = QTreeWidgetItem([project.name])
parent.project = project
self.ui.projects.addTopLevelItem(parent)
for ticket in project.tickets:
child = QTreeWidgetItem([ticket.name])
child.ticket = ticket
parent.addChild(child)
for timer in session.query(Timer):
item = QTreeWidgetItem([
timer.status,
timer.name,
format_time(timer.start),
str(self.backend.roundTime(timer.duration)),
str(timer.posted)
])
item.timer = timer
self.ui.timers.addTopLevelItem(item)
self.ui.date_from.setDate(QDate.currentDate())
self.ui.date_to.setDate(QDate.currentDate())
self.actionMenu = QMenu()
self.actionMenu.triggered.connect(self.onActionClicked)
self.actionMenu.addAction('Start')
self.actionMenu.addAction('Stop')
self.actionMenu.addAction('Post')
self.ui.actions.setMenu(self.actionMenu)
self.readSettings()
self.dateFromChanged = lambda date: self.onDateChanged(self.ui.date_from_label)
self.dateToChanged = lambda date: self.onDateChanged(self.ui.date_to_label)
self.ui.projects.itemClicked.connect(self.onFilterClicked)
self.ui.date_from.dateChanged.connect(self.dateFromChanged)
self.ui.date_to.dateChanged.connect(self.dateToChanged)
self.durationTimer = QTimer(self)
self.durationTimer.timeout.connect(self.onRefreshDurations)
self.durationTimer.start(int(self.backend.config.timers.rounding) * 1000)
class QTimerMainWindow(QMainWindow):
def __init__(self, backend, parent=None):
super().__init__(parent)
self.backend = backend
self.ui = Ui_mainwindow()
self.ui.setupUi(self)
self.setWindowTitle('qTimer v%s' % VERSION)
session = self.backend.session
# Setup initial projects
self.backend.syncConditionally()
for project in session.query(Project):
parent = QTreeWidgetItem([project.name])
parent.project = project
self.ui.projects.addTopLevelItem(parent)
for ticket in project.tickets:
child = QTreeWidgetItem([ticket.name])
child.ticket = ticket
parent.addChild(child)
for timer in session.query(Timer):
item = QTreeWidgetItem([
timer.status,
timer.name,
format_time(timer.start),
str(self.backend.roundTime(timer.duration)),
str(timer.posted)
])
item.timer = timer
self.ui.timers.addTopLevelItem(item)
self.ui.date_from.setDate(QDate.currentDate())
self.ui.date_to.setDate(QDate.currentDate())
self.actionMenu = QMenu()
self.actionMenu.triggered.connect(self.onActionClicked)
self.actionMenu.addAction('Start')
self.actionMenu.addAction('Stop')
self.actionMenu.addAction('Post')
self.ui.actions.setMenu(self.actionMenu)
self.readSettings()
self.dateFromChanged = lambda date: self.onDateChanged(self.ui.date_from_label)
self.dateToChanged = lambda date: self.onDateChanged(self.ui.date_to_label)
self.ui.projects.itemClicked.connect(self.onFilterClicked)
self.ui.date_from.dateChanged.connect(self.dateFromChanged)
self.ui.date_to.dateChanged.connect(self.dateToChanged)
self.durationTimer = QTimer(self)
self.durationTimer.timeout.connect(self.onRefreshDurations)
self.durationTimer.start(int(self.backend.config.timers.rounding) * 1000)
def onDateChanged(self, dateEdit):
if dateEdit.checkState() == Qt.Unchecked:
dateEdit.setCheckState(Qt.Checked)
def writeSettings(self):
settingsPath = path.join(DATA_DIR, 'qtimer.gui.ini')
settings = QSettings(settingsPath, QSettings.IniFormat)
settings.beginGroup('MainWindow')
settings.setValue('size', self.size())
settings.setValue('pos', self.pos())
settings.setValue('timersState', self.ui.timers.header().saveState())
settings.setValue('splitterState', self.ui.splitter.saveState())
settings.endGroup()
def readSettings(self):
settingsPath = path.join(DATA_DIR, 'qtimer.gui.ini')
settings = QSettings(settingsPath, QSettings.IniFormat)
settings.beginGroup('MainWindow')
self.resize(settings.value('size', QSize(420, 460)))
self.move(settings.value('pos', QPoint(200, 200)))
splitterState = settings.value('splitterState', None)
if splitterState:
self.ui.splitter.restoreState(splitterState)
listColState = settings.value('timersState', None)
if listColState:
self.ui.timers.header().restoreState(listColState)
settings.endGroup()
def createErrorText(self, text):
msgBox = QMessageBox()
msgBox.setWindowTitle("I'm sorry, Dave. I'm afraid I can't do that.")
msgBox.setText(text)
msgBox.exec_()
return
setattr(self.ui, text.replace(' .', '_').lower(), msgBox)
def onRefreshDurations(self):
for i in range(self.ui.timers.topLevelItemCount()):
# GET CHILD & REFRESH
item = self.ui.timers.topLevelItem(i)
item.setText(3, str(self.backend.roundTime(item.timer.duration)))
def onFilterClicked(self, item, column):
self.ui.timers.clear()
query = self.backend.session.query(Timer)
if hasattr(item, 'ticket'):
query = query.filter(Timer.ticket_id == item.ticket.id)
elif hasattr(item, 'project'):
query = query.join(Ticket).filter(Ticket.project_id == item.project.id)
for timer in query:
item = QTreeWidgetItem([
timer.status,
timer.name,
format_time(timer.start),
str(self.backend.roundTime(timer.duration)),
str(timer.posted)
])
item.timer = timer
self.ui.timers.addTopLevelItem(item)
def onActionClicked(self, action):
print('onActionClicked:', repr(action.text()))
items = self.ui.timers.selectedItems()
{
'start': self.onStartTimers,
'stop': self.onStopTimers,
'post': self.onPostTimers
}.get(action.text().lower(), self.noAction)(items)
def onStartTimers(self, items):
if items:
# Start any paused timers
with qtimer.autocommit(self.backend.session) as sql:
for item in items:
status = item.timer.status
if not status == STATUS_IDLE:
continue
session = Session(start=self.backend.roundTime(datetime.utcnow()), timer_id=item.timer.id)
sql.add(session)
else:
# Create timer
text, ok = QInputDialog.getText(self, 'Enter Timer Name', 'Enter new timer name:')
if not ok:
return
item = self.ui.projects.selectedItems()
tid = None
if hasattr(item, 'ticket'):
tid = item.ticket.id
session = Session(start=self.backend.roundTime(datetime.utcnow()))
timer = Timer(name=text, ticket_id=tid, sessions=[session])
self.backend.session.add(timer)
self.onFilterClicked(item, None)
def onStopTimers(self, items):
if not items:
self.createErrorText('No timers selected.')
return
ids = []
for item in items:
ids.append(item.timer.id)
values = {
Session.end: self.backend.roundTime(datetime.utcnow())
}
with qtimer.autocommit(self.backend.session) as session:
session.query(Session).filter(Session.timer_id.in_(ids))\
.filter(Session.end == None).update(values, 'fetch')
self.onFilterClicked(item, None)
def onPostTimers(self, items):
pass
def noAction(self, items):
self.createErrorText('There is no implementation for this action')
raise RuntimeError('There is no implementation for this action')
def closeEvent(self, event):
self.writeSettings()
