+ self._app = app
+ self._appIconPath = appIconPath
+ self._recent = []
+ self._hiddenCategories = set()
+ self._hiddenUnits = {}
+ self._clipboard = QtGui.QApplication.clipboard()
+
+ self._jumpWindow = None
+ self._recentWindow = None
+ self._catWindow = None
+
+ self._jumpAction = QtGui.QAction(None)
+ self._jumpAction.setText("Quick Jump")
+ self._jumpAction.setStatusTip("Search for a unit and jump straight to it")
+ self._jumpAction.setToolTip("Search for a unit and jump straight to it")
+ self._jumpAction.setShortcut(QtGui.QKeySequence("CTRL+j"))
+ self._jumpAction.triggered.connect(self._on_jump_start)
+
+ self._recentAction = QtGui.QAction(None)
+ self._recentAction.setText("Recent Units")
+ self._recentAction.setStatusTip("View the recent units")
+ self._recentAction.setToolTip("View the recent units")
+ self._recentAction.setShortcut(QtGui.QKeySequence("CTRL+r"))
+ self._recentAction.triggered.connect(self._on_recent_start)
+
+ self._fullscreenAction = QtGui.QAction(None)
+ self._fullscreenAction.setText("Fullscreen")
+ self._fullscreenAction.setCheckable(True)
+ self._fullscreenAction.setShortcut(QtGui.QKeySequence("CTRL+Enter"))
+ self._fullscreenAction.toggled.connect(self._on_toggle_fullscreen)
+
+ self._showFavoritesAction = QtGui.QAction(None)
+ self._showFavoritesAction.setCheckable(True)
+ self._showFavoritesAction.setText("Favorites Only")
+
+ self._logAction = QtGui.QAction(None)
+ self._logAction.setText("Log")
+ self._logAction.setShortcut(QtGui.QKeySequence("CTRL+l"))
+ self._logAction.triggered.connect(self._on_log)
+
+ self._quitAction = QtGui.QAction(None)
+ self._quitAction.setText("Quit")
+ self._quitAction.setShortcut(QtGui.QKeySequence("CTRL+q"))
+ self._quitAction.triggered.connect(self._on_quit)
+
+ self._app.lastWindowClosed.connect(self._on_app_quit)
+ self.request_category()
+ self.load_settings()
+
+ def request_category(self):
+ if self._catWindow is not None:
+ self._catWindow.close()
+ self._catWindow = None
+ self._catWindow = CategoryWindow(None, self)
+ self._catWindow.window.destroyed.connect(lambda obj = None: self._on_child_close("_catWindow", obj))
+ return self._catWindow
+
+ def search_units(self):
+ if self._jumpWindow is not None:
+ self._jumpWindow.close()
+ self._jumpWindow = None
+ self._jumpWindow = QuickJump(None, self)
+ self._jumpWindow.window.destroyed.connect(lambda obj = None: self._on_child_close("_jumpWindow", obj))
+ return self._jumpWindow
+
+ def show_recent(self):
+ if self._recentWindow is not None:
+ self._recentWindow.close()
+ self._recentWindow = None
+ self._recentWindow = Recent(None, self)
+ self._recentWindow.window.destroyed.connect(lambda obj = None: self._on_child_close("_recentWindow", obj))
+ return self._recentWindow
+
+ def add_recent(self, categoryName, unitName):
+ catUnit = categoryName, unitName
+ try:
+ self._recent.remove(catUnit)
+ except ValueError:
+ pass # ignore if its not already in the recent history
+ assert catUnit not in self._recent
+ self._recent.append(catUnit)
+
+ def get_recent_unit(self, categoryName):
+ for catName, unitName in reversed(self._recent):
+ if catName == categoryName:
+ return unitName
+ else:
+ return ""
+
+ def get_recent(self):
+ return reversed(self._recent)
+
+ @property
+ def hiddenCategories(self):
+ return self._hiddenCategories
+
+ def get_hidden_units(self, categoryName):
+ try:
+ return self._hiddenUnits[categoryName]
+ except KeyError:
+ self._hiddenUnits[categoryName] = set()
+ return self._hiddenUnits[categoryName]
+
+ def load_settings(self):
+ try:
+ with open(constants._user_settings_, "r") as settingsFile:
+ settings = simplejson.load(settingsFile)
+ except IOError, e:
+ _moduleLogger.info("No settings")
+ settings = {}
+ except ValueError:
+ _moduleLogger.info("Settings were corrupt")
+ settings = {}
+
+ self._fullscreenAction.setChecked(settings.get("isFullScreen", False))
+
+ recent = settings.get("recent", self._recent)
+ for category, unit in recent:
+ self.add_recent(category, unit)
+
+ self._hiddenCategories = set(settings.get("hiddenCategories", set()))
+ self._hiddenUnits = dict(
+ (catName, set(units))
+ for (catName, units) in settings.get("hiddenUnits", {}).iteritems()
+ )
+
+ self._showFavoritesAction.setChecked(settings.get("showFavorites", True))
+
+ if self._recent:
+ self._catWindow.select_category(self._recent[-1][0])
+
+ def save_settings(self):
+ settings = {
+ "isFullScreen": self._fullscreenAction.isChecked(),
+ "recent": self._recent,
+ "hiddenCategories": list(self._hiddenCategories),
+ "hiddenUnits": dict(
+ (catName, list(units))
+ for (catName, units) in self._hiddenUnits.iteritems()
+ ),
+ "showFavorites": self._showFavoritesAction.isChecked(),
+ }
+ with open(constants._user_settings_, "w") as settingsFile:
+ simplejson.dump(settings, settingsFile)
+
+ @property
+ def appIconPath(self):
+ return self._appIconPath
+
+ @property
+ def jumpAction(self):
+ return self._jumpAction
+
+ @property
+ def recentAction(self):
+ return self._recentAction
+
+ @property
+ def fullscreenAction(self):
+ return self._fullscreenAction
+
+ @property
+ def logAction(self):
+ return self._logAction
+
+ @property
+ def quitAction(self):
+ return self._quitAction
+
+ @property
+ def showFavoritesAction(self):
+ return self._showFavoritesAction
+
+ def _walk_children(self):
+ if self._catWindow is not None:
+ yield self._catWindow
+ if self._jumpWindow is not None:
+ yield self._jumpWindow
+ if self._recentWindow is not None:
+ yield self._recentWindow
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_app_quit(self, checked = False):
+ self.save_settings()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_child_close(self, name, obj = None):
+ if not hasattr(self, name):
+ _moduleLogger.info("Something weird going on when we don't have a %s" % name)
+ return
+ setattr(self, name, None)
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_toggle_fullscreen(self, checked = False):
+ for window in self._walk_children():
+ window.set_fullscreen(checked)
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_jump_start(self, checked = False):
+ self.search_units()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_recent_start(self, checked = False):
+ self.show_recent()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_log(self, checked = False):
+ with open(constants._user_logpath_, "r") as f:
+ logLines = f.xreadlines()
+ log = "".join(logLines)
+ self._clipboard.setText(log)
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_quit(self, checked = False):
+ for window in self._walk_children():
+ window.close()
+
+
+class QuickJump(object):
+
+ MINIMAL_ENTRY = 3
+
+ def __init__(self, parent, app):
+ self._app = app
+
+ self._searchLabel = QtGui.QLabel("Search:")
+ self._searchEntry = QtGui.QLineEdit("")
+ self._searchEntry.textEdited.connect(self._on_search_edited)
+
+ self._entryLayout = QtGui.QHBoxLayout()
+ self._entryLayout.addWidget(self._searchLabel)
+ self._entryLayout.addWidget(self._searchEntry)
+
+ self._resultsBox = QtGui.QTreeWidget()
+ self._resultsBox.setHeaderLabels(["Categories", "Units"])
+ self._resultsBox.setHeaderHidden(True)
+ self._resultsBox.setAlternatingRowColors(True)
+ self._resultsBox.itemClicked.connect(self._on_result_clicked)
+
+ self._layout = QtGui.QVBoxLayout()
+ self._layout.addLayout(self._entryLayout)
+ self._layout.addWidget(self._resultsBox)
+
+ centralWidget = QtGui.QWidget()
+ centralWidget.setLayout(self._layout)
+
+ self._window = QtGui.QMainWindow(parent)
+ self._window.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
+ if parent is not None:
+ self._window.setWindowModality(QtCore.Qt.WindowModal)
+ self._window.setWindowTitle("%s - Quick Jump" % constants.__pretty_app_name__)
+ self._window.setWindowIcon(QtGui.QIcon(self._app.appIconPath))
+ self._window.setCentralWidget(centralWidget)
+
+ self._closeWindowAction = QtGui.QAction(None)
+ self._closeWindowAction.setText("Window")
+ self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
+ self._closeWindowAction.triggered.connect(self._on_close_window)
+
+ fileMenu = self._window.menuBar().addMenu("&Units")
+ fileMenu.addAction(self._closeWindowAction)
+ fileMenu.addAction(self._app.quitAction)
+
+ viewMenu = self._window.menuBar().addMenu("&View")
+ viewMenu.addAction(self._app.fullscreenAction)
+
+ self._window.addAction(self._app.logAction)
+
+ self.set_fullscreen(self._app.fullscreenAction.isChecked())
+ self._window.show()
+
+ @property
+ def window(self):
+ return self._window
+
+ def close(self):
+ self._window.close()
+
+ def set_fullscreen(self, isFullscreen):
+ if isFullscreen:
+ self._window.showFullScreen()
+ else:
+ self._window.showNormal()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_close_window(self, checked = True):
+ self.close()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_result_clicked(self, item, columnIndex):
+ categoryName = unicode(item.text(0))
+ unitName = unicode(item.text(1))
+ catWindow = self._app.request_category()
+ unitsWindow = catWindow.select_category(categoryName)
+ unitsWindow.select_unit(unitName)
+ self.close()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_search_edited(self, *args):
+ userInput = self._searchEntry.text()
+ if len(userInput) < self.MINIMAL_ENTRY:
+ return
+
+ self._resultsBox.clear()
+ lowerInput = str(userInput).lower()
+ for catIndex, category in enumerate(unit_data.UNIT_CATEGORIES):
+ units = unit_data.get_units(category)
+ for unitIndex, unit in enumerate(units):
+ loweredUnit = unit.lower()
+ if lowerInput in loweredUnit:
+ twi = QtGui.QTreeWidgetItem(self._resultsBox)
+ twi.setText(0, category)
+ twi.setText(1, unit)
+
+
+class Recent(object):
+
+ def __init__(self, parent, app):
+ self._app = app
+
+ self._resultsBox = QtGui.QTreeWidget()
+ self._resultsBox.setHeaderLabels(["Categories", "Units"])
+ self._resultsBox.setHeaderHidden(True)
+ self._resultsBox.setAlternatingRowColors(True)
+ self._resultsBox.itemClicked.connect(self._on_result_clicked)
+
+ self._layout = QtGui.QVBoxLayout()
+ self._layout.addWidget(self._resultsBox)
+
+ centralWidget = QtGui.QWidget()
+ centralWidget.setLayout(self._layout)
+
+ self._window = QtGui.QMainWindow(parent)
+ self._window.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
+ if parent is not None:
+ self._window.setWindowModality(QtCore.Qt.WindowModal)
+ self._window.setWindowTitle("%s - Recent" % constants.__pretty_app_name__)
+ self._window.setWindowIcon(QtGui.QIcon(self._app.appIconPath))
+ self._window.setCentralWidget(centralWidget)
+
+ for cat, unit in self._app.get_recent():
+ twi = QtGui.QTreeWidgetItem(self._resultsBox)
+ twi.setText(0, cat)
+ twi.setText(1, unit)
+
+ self._closeWindowAction = QtGui.QAction(None)
+ self._closeWindowAction.setText("Window")
+ self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
+ self._closeWindowAction.triggered.connect(self._on_close_window)
+
+ fileMenu = self._window.menuBar().addMenu("&Units")
+ fileMenu.addAction(self._closeWindowAction)
+ fileMenu.addAction(self._app.quitAction)
+
+ viewMenu = self._window.menuBar().addMenu("&View")
+ viewMenu.addAction(self._app.fullscreenAction)
+
+ self._window.addAction(self._app.logAction)
+
+ self.set_fullscreen(self._app.fullscreenAction.isChecked())
+ self._window.show()
+
+ @property
+ def window(self):
+ return self._window
+
+ def close(self):
+ self._window.close()
+
+ def set_fullscreen(self, isFullscreen):
+ if isFullscreen:
+ self._window.showFullScreen()
+ else:
+ self._window.showNormal()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_close_window(self, checked = True):
+ self.close()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_result_clicked(self, item, columnIndex):
+ categoryName = unicode(item.text(0))
+ unitName = unicode(item.text(1))
+ catWindow = self._app.request_category()
+ unitsWindow = catWindow.select_category(categoryName)
+ unitsWindow.select_unit(unitName)
+ self.close()
+
+
+class FavoriteCategoriesWindow(object):
+
+ def __init__(self, parent, app, source, hidden):
+ self._app = app
+ self._source = list(source)
+ self._hidden = hidden
+
+ self._categories = QtGui.QTreeWidget()
+ self._categories.setHeaderLabels(["Categories"])
+ self._categories.setHeaderHidden(True)
+ self._categories.setAlternatingRowColors(True)
+ self._categories.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
+ self._categories.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
+ self._childWidgets = []
+ for catName in self._source:
+ twi = QtGui.QTreeWidgetItem(self._categories)
+ twi.setText(0, catName)
+ self._childWidgets.append(twi)
+ if catName not in self._hidden:
+ self._categories.setItemSelected(twi, True)
+ self._selection = self._categories.selectionModel()
+ self._selection.selectionChanged.connect(self._on_selection_changed)
+
+ self._allButton = QtGui.QPushButton("All")
+ self._allButton.clicked.connect(self._on_select_all)
+ self._invertButton = QtGui.QPushButton("Invert")
+ self._invertButton.clicked.connect(self._on_invert_selection)
+ self._noneButton = QtGui.QPushButton("None")
+ self._noneButton.clicked.connect(self._on_select_none)
+
+ self._buttonLayout = QtGui.QHBoxLayout()
+ self._buttonLayout.addWidget(self._allButton)
+ self._buttonLayout.addWidget(self._invertButton)
+ self._buttonLayout.addWidget(self._noneButton)
+
+ self._layout = QtGui.QVBoxLayout()
+ self._layout.addWidget(self._categories)
+ self._layout.addLayout(self._buttonLayout)
+
+ centralWidget = QtGui.QWidget()
+ centralWidget.setLayout(self._layout)
+
+ self._window = QtGui.QMainWindow(parent)
+ self._window.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
+ if parent is not None:
+ self._window.setWindowModality(QtCore.Qt.WindowModal)
+ self._window.setWindowTitle("%s - Favorites" % constants.__pretty_app_name__)
+ self._window.setWindowIcon(QtGui.QIcon(self._app.appIconPath))
+ self._window.setCentralWidget(centralWidget)
+
+ self._closeWindowAction = QtGui.QAction(None)
+ self._closeWindowAction.setText("Window")
+ self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
+ self._closeWindowAction.triggered.connect(self._on_close_window)
+
+ fileMenu = self._window.menuBar().addMenu("&Units")
+ fileMenu.addAction(self._closeWindowAction)
+ fileMenu.addAction(self._app.quitAction)
+
+ viewMenu = self._window.menuBar().addMenu("&View")
+ viewMenu.addAction(self._app.fullscreenAction)
+
+ self._window.addAction(self._app.logAction)
+
+ self.set_fullscreen(self._app.fullscreenAction.isChecked())
+ self._window.show()
+
+ @property
+ def window(self):
+ return self._window
+
+ def close(self):
+ self._window.close()
+
+ def set_fullscreen(self, isFullscreen):
+ if isFullscreen:
+ self._window.showFullScreen()
+ else:
+ self._window.showNormal()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_select_all(self, checked = False):
+ for child in self._childWidgets:
+ self._categories.setItemSelected(child, True)
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_invert_selection(self, checked = False):
+ for child in self._childWidgets:
+ isSelected = self._categories.isItemSelected(child)
+ self._categories.setItemSelected(child, not isSelected)
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_select_none(self, checked = False):
+ for child in self._childWidgets:
+ self._categories.setItemSelected(child, False)