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.set_fullscreen(self._app.fullscreenAction.isChecked())
 222                 self.set_orientation(self._app.orientationAction.isChecked())
 223
 224         def walk_children(self):
 225                 return ()
 226
 227         def set_orientation(self, isPortrait):
 228                 qwrappers.WindowWrapper.set_orientation(self, isPortrait)
 229                 if isPortrait:
 230                         defaultLayoutOrientation = QtGui.QBoxLayout.TopToBottom
 231                         #tabPosition = QtGui.QTabWidget.South
 232                         tabPosition = QtGui.QTabWidget.West
 233                 else:
 234                         defaultLayoutOrientation = QtGui.QBoxLayout.LeftToRight
 235                         tabPosition = QtGui.QTabWidget.North
 236                 self._layout.setDirection(defaultLayoutOrientation)
 237                 self._keyboardTabs.setTabPosition(tabPosition)
 238
 239         def enable_plugin(self, pluginId):
 240                 self._keyboardPlugins.enable_plugin(pluginId)
 241                 pluginData = self._keyboardPlugins.plugin_info(pluginId)
 242                 pluginName = pluginData[0]
 243                 plugin = self._keyboardPlugins.keyboards[pluginName].construct_keyboard()
 244                 relIcon = self._keyboardPlugins.keyboards[pluginName].icon
 245                 for iconPath in self._keyboardPlugins.keyboards[pluginName].iconPaths:
 246                         absIconPath = os.path.join(iconPath, relIcon)
 247                         if os.path.exists(absIconPath):
 248                                 icon = QtGui.QIcon(absIconPath)
 249                                 break
 250                 else:
 251                         icon = None
 252                 pluginKeyboard = plugin.setup(self._history, self._handler)
 253
 254                 self._activeKeyboards.append({
 255                         "pluginName": pluginName,
 256                         "plugin": plugin,
 257                         "pluginKeyboard": pluginKeyboard,
 258                 })
 259                 if icon is None:
 260                         self._keyboardTabs.addTab(pluginKeyboard.toplevel, pluginName)
 261                 else:
 262                         self._keyboardTabs.addTab(pluginKeyboard.toplevel, icon, "")
 263
 264         def close(self):
 265                 qwrappers.WindowWrapper.close(self)
 266                 self._save_history()
 267
 268         def _load_history(self):
 269                 serialized = []
 270                 try:
 271                         with open(self._user_history, "rU") as f:
 272                                 serialized = (
 273                                         (part.strip() for part in line.split(" "))
 274                                         for line in f.readlines()
 275                                 )
 276                 except IOError, e:
 277                         if e.errno != 2:
 278                                 raise
 279                 self._history.deserialize_stack(serialized)
 280
 281         def _save_history(self):
 282                 serialized = self._history.serialize_stack()
 283                 with open(self._user_history, "w") as f:
 284                         for lineData in serialized:
 285                                 line = " ".join(data for data in lineData)
 286                                 f.write("%s\n" % line)
 287
 288         @misc_utils.log_exception(_moduleLogger)
 289         def _on_child_close(self, something = None):
 290                 with qui_utils.notify_error(self._app.errorLog):
 291                         self._child = None
 292
 293         @misc_utils.log_exception(_moduleLogger)
 294         def _on_copy(self, *args):
 295                 with qui_utils.notify_error(self._app.errorLog):
 296                         eqNode = self._historyView.peek()
 297                         resultNode = eqNode.simplify()
 298                         self._app._clipboard.setText(str(resultNode))
 299
 300         @misc_utils.log_exception(_moduleLogger)
 301         def _on_paste(self, *args):
 302                 with qui_utils.notify_error(self._app.errorLog):
 303                         result = str(self._app._clipboard.text())
 304                         self._userEntry.append(result)
 305
 306         @misc_utils.log_exception(_moduleLogger)
 307         def _on_entry_direct(self, keys, modifiers):
 308                 with qui_utils.notify_error(self._app.errorLog):
 309                         if "shift" in modifiers:
 310                                 keys = keys.upper()
 311                         self._userEntry.append(keys)
 312
 313         @misc_utils.log_exception(_moduleLogger)
 314         def _on_push(self, *args):
 315                 with qui_utils.notify_error(self._app.errorLog):
 316                         self._history.push_entry()
 317
 318         @misc_utils.log_exception(_moduleLogger)
 319         def _on_unpush(self, *args):
 320                 with qui_utils.notify_error(self._app.errorLog):
 321                         self._historyView.unpush()
 322
 323         @misc_utils.log_exception(_moduleLogger)
 324         def _on_entry_backspace(self, *args):
 325                 with qui_utils.notify_error(self._app.errorLog):
 326                         self._userEntry.pop()
 327
 328         @misc_utils.log_exception(_moduleLogger)
 329         def _on_entry_clear(self, *args):
 330                 with qui_utils.notify_error(self._app.errorLog):
 331                         self._userEntry.clear()
 332
 333         @misc_utils.log_exception(_moduleLogger)
 334         def _on_clear_all(self, *args):
 335                 with qui_utils.notify_error(self._app.errorLog):
 336                         self._history.clear()
 337
 338
 339 def run():
 340         app = QtGui.QApplication([])
 341         handle = Calculator(app)
 342         qtpie.init_pies()
 343         return app.exec_()
 344
 345
 346 if __name__ == "__main__":
 347         logging.basicConfig(level = logging.DEBUG)
 348         try:
 349                 os.makedirs(constants._data_path_)
 350         except OSError, e:
 351                 if e.errno != 17:
 352                         raise
 353
 354         val = run()
 355         sys.exit(val)