1 files changed, 287 insertions, 0 deletions
diff --git a/invest-applet/invest/preferences.py b/invest-applet/invest/preferences.py
new file mode 100644
index 00000000..3c48ee42
--- /dev/null
+++ b/invest-applet/invest/preferences.py
@@ -0,0 +1,287 @@
+from gettext import gettext as _
+import locale
+from os.path import join
+import gtk, gobject, mateconf
+import invest
+import currencies
+import cPickle
+
+class PrefsDialog:
+ def __init__(self, applet):
+ self.ui = gtk.Builder()
+ self.ui.add_from_file(join(invest.BUILDER_DATA_DIR, "prefs-dialog.ui"))
+
+ self.dialog = self.ui.get_object("preferences")
+ self.treeview = self.ui.get_object("stocks")
+ self.currency = self.ui.get_object("currency")
+ self.currency_code = None
+ self.currencies = currencies.Currencies.currencies
+
+ self.ui.get_object("add").connect('clicked', self.on_add_stock)
+ self.ui.get_object("add").connect('activate', self.on_add_stock)
+ self.ui.get_object("remove").connect('clicked', self.on_remove_stock)
+ self.ui.get_object("remove").connect('activate', self.on_remove_stock)
+ self.ui.get_object("help").connect('clicked', self.on_help)
+ self.treeview.connect('key-press-event', self.on_tree_keypress)
+ self.currency.connect('key-press-event', self.on_entry_keypress)
+ self.currency.connect('activate', self.on_activate_entry)
+ self.currency.connect('focus-out-event', self.on_focus_out_entry)
+
+ self.typs = (str, str, float, float, float, float)
+ self.names = (_("Symbol"), _("Label"), _("Amount"), _("Price"), _("Commission"), _("Currency Rate"))
+ store = gtk.ListStore(*self.typs)
+ store.set_sort_column_id(0, gtk.SORT_ASCENDING)
+ self.treeview.set_model(store)
+ self.model = store
+
+ completion = gtk.EntryCompletion()
+ self.currency.set_completion(completion)
+ liststore = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
+ completion.set_model(liststore)
+ completion.set_text_column(0)
+ for code, label in self.currencies.items():
+ liststore.append([self.format_currency(label, code), code])
+ completion.set_match_func(self.match_func, 0)
+ completion.connect("match-selected", self.on_completion_selection, 1)
+
+ if invest.CONFIG.has_key("currency"):
+ code = invest.CONFIG["currency"];
+ if self.currencies.has_key(code):
+ self.currency_code = code;
+ currency = self.format_currency(self.currencies[self.currency_code], self.currency_code)
+ self.currency.set_text(currency)
+
+ for n in xrange (0, 5):
+ self.create_cell (self.treeview, n, self.names[n], self.typs[n])
+ if self.currency_code != None:
+ self.add_exchange_column()
+
+ stock_items = invest.STOCKS.items ()
+ stock_items.sort ()
+ for key, data in stock_items:
+ label = data["label"]
+ purchases = data["purchases"]
+ for purchase in purchases:
+ if purchase.has_key("exchange"):
+ exchange = purchase["exchange"]
+ else:
+ exchange = 0.0
+ store.append([key, label, purchase["amount"], purchase["bought"], purchase["comission"], exchange])
+
+ self.sync_ui()
+
+ def on_cell_edited(self, cell, path, new_text, col, typ):
+ try:
+ if col == 0: # stock symbols must be uppercase
+ new_text = str.upper(new_text)
+ if col < 2:
+ self.model[path][col] = new_text
+ else:
+ value = locale.atof(new_text)
+ self.model[path][col] = value
+ except Exception, msg:
+ invest.error('Exception while processing cell change: %s' % msg)
+ pass
+
+ def format(self, fmt, value):
+ return locale.format(fmt, value, True)
+
+ def get_cell_data(self, column, cell, model, iter, data):
+ typ, col = data
+ if typ == int:
+ cell.set_property('text', "%d" % typ(model[iter][col]))
+ elif typ == float:
+ # provide float numbers with at least 2 fractional digits
+ val = model[iter][col]
+ digits = self.fraction_digits(val)
+ fmt = "%%.%df" % max(digits, 2)
+ cell.set_property('text', self.format(fmt, val))
+ else:
+ cell.set_property('text', typ(model[iter][col]))
+
+ # determine the number of non zero digits in the fraction of the value
+ def fraction_digits(self, value):
+ text = "%g" % value # do not use locale here, so that %g always is rendered to a number with . as decimal separator
+ if text.find(".") < 0:
+ return 0
+ return len(text) - text.find(".") - 1
+
+ def create_cell (self, view, column, name, typ):
+ cell_description = gtk.CellRendererText ()
+ if typ == float:
+ cell_description.set_property("xalign", 1.0)
+ cell_description.set_property("editable", True)
+ cell_description.connect("edited", self.on_cell_edited, column, typ)
+ column_description = gtk.TreeViewColumn (name, cell_description)
+ if typ == str:
+ column_description.set_attributes (cell_description, text=column)
+ column_description.set_sort_column_id(column)
+ if typ == float:
+ column_description.set_cell_data_func(cell_description, self.get_cell_data, (float, column))
+ view.append_column(column_description)
+
+ def add_exchange_column(self):
+ self.create_cell (self.treeview, 5, self.names[5], self.typs[5])
+
+ def remove_exchange_column(self):
+ column = self.treeview.get_column(5)
+ self.treeview.remove_column(column)
+
+ def show_run_hide(self, explanation = ""):
+ expl = self.ui.get_object("explanation")
+ expl.set_markup(explanation)
+ self.dialog.show_all()
+ if explanation == "":
+ expl.hide()
+ # returns 1 if help is clicked
+ while self.dialog.run() == 1:
+ pass
+ self.dialog.destroy()
+
+ invest.STOCKS = {}
+
+ def save_symbol(model, path, iter):
+ #if int(model[iter][1]) == 0 or float(model[iter][2]) < 0.0001:
+ # return
+
+ if not model[iter][0] in invest.STOCKS:
+ invest.STOCKS[model[iter][0]] = { 'label': model[iter][1], 'purchases': [] }
+ 
+ invest.STOCKS[model[iter][0]]["purchases"].append({
+ "amount": float(model[iter][2]),
+ "bought": float(model[iter][3]),
+ "comission": float(model[iter][4]),
+ "exchange": float(model[iter][5])
+ })
+ self.model.foreach(save_symbol)
+ try:
+ cPickle.dump(invest.STOCKS, file(invest.STOCKS_FILE, 'w'))
+ invest.debug('Stocks written to file')
+ except Exception, msg:
+ invest.error('Could not save stocks file: %s' % msg)
+
+ invest.CONFIG = {}
+ if self.currency_code != None and len(self.currency_code) == 3:
+ invest.CONFIG['currency'] = self.currency_code
+ try:
+ cPickle.dump(invest.CONFIG, file(invest.CONFIG_FILE, 'w'))
+ invest.debug('Configuration written to file')
+ except Exception, msg:
+ invest.debug('Could not save configuration file: %s' % msg)
+
+ def sync_ui(self):
+ pass
+
+ def on_add_stock(self, w):
+ iter = self.model.append(["GOOG", "Google Inc.", 0, 0, 0, 0])
+ path = self.model.get_path(iter)
+ self.treeview.set_cursor(path, self.treeview.get_column(0), True)
+
+ def on_remove_stock(self, w):
+ model, paths = self.treeview.get_selection().get_selected_rows()
+ for path in paths:
+ model.remove(model.get_iter(path))
+
+ def on_help(self, w):
+ invest.help.show_help_section("invest-applet-usage")
+
+ def on_tree_keypress(self, w, event):
+ if event.keyval == 65535:
+ self.on_remove_stock(w)
+
+ return False
+
+ def format_currency(self, label, code):
+ if code == None:
+ return label
+ return label + " (" + code + ")"
+
+ def on_entry_keypress(self, w, event):
+ # enter key was pressed
+ if event.keyval == 65293:
+ self.match_currency()
+ return False
+
+ # entry was activated (Enter)
+ def on_activate_entry(self, entry):
+ self.match_currency()
+
+ # entry left focus
+ def on_focus_out_entry(self, w, event):
+ self.match_currency()
+ return False
+
+ # tries to find a currency for the text in the currency entry
+ def match_currency(self):
+ # get the text
+ text = self.currency.get_text().upper()
+
+ # if there is none, finish
+ if len(text) == 0:
+ self.currency_code = None
+ self.pick_currency(None)
+ return
+
+ # if it is a currency code, take that one
+ if len(text) == 3:
+ # try to find the string as code
+ if self.currencies.has_key(text):
+ self.pick_currency(text)
+ return
+ else:
+ # try to find the code for the string
+ for code, label in self.currencies.items():
+ # if the entry equals to the full label,
+ # or the entry equals to the label+code concat
+ if label.upper() == text or self.format_currency(label.upper(), code) == text:
+ # then we take that code
+ self.pick_currency(code)
+ return
+
+ # the entry is not valid, reuse the old one
+ self.pick_currency(self.currency_code)
+
+ # pick this currency, stores it and sets the entry text
+ def pick_currency(self, code):
+ if code == None:
+ label = ""
+ if len(self.treeview.get_columns()) == 6:
+ self.remove_exchange_column()
+ else:
+ label = self.currencies[code]
+ if len(self.treeview.get_columns()) == 5:
+ self.add_exchange_column()
+ self.currency.set_text(self.format_currency(label, code))
+ self.currency_code = code
+
+ # finds matches by testing candidate strings to have tokens starting with the entered text's tokens
+ def match_func(self, completion, key, iter, column):
+ keys = key.split()
+ model = completion.get_model()
+ text = model.get_value(iter, column).lower()
+ tokens = text.split()
+
+ # each key must have a match
+ for key in keys:
+ found_key = False
+ # check any token of the completions start with the key
+ for token in tokens:
+ # remove the ( from the currency code
+ if token.startswith("("):
+ token = token[1:]
+ if token.startswith(key):
+ found_key = True
+ break
+ # this key does not have a match, this is not a completion
+ if not found_key:
+ return False
+ # all keys matched, this is a completion
+ return True
+
+ # stores the selected currency code
+ def on_completion_selection(self, completion, model, iter, column):
+ self.pick_currency(model.get_value(iter, column))
+
+
+def show_preferences(applet, explanation = ""):
+ PrefsDialog(applet).show_run_hide(explanation)