diff options
Diffstat (limited to 'invest-applet/invest/widgets.py')
| -rw-r--r-- | invest-applet/invest/widgets.py | 263 |
1 files changed, 0 insertions, 263 deletions
diff --git a/invest-applet/invest/widgets.py b/invest-applet/invest/widgets.py deleted file mode 100644 index 8d15a5ed..00000000 --- a/invest-applet/invest/widgets.py +++ /dev/null @@ -1,263 +0,0 @@ -import os, time -from os.path import * - -import gi -gi.require_version("Gtk", "3.0") -from gi.repository import Gtk -from gi.repository import Gdk -from gi.repository import GObject -from gi.repository import Pango - -from gettext import gettext as _ -import locale -import csv -import mate_invest, mate_invest.about, mate_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_property("reorderable", True) - self.set_property("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, userdata): - label = model[iter][model.LABEL] - cell.set_property('text', label) - - def _getcelldata_value(self, column, cell, model, iter, userdata): - value = model[iter][model.VALUE]; - currency = model[iter][model.CURRENCY]; - if value == None or currency == None: - cell.set_property('text', "") - else: - cell.set_property('text', self.format_currency(value, 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, userdata): - 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, userdata): - 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, userdata): - 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 - - mate_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 = GdkPixbuf.Pixbuf(GdkPixbuf.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: - mate_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 mate_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) |