def __init__(self, parent=None):
super(SearchEventApp, self).__init__(parent)
self.setupUi(self)
self.searchButton.clicked.connect(self.search_events)
self.progressBar.setRange(0, 1)
self.progressBar.setStyleSheet(
read_file("config/progres_bar_stylesheet.css"))
self.paint_cell_calendar_old = self.calendarWidget.paintCell
self.calendarWidget.paintCell = self.paint_calendar_cell
self.searchInput.returnPressed.connect(self.search_events)
self.worker = Worker()
self.connect(self.worker, QtCore.SIGNAL("worker_finished()"),
self.search_done)
# change calendar when worker finishes
self.calendarWidget.connect(self.worker,
QtCore.SIGNAL("worker_finished()"),
self.calendarWidget.updateCells)
self.connect(self.worker, QtCore.SIGNAL("worker_error()"),
self.search_error)
self.events = []
def __init__(self, options, parent=None):
self.title = "Export ... "
self.options = options
self.exportFunction = None
if self.options["fileType"] == "STATA":
self.steps = ["Prepare Data", "Writing", "Saving"]
self.exportFunction = exportStata
elif self.options["fileType"] == "EXCEL":
self.steps = ["Prepare Data", "Sorting", "Writing", "Saving"]
self.exportFunction = exportExcel
Worker.__init__(self, parent)
def __init__(self, parent = None):
Worker.__init__(self, parent)
class SearchEventApp(QtGui.QMainWindow, searchUI.Ui_MainWindow):
def __init__(self, parent=None):
super(SearchEventApp, self).__init__(parent)
self.setupUi(self)
self.searchButton.clicked.connect(self.search_events)
self.progressBar.setRange(0, 1)
self.progressBar.setStyleSheet(
read_file("config/progres_bar_stylesheet.css"))
self.paint_cell_calendar_old = self.calendarWidget.paintCell
self.calendarWidget.paintCell = self.paint_calendar_cell
self.searchInput.returnPressed.connect(self.search_events)
self.worker = Worker()
self.connect(self.worker, QtCore.SIGNAL("worker_finished()"),
self.search_done)
# change calendar when worker finishes
self.calendarWidget.connect(self.worker,
QtCore.SIGNAL("worker_finished()"),
self.calendarWidget.updateCells)
self.connect(self.worker, QtCore.SIGNAL("worker_error()"),
self.search_error)
self.events = []
def search_error(self):
self.searchInput.setEnabled(True)
self.searchButton.setEnabled(True)
self.progressBar.setRange(0, 1)
t = self.worker.exception
msg = "Cannot perform this task."
if isinstance(t, IOError):
msg = "check your connection. Network is unreachable."
elif isinstance(t, logic.EngineQueryError):
msg = "cannot retrieve results. %s" % str(t)
mbox = QtGui.QMessageBox(QtGui.QMessageBox.Critical,
"Application error", msg)
mbox.exec_()
def paint_calendar_cell(self, painter, rect, date):
# TODO: handle drawing date
# self.paint_cell_calendar_old(painter, rect, date)
# self.calendarWidget.par.paintCell(self, painter, rect, date)
painter.drawText(rect.topLeft() + QtCore.QPoint(20, 20),
str(date.day()))
text_option = QtGui.QTextOption()
text_option.setWrapMode(QtGui.QTextOption.WordWrap)
pydate = date.toPyDate()
devent = []
if self.events:
for event in self.events:
if event.startDate == pydate:
devent.append(event)
elif event.endDate:
if event.startDate <= pydate <= event.endDate:
devent.append(event)
text = ""
if 1 < len(devent) < 5:
# TODO: kilka wydarzen
text = "%d wydarzenia" % len(devent)
elif len(devent) >= 5:
text = "%d wydarzen" % len(devent)
elif len(devent) == 1:
text = devent[0].title
color = QtGui.QColor(self.calendarWidget.palette().color(
QtGui.QPalette.Highlight))
if devent:
painter.fillRect(rect, color)
painter.drawText(rect.topLeft() + QtCore.QPoint(20, 20),
str(date.day()))
w = rect.width()
h = rect.height()
new_rect = QtCore.QRectF(rect)
new_rect.setTopLeft(new_rect.topLeft() +
QtCore.QPointF(0.1 * w, 0.5 * h))
painter.drawText(new_rect, text, text_option)
def search_done(self):
# self.disconnect(self.worker, QtCore.SIGNAL('terminated()'), self.search_done)
self.searchInput.setEnabled(True)
self.searchButton.setEnabled(True)
self.progressBar.setRange(0, 1)
events = self.worker.events
self.events = events
if events:
self.listView.clear()
i = 0
for event in events:
# setup event item look
item = QEventItemWidget(self.listView)
item.set_text(event.title)
if event.endDate:
item.set_date(
str(event.startDate) + " -- " + str(event.endDate))
else:
item.set_date(event.startDate)
# some code that is necessary evil
event_list_widget_item = QtGui.QListWidgetItem(self.listView)
event_list_widget_item.setSizeHint(item.sizeHint())
self.listView.addItem(event_list_widget_item)
self.listView.setItemWidget(event_list_widget_item, item)
def search_events(self):
self.searchInput.setEnabled(False)
self.searchButton.setEnabled(False)
self.worker.set_query(self.searchInput.text(), 10)
self.worker.start()
self.progressBar.setRange(0, 0)