vcs.maemo.org Git - ejpi/blob - src/ejpi_qt.py

   1 #!/usr/bin/env python
   2 # -*- coding: UTF8 -*-
   3
   4 from __future__ import with_statement
   5
   6 import sys
   7 import os
   8 import simplejson
   9 import string
  10 import logging
  11
  12 from PyQt4 import QtCore
  13 from PyQt4 import QtGui
  14
  15 import constants
  16 from util import misc as misc_utils
  17
  18 from util import qui_utils
  19 from util import qwrappers
  20 from util import qtpie
  21 from util import qtpieboard
  22 import plugin_utils
  23 import history
  24 import qhistory
  25
  26
  27 _moduleLogger = logging.getLogger(__name__)
  28
  29
  30 class Calculator(qwrappers.ApplicationWrapper):
  31
  32         def __init__(self, app):
  33                 self._recent = []
  34                 self._hiddenCategories = set()
  35                 self._hiddenUnits = {}
  36                 qwrappers.ApplicationWrapper.__init__(self, app, constants)
  37
  38         def load_settings(self):
  39                 try:
  40                         with open(constants._user_settings_, "r") as settingsFile:
  41                                 settings = simplejson.load(settingsFile)
  42                 except IOError, e:
  43                         _moduleLogger.info("No settings")
  44                         settings = {}
  45                 except ValueError:
  46                         _moduleLogger.info("Settings were corrupt")
  47                         settings = {}
  48
  49                 isPortraitDefault = qui_utils.screen_orientation() == QtCore.Qt.Vertical
  50                 self._fullscreenAction.setChecked(settings.get("isFullScreen", False))
  51                 self._orientationAction.setChecked(settings.get("isPortrait", isPortraitDefault))
  52
  53         def save_settings(self):
  54                 settings = {
  55                         "isFullScreen": self._fullscreenAction.isChecked(),
  56                         "isPortrait": self._orientationAction.isChecked(),
  57                 }
  58                 with open(constants._user_settings_, "w") as settingsFile:
  59                         simplejson.dump(settings, settingsFile)
  60
  61         @property
  62         def dataPath(self):
  63                 return self._dataPath
  64
  65         def _new_main_window(self):
  66                 return MainWindow(None, self)
  67
  68         @misc_utils.log_exception(_moduleLogger)
  69         def _on_about(self, checked = True):
  70                 raise NotImplementedError("Booh")
  71
  72
  73 class QValueEntry(object):
  74
  75         def __init__(self):
  76                 self._widget = QtGui.QLineEdit("")
  77                 qui_utils.mark_numbers_preferred(self._widget)
  78
  79         @property
  80         def toplevel(self):
  81                 return self._widget
  82
  83         @property
  84         def entry(self):
  85                 return self._widget
  86
  87         def get_value(self):
  88                 value = str(self._widget.text()).strip()
  89                 if any(
  90                         0 < value.find(whitespace)
  91                         for whitespace in string.whitespace
  92                 ):
  93                         self.clear()
  94                         raise ValueError('Invalid input "%s"' % value)
  95                 return value
  96
  97         def set_value(self, value):
  98                 value = value.strip()
  99                 if any(
 100                         0 < value.find(whitespace)
 101                         for whitespace in string.whitespace
 102                 ):
 103                         raise ValueError('Invalid input "%s"' % value)
 104                 self._widget.setText(value)
 105
 106         def append(self, value):
 107                 value = value.strip()
 108                 if any(
 109                         0 < value.find(whitespace)
 110                         for whitespace in string.whitespace
 111                 ):
 112                         raise ValueError('Invalid input "%s"' % value)
 113                 self.set_value(self.get_value() + value)
 114
 115         def pop(self):
 116                 value = self.get_value()[0:-1]
 117                 self.set_value(value)
 118
 119         def clear(self):
 120                 self.set_value("")
 121
 122         value = property(get_value, set_value, clear)
 123
 124
 125 class MainWindow(qwrappers.WindowWrapper):
 126
 127         _plugin_search_paths = [
 128                 os.path.join(os.path.dirname(__file__), "plugins/"),
 129         ]
 130
 131         _user_history = "%s/history.stack" % constants._data_path_
 132
 133         def __init__(self, parent, app):
 134                 qwrappers.WindowWrapper.__init__(self, parent, app)
 135                 self._window.setWindowTitle("%s" % constants.__pretty_app_name__)
 136
 137                 self._historyView = qhistory.QCalcHistory(self._app.errorLog)
 138                 self._userEntry = QValueEntry()
 139                 self._userEntry.entry.returnPressed.connect(self._on_push)
 140                 self._userEntryLayout = QtGui.QHBoxLayout()
 141                 self._userEntryLayout.setContentsMargins(0, 0, 0, 0)
 142                 self._userEntryLayout.addWidget(self._userEntry.toplevel, 10)
 143
 144                 self._controlLayout = QtGui.QVBoxLayout()
 145                 self._controlLayout.setContentsMargins(0, 0, 0, 0)
 146                 self._controlLayout.addWidget(self._historyView.toplevel, 1000)
 147                 self._controlLayout.addLayout(self._userEntryLayout, 0)
 148
 149                 self._keyboardTabs = QtGui.QTabWidget()
 150
 151                 self._layout.addLayout(self._controlLayout)
 152                 self._layout.addWidget(self._keyboardTabs)
 153
 154                 self._copyItemAction = QtGui.QAction(None)
 155                 self._copyItemAction.setText("Copy")
 156                 self._copyItemAction.setShortcut(QtGui.QKeySequence("CTRL+c"))
 157                 self._copyItemAction.triggered.connect(self._on_copy)
 158
 159                 self._pasteItemAction = QtGui.QAction(None)
 160                 self._pasteItemAction.setText("Paste")
 161                 self._pasteItemAction.setShortcut(QtGui.QKeySequence("CTRL+v"))
 162                 self._pasteItemAction.triggered.connect(self._on_paste)
 163
 164                 self._closeWindowAction = QtGui.QAction(None)
 165                 self._closeWindowAction.setText("Close")
 166                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
 167                 self._closeWindowAction.triggered.connect(self._on_close_window)
 168
 169                 self._window.addAction(self._copyItemAction)
 170                 self._window.addAction(self._pasteItemAction)
 171
 172                 self._constantPlugins = plugin_utils.ConstantPluginManager()
 173                 self._constantPlugins.add_path(*self._plugin_search_paths)
 174                 for pluginName in ["Builtins", "Trigonometry", "Computer", "Alphabet"]:
 175                         try:
 176                                 pluginId = self._constantPlugins.lookup_plugin(pluginName)
 177                                 self._constantPlugins.enable_plugin(pluginId)
 178                         except:
 179                                 _moduleLogger.info("Failed to load plugin %s" % pluginName)
 180
 181                 self._operatorPlugins = plugin_utils.OperatorPluginManager()
 182                 self._operatorPlugins.add_path(*self._plugin_search_paths)
 183                 for pluginName in ["Builtins", "Trigonometry", "Computer", "Alphabet"]:
 184                         try:
 185                                 pluginId = self._operatorPlugins.lookup_plugin(pluginName)
 186                                 self._operatorPlugins.enable_plugin(pluginId)
 187                         except:
 188                                 _moduleLogger.info("Failed to load plugin %s" % pluginName)
 189
 190                 self._keyboardPlugins = plugin_utils.KeyboardPluginManager()
 191                 self._keyboardPlugins.add_path(*self._plugin_search_paths)
 192                 self._activeKeyboards = []
 193
 194                 self._history = history.RpnCalcHistory(
 195                         self._historyView,
 196                         self._userEntry, self._app.errorLog,
 197                         self._constantPlugins.constants, self._operatorPlugins.operators
 198                 )
 199                 self._load_history()
 200
 201                 # Basic keyboard stuff
 202                 self._handler = qtpieboard.KeyboardHandler(self._on_entry_direct)
 203                 self._handler.register_command_handler("push", self._on_push)
 204                 self._handler.register_command_handler("unpush", self._on_unpush)
 205                 self._handler.register_command_handler("backspace", self._on_entry_backspace)
 206                 self._handler.register_command_handler("clear", self._on_entry_clear)
 207
 208                 # Main keyboard
 209                 entryKeyboardId = self._keyboardPlugins.lookup_plugin("Entry")
 210                 self._keyboardPlugins.enable_plugin(entryKeyboardId)
 211                 entryPlugin = self._keyboardPlugins.keyboards["Entry"].construct_keyboard()
 212                 entryKeyboard = entryPlugin.setup(self._history, self._handler)
 213                 self._userEntryLayout.addWidget(entryKeyboard.toplevel)
 214
 215                 # Plugins
 216                 self.enable_plugin(self._keyboardPlugins.lookup_plugin("Builtins"))
 217                 self.enable_plugin(self._keyboardPlugins.lookup_plugin("Trigonometry"))
 218                 self.enable_plugin(self._keyboardPlugins.lookup_plugin("Computer"))
 219                 self.enable_plugin(self._keyboardPlugins.lookup_plugin("Alphabet"))
 220
 221                 self._scrollTimer = QtCore.QTimer()
 222                 self._scrollTimer.setInterval(0)
 223                 self._scrollTimer.setSingleShot(True)
 224                 self._scrollTimer.timeout.connect(self._on_delayed_scroll_to_bottom)
 225                 self._scrollTimer.start()
 226
 227                 self.set_fullscreen(self._app.fullscreenAction.isChecked())
 228                 self.set_orientation(self._app.orientationAction.isChecked())
 229
 230         def walk_children(self):
 231                 return ()
 232
 233         def set_orientation(self, isPortrait):
 234                 qwrappers.WindowWrapper.set_orientation(self, isPortrait)
 235                 if isPortrait:
 236                         defaultLayoutOrientation = QtGui.QBoxLayout.TopToBottom
 237                         #tabPosition = QtGui.QTabWidget.South
 238                         tabPosition = QtGui.QTabWidget.West
 239                 else:
 240                         defaultLayoutOrientation = QtGui.QBoxLayout.LeftToRight
 241                         tabPosition = QtGui.QTabWidget.North
 242                 self._layout.setDirection(defaultLayoutOrientation)
 243                 self._keyboardTabs.setTabPosition(tabPosition)
 244
 245         def enable_plugin(self, pluginId):
 246                 self._keyboardPlugins.enable_plugin(pluginId)
 247                 pluginData = self._keyboardPlugins.plugin_info(pluginId)
 248                 pluginName = pluginData[0]
 249                 plugin = self._keyboardPlugins.keyboards[pluginName].construct_keyboard()
 250                 relIcon = self._keyboardPlugins.keyboards[pluginName].icon
 251                 for iconPath in self._keyboardPlugins.keyboards[pluginName].iconPaths:
 252                         absIconPath = os.path.join(iconPath, relIcon)
 253                         if os.path.exists(absIconPath):
 254                                 icon = QtGui.QIcon(absIconPath)
 255                                 break
 256                 else:
 257                         icon = None
 258                 pluginKeyboard = plugin.setup(self._history, self._handler)
 259
 260                 self._activeKeyboards.append({
 261                         "pluginName": pluginName,
 262                         "plugin": plugin,
 263                         "pluginKeyboard": pluginKeyboard,
 264                 })
 265                 if icon is None:
 266                         self._keyboardTabs.addTab(pluginKeyboard.toplevel, pluginName)
 267                 else:
 268                         self._keyboardTabs.addTab(pluginKeyboard.toplevel, icon, "")
 269
 270         def close(self):
 271                 qwrappers.WindowWrapper.close(self)
 272                 self._save_history()
 273
 274         def _load_history(self):
 275                 serialized = []
 276                 try:
 277                         with open(self._user_history, "rU") as f:
 278                                 serialized = (
 279                                         (part.strip() for part in line.split(" "))
 280                                         for line in f.readlines()
 281                                 )
 282                 except IOError, e:
 283                         if e.errno != 2:
 284                                 raise
 285                 self._history.deserialize_stack(serialized)
 286
 287         def _save_history(self):
 288                 serialized = self._history.serialize_stack()
 289                 with open(self._user_history, "w") as f:
 290                         for lineData in serialized:
 291                                 line = " ".join(data for data in lineData)
 292                                 f.write("%s\n" % line)
 293
 294         @misc_utils.log_exception(_moduleLogger)
 295         def _on_delayed_scroll_to_bottom(self):
 296                 with qui_utils.notify_error(self._app.errorLog):
 297                         self._historyView.scroll_to_bottom()
 298
 299         @misc_utils.log_exception(_moduleLogger)
 300         def _on_child_close(self, something = None):
 301                 with qui_utils.notify_error(self._app.errorLog):
 302                         self._child = None
 303
 304         @misc_utils.log_exception(_moduleLogger)
 305         def _on_copy(self, *args):
 306                 with qui_utils.notify_error(self._app.errorLog):
 307                         eqNode = self._historyView.peek()
 308                         resultNode = eqNode.simplify()
 309                         self._app._clipboard.setText(str(resultNode))
 310
 311         @misc_utils.log_exception(_moduleLogger)
 312         def _on_paste(self, *args):
 313                 with qui_utils.notify_error(self._app.errorLog):
 314                         result = str(self._app._clipboard.text())
 315                         self._userEntry.append(result)
 316
 317         @misc_utils.log_exception(_moduleLogger)
 318         def _on_entry_direct(self, keys, modifiers):
 319                 with qui_utils.notify_error(self._app.errorLog):
 320                         if "shift" in modifiers:
 321                                 keys = keys.upper()
 322                         self._userEntry.append(keys)
 323
 324         @misc_utils.log_exception(_moduleLogger)
 325         def _on_push(self, *args):
 326                 with qui_utils.notify_error(self._app.errorLog):
 327                         self._history.push_entry()
 328
 329         @misc_utils.log_exception(_moduleLogger)
 330         def _on_unpush(self, *args):
 331                 with qui_utils.notify_error(self._app.errorLog):
 332                         self._historyView.unpush()
 333
 334         @misc_utils.log_exception(_moduleLogger)
 335         def _on_entry_backspace(self, *args):
 336                 with qui_utils.notify_error(self._app.errorLog):
 337                         self._userEntry.pop()
 338
 339         @misc_utils.log_exception(_moduleLogger)
 340         def _on_entry_clear(self, *args):
 341                 with qui_utils.notify_error(self._app.errorLog):
 342                         self._userEntry.clear()
 343
 344         @misc_utils.log_exception(_moduleLogger)
 345         def _on_clear_all(self, *args):
 346                 with qui_utils.notify_error(self._app.errorLog):
 347                         self._history.clear()
 348
 349
 350 def run():
 351         app = QtGui.QApplication([])
 352         handle = Calculator(app)
 353         qtpie.init_pies()
 354         return app.exec_()
 355
 356
 357 if __name__ == "__main__":
 358         logging.basicConfig(level = logging.DEBUG)
 359         try:
 360                 os.makedirs(constants._data_path_)
 361         except OSError, e:
 362                 if e.errno != 17:
 363                         raise
 364
 365         val = run()
 366         sys.exit(val)