diff options
Diffstat (limited to 'invest-applet/invest/widgets.py')
| -rw-r--r-- | invest-applet/invest/widgets.py | 250 |
1 files changed, 250 insertions, 0 deletions
diff --git a/invest-applet/invest/widgets.py b/invest-applet/invest/widgets.py new file mode 100644 index 00000000..3f5bc91b --- /dev/null +++ b/invest-applet/invest/widgets.py @@ -0,0 +1,250 @@ +import os, time +from os.path import * +import mateapplet, gtk, gtk.gdk, mateconf, gobject, pango +from gettext import gettext as _ +import locale +import csv +import invest, invest.about, invest.chart + +COLORSCALE_POSITIVE = [ + "white", + "#ad7fa8", + "#75507b", + "#5c3566", + "#729fcf", + "#3465a4", + "#204a87", + "#8ae234", + "#73d216", + "#4e9a06", +] +COLORSCALE_NEGATIVE = [ + "white", + "#fce94f", + "#e9b96e", + "#fcaf3e", + "#c17d11", + "#f57900", + "#ce5c00", + "#ef2929", + "#cc0000", + "#a40000", +] +LIGHT = -3 +MEDIUM = -1 + +TICKER_TIMEOUT = 10000#3*60*1000# + +class InvestWidget(gtk.TreeView): + def __init__(self, quotes_updater): + gtk.TreeView.__init__(self) + self.set_property("rules-hint", True) + self.set_reorderable(True) + self.set_hover_selection(True) + + simple_quotes_only = quotes_updater.simple_quotes_only() + + # model: SYMBOL, LABEL, TICKER_ONLY, BALANCE, BALANCE_PCT, VALUE, VARIATION_PCT, PB + # Translators: these words all refer to a stock. Last is short + # for "last price". Gain is referring to the gain since the + # stock was purchased. + col_names = [_('Ticker'), _('Last'), _('Change %'), _('Chart'), _('Gain'), _('Gain %')] + col_cellgetdata_functions = [self._getcelldata_label, self._getcelldata_value, + self._getcelldata_variation, None, self._getcelldata_balance, + self._getcelldata_balancepct] + for i, col_name in enumerate(col_names): + if i < 3: + cell = gtk.CellRendererText() + if i > 0: + cell.set_property("xalign", 1.0) + column = gtk.TreeViewColumn (col_name, cell) + if i == 0: + column.set_sort_column_id(quotes_updater.LABEL) + elif i == 2: + column.set_sort_column_id(quotes_updater.VARIATION_PCT) + column.set_cell_data_func(cell, col_cellgetdata_functions[i]) + self.append_column(column) + elif i == 3: + cell_pb = gtk.CellRendererPixbuf() + column = gtk.TreeViewColumn (col_name, cell_pb, pixbuf=quotes_updater.PB) + self.append_column(column) + else: + # add the last two column only if we have any positions + if simple_quotes_only == False: + cell = gtk.CellRendererText() + cell.set_property("xalign", 1.0) + column = gtk.TreeViewColumn (col_name, cell) + if i == 4: + column.set_sort_column_id(quotes_updater.BALANCE) + elif i == 5: + column.set_sort_column_id(quotes_updater.BALANCE_PCT) + column.set_cell_data_func(cell, col_cellgetdata_functions[i]) + self.append_column(column) + + if simple_quotes_only == True: + self.set_property('headers-visible', False) + + self.connect('row-activated', self.on_row_activated) + self.set_model(quotes_updater) + + + # locale-aware formatting of the value as monetary, without currency symbol, using 2 decimal digits + def format_currency(self, value, currency): + return locale.format("%.2f", value, True, True) + " " + currency + + # locale-aware formatting of the percent float (decimal point, thousand grouping point) with 2 decimal digits + def format_percent(self, value): + return locale.format("%+.2f", value, True) + "%" + + # locale-aware formatting of the float value (decimal point, thousand grouping point) with sign and 2 decimal digits + def format_difference(self, value, currency): + return locale.format("%+.2f", value, True, True) + " " + currency + + + def _getcelldata_label(self, column, cell, model, iter): + cell.set_property('text', model[iter][model.LABEL]) + + def _getcelldata_value(self, column, cell, model, iter): + cell.set_property('text', self.format_currency(model[iter][model.VALUE], model[iter][model.CURRENCY])) + + def is_selected(self, model, iter): + m, it = self.get_selection().get_selected() + return it != None and model.get_path(iter) == m.get_path(it) + + def get_color(self, model, iter, field): + palette = COLORSCALE_POSITIVE + intensity = MEDIUM + if model[iter][field] < 0: + palette = COLORSCALE_NEGATIVE + if self.is_selected(model, iter): + intensity = LIGHT + return palette[intensity] + + def _getcelldata_variation(self, column, cell, model, iter): + color = self.get_color(model, iter, model.VARIATION_PCT) + change_pct = self.format_percent(model[iter][model.VARIATION_PCT]) + cell.set_property('markup', + "<span foreground='%s'>%s</span>" % + (color, change_pct)) + + def _getcelldata_balance(self, column, cell, model, iter): + is_ticker_only = model[iter][model.TICKER_ONLY] + color = self.get_color(model, iter, model.BALANCE) + if is_ticker_only: + cell.set_property('text', '') + else: + balance = self.format_difference(model[iter][model.BALANCE], model[iter][model.CURRENCY]) + cell.set_property('markup', + "<span foreground='%s'>%s</span>" % + (color, balance)) + + def _getcelldata_balancepct(self, column, cell, model, iter): + is_ticker_only = model[iter][model.TICKER_ONLY] + color = self.get_color(model, iter, model.BALANCE_PCT) + if is_ticker_only: + cell.set_property('text', '') + else: + balance_pct = self.format_percent(model[iter][model.BALANCE_PCT]) + cell.set_property('markup', + "<span foreground='%s'>%s</span>" % + (color, balance_pct)) + + def on_row_activated(self, treeview, path, view_column): + ticker = self.get_model()[self.get_model().get_iter(path)][0] + if ticker == None: + return + + invest.chart.show_chart([ticker]) + +#class InvestTicker(gtk.Label): +# def __init__(self): +# gtk.Label.__init__(self, _("Waiting...")) +# +# self.quotes = [] +# gobject.timeout_add(TICKER_TIMEOUT, self.scroll_quotes) +# +# get_quotes_updater().connect('quotes-updated', self.on_quotes_update) +# +# def on_quotes_update(self, updater): +# self.quotes = [] +# updater.foreach(self.update_quote, None) +# +# def update_quote(self, model, path, iter, user_data): +# color = GREEN +# if model[iter][model.BALANCE] < 0: +# color = RED +# +# self.quotes.append( +# "%s: <span foreground='%s'>%+.2f (%+.2f%%)</span> %.2f" % +# (model[iter][model.SYMBOL], color, model[iter][model.BALANCE], model[iter][model.BALANCE_PCT], model[iter][model.VALUE])) +# +# def scroll_quotes(self): +# if len(self.quotes) == 0: +# return True +# +# q = self.quotes.pop() +# self.set_markup("<span face='Monospace'>%s</span>" % q) +# self.quotes.insert(0, q) +# +# return True +# +#gobject.type_register(InvestTicker) + +class InvestTrend(gtk.Image): + def __init__(self): + gtk.Image.__init__(self) + self.pixbuf = None + self.previous_allocation = (0,0) + self.connect('size-allocate', self.on_size_allocate) + get_quotes_updater().connect('quotes-updated', self.on_quotes_update) + + def on_size_allocate(self, widget, allocation): + if self.previous_allocation == (allocation.width, allocation.height): + return + + self.pixbuf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, True, 8, allocation.height, allocation.height) + self.set_color("grey") + self.previous_allocation = (allocation.width, allocation.height) + + def set_color(self, color, opacity=0xFF): + if self.pixbuf != None: + try: + color = pango.Color(color) + factor = float(0xFF)/0xFFFF + self.pixbuf.fill( + int(color.red*factor)<<24|int(color.green*factor)<<16|int(color.blue*factor)<<8|opacity) + self.set_from_pixbuf(self.pixbuf) + except Exception, msg: + invest.error("Could not set color: %s" % msg) + + def on_quotes_update(self, updater): + start_total = 0 + now_total = 0 + for row in updater: + # Don't count the ticker only symbols in the color-trend + if row[updater.TICKER_ONLY]: + continue + + var = row[updater.VARIATION]/100 + now = row[updater.VALUE] + + start = now / (1 + var) + + portfolio_number = sum([purchase["amount"] for purchase in invest.STOCKS[row[updater.SYMBOL]]["purchases"]]) + start_total += start * portfolio_number + now_total += now * portfolio_number + + day_var = 0 + if start_total != 0: + day_var = (now_total - start_total) / start_total * 100 + + color = int(2*day_var) + opacity = min(0xFF, int(abs(127.5*day_var))) + if day_var < 0: + color = COLORSCALE_NEGATIVE[min(len(COLORSCALE_NEGATIVE)-1, abs(color))] + else: + color = COLORSCALE_POSITIVE[min(len(COLORSCALE_POSITIVE)-1, abs(color))] + + self.set_color(color, opacity) + +gobject.type_register(InvestTrend) |