initial

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)