diff options
Diffstat (limited to 'invest-applet/invest/quotes.py')
| -rw-r--r-- | invest-applet/invest/quotes.py | 205 |
1 files changed, 176 insertions, 29 deletions
diff --git a/invest-applet/invest/quotes.py b/invest-applet/invest/quotes.py index c12397879..93b8b6880 100644 --- a/invest-applet/invest/quotes.py +++ b/invest-applet/invest/quotes.py @@ -8,14 +8,15 @@ import datetime from threading import Thread import invest, invest.about, invest.chart +import currencies CHUNK_SIZE = 512*1024 # 512 kB AUTOREFRESH_TIMEOUT = 15*60*1000 # 15 minutes -QUOTES_URL="http://finance.yahoo.com/d/quotes.csv?s=%(s)s&f=sl1d1t1c1ohgv&e=.csv" +QUOTES_URL="http://finance.yahoo.com/d/quotes.csv?s=%(s)s&f=snc4l1d1t1c1ohgv&e=.csv" -# Sample (25/4/2008): UCG.MI,"4,86",09:37:00,2008/04/25,"0,07","4,82","4,87","4,82",11192336 -QUOTES_CSV_FIELDS=["ticker", ("trade", float), "time", "date", ("variation", float), ("open", float)] +# Sample (09/2/2010): "UCG.MI","UNICREDIT","EUR","UCG.MI",1.9410,"2/9/2010","6:10am",+0.0210,1.9080,1.9810,1.8920,166691232 +QUOTES_CSV_FIELDS=["ticker", "label", "currency", ("trade", float), "date", "time", ("variation", float), ("open", float), ("high", float), ("low", float), ("volume", int)] # based on http://www.johnstowers.co.nz/blog/index.php/2007/03/12/threading-and-pygtk/ class _IdleObject(gobject.GObject): @@ -49,6 +50,7 @@ class QuotesRetriever(Thread, _IdleObject): self.tickers = tickers self.retrieved = False self.data = [] + self.currencies = [] def run(self): quotes_url = QUOTES_URL % {"s": self.tickers} @@ -68,9 +70,9 @@ class QuoteUpdater(gtk.ListStore): last_updated = None quotes_valid = False timeout_id = None - SYMBOL, LABEL, TICKER_ONLY, BALANCE, BALANCE_PCT, VALUE, VARIATION_PCT, PB = range(8) + SYMBOL, LABEL, CURRENCY, TICKER_ONLY, BALANCE, BALANCE_PCT, VALUE, VARIATION_PCT, PB = range(9) def __init__ (self, change_icon_callback, set_tooltip_callback): - gtk.ListStore.__init__ (self, gobject.TYPE_STRING, gobject.TYPE_STRING, bool, float, float, float, float, gtk.gdk.Pixbuf) + gtk.ListStore.__init__ (self, gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING, bool, float, float, float, float, gtk.gdk.Pixbuf) self.set_update_interval(AUTOREFRESH_TIMEOUT) self.change_icon_callback = change_icon_callback self.set_tooltip_callback = set_tooltip_callback @@ -134,16 +136,31 @@ class QuoteUpdater(gtk.ListStore): self.populate(self.parse_yahoo_csv(csv.reader(retriever.data))) self.updated = True self.last_updated = datetime.datetime.now() - tooltip = [] - if self.simple_quotes_count > 0: - # Translators: This is share-market jargon. It is the percentage change in the price of a stock. The %% gets changed to a single percent sign and the %s gets replaced with the string value of the change (localized), including the percent sign. - tooltip.append(_('Quotes average change %%: %s') % self.format_percent(self.avg_simple_quotes_change)) - if self.positions_count > 0: - # Translators: This is share-market jargon. It refers to the total difference between the current price and purchase price for all the shares put together. i.e. How much money would be earned if they were sold right now. - tooltip.append(_('Positions balance: %s') % self.format_difference(self.positions_balance)) - tooltip.append(_('Updated at %s') % self.last_updated.strftime("%H:%M")) - self.set_tooltip_callback('\n'.join(tooltip)) + self.update_tooltip() + def on_currency_retriever_completed(self, retriever): + if retriever.retrieved == False: + invest.error("Failed to retrieve currency rates!") + else: + self.convert_currencies(self.parse_yahoo_csv(csv.reader(retriever.data))) + self.update_tooltip() + + def update_tooltip(self): + tooltip = [] + if self.simple_quotes_count > 0: + # Translators: This is share-market jargon. It is the average percentage change of all stock prices. The %s gets replaced with the string value of the change (localized), including the percent sign. + tooltip.append(_('Average change: %s') % self.format_percent(self.avg_simple_quotes_change)) + for currency, stats in self.statistics.items(): + # get the statsitics + balance = stats["balance"] + paid = stats["paid"] + change = self.format_percent(balance / paid * 100) + balance = self.format_difference(balance) + + # Translators: This is share-market jargon. It refers to the total difference between the current price and purchase price for all the shares put together for a particular currency. i.e. How much money would be earned if they were sold right now. The first string is the change value, the second the currency, and the third value is the percentage of the change, formatted using user's locale. + tooltip.append(_('Positions balance: %s %s (%s)') % (balance, currency, change)) + tooltip.append(_('Updated at %s') % self.last_updated.strftime("%H:%M")) + self.set_tooltip_callback('\n'.join(tooltip)) def parse_yahoo_csv(self, csvreader): @@ -168,20 +185,54 @@ class QuoteUpdater(gtk.ListStore): result[fields[0]]['variation_pct'] = 0 return result + # Computes the balance of the given purchases using a certain current value + # and optionally a current exchange rate. + def balance(self, purchases, value, currentrate=None): + current = 0 + paid = 0 + + for purchase in purchases: + if purchase["amount"] != 0: + buyrate = purchase["exchange"] + # if the buy rate is invalid, use 1.0 + if buyrate <= 0: + buyrate = 1.0 + + # if no current rate is given, ignore buy rate + if currentrate == None: + buyrate = 1.0 + rate = 1.0 + else: + # otherwise, take use buy rate and current rate to compute the balance + rate = currentrate + + # current value is the current rate * amount * value + current += rate * purchase["amount"] * value + # paid is buy rate * ( amount * price + commission ) + paid += buyrate * (purchase["amount"] * purchase["bought"] + purchase["comission"]) + + balance = current - paid + if paid != 0: + change = 100*balance/paid + else: + change = 100 # Not technically correct, but it will look more intuitive than the real result of infinity. + + return (balance, change) + def populate(self, quotes): if (len(quotes) == 0): return self.clear() - + self.currencies = [] + try: quote_items = quotes.items () quote_items.sort () simple_quotes_change = 0 self.simple_quotes_count = 0 - self.positions_balance = 0 - self.positions_count = 0 + self.statistics = {} for ticker, val in quote_items: pb = None @@ -189,7 +240,31 @@ class QuoteUpdater(gtk.ListStore): # get the label of this stock for later reuse label = invest.STOCKS[ticker]["label"] if len(label) == 0: - label = ticker + if len(val["label"]) != 0: + label = val["label"] + else: + label = ticker + + # make sure the currency field is upper case + val["currency"] = val["currency"].upper(); + + # the currency of currency conversion rates like EURUSD=X is wrong in csv + # this can be fixed easily by reusing the latter currency in the symbol + if len(ticker) == 8 and ticker.endswith("=X"): + val["currency"] = ticker[3:6] + + # indices should not have a currency, though yahoo says so + if ticker.startswith("^"): + val["currency"] = "" + + # sometimes, funny currencies are returned (special characters), only consider known currencies + if len(val["currency"]) > 0 and val["currency"] not in currencies.Currencies.currencies: + invest.debug("Currency '%s' is not known, dropping" % val["currency"]) + val["currency"] = "" + + # if this is a currency not yet seen and different from the target currency, memorize it + if val["currency"] not in self.currencies and len(val["currency"]) > 0: + self.currencies.append(val["currency"]) # Check whether the symbol is a simple quote, or a portfolio value is_simple_quote = True @@ -200,19 +275,12 @@ class QuoteUpdater(gtk.ListStore): if is_simple_quote: self.simple_quotes_count += 1 - row = self.insert(0, [ticker, label, True, 0, 0, val["trade"], val["variation_pct"], pb]) + row = self.insert(0, [ticker, label, val["currency"], True, 0, 0, val["trade"], val["variation_pct"], pb]) simple_quotes_change += val['variation_pct'] else: - self.positions_count += 1 - current = sum([purchase["amount"]*val["trade"] for purchase in invest.STOCKS[ticker]["purchases"] if purchase["amount"] != 0]) - paid = sum([purchase["amount"]*purchase["bought"] + purchase["comission"] for purchase in invest.STOCKS[ticker]["purchases"] if purchase["amount"] != 0]) - balance = current - paid - if paid != 0: - change = 100*balance/paid - else: - change = 100 # Not technically correct, but it will look more intuitive than the real result of infinity. - row = self.insert(0, [ticker, label, False, balance, change, val["trade"], val["variation_pct"], pb]) - self.positions_balance += balance + (balance, change) = self.balance(invest.STOCKS[ticker]["purchases"], val["trade"]) + row = self.insert(0, [ticker, label, val["currency"], False, balance, change, val["trade"], val["variation_pct"], pb]) + self.add_balance_change(balance, change, val["currency"]) if len(ticker.split('.')) == 2: url = 'http://ichart.europe.yahoo.com/h?s=%s' % ticker @@ -248,6 +316,85 @@ class QuoteUpdater(gtk.ListStore): invest.debug(quotes) self.quotes_valid = False + # start retrieving currency conversion rates + if invest.CONFIG.has_key("currency"): + target_currency = invest.CONFIG["currency"] + symbols = [] + + invest.debug("These currencies occur: %s" % self.currencies) + for currency in self.currencies: + if currency == target_currency: + continue + + invest.debug("%s will be converted to %s" % ( currency, target_currency )) + symbol = currency + target_currency + "=X" + symbols.append(symbol) + + if len(symbols) > 0: + tickers = '+'.join(symbols) + quotes_retriever = QuotesRetriever(tickers) + quotes_retriever.connect("completed", self.on_currency_retriever_completed) + quotes_retriever.start() + + def convert_currencies(self, quotes): + # if there is no target currency, this method should never have been called + if not invest.CONFIG.has_key("currency"): + return + + # reset the overall balance + self.statistics = {} + + # collect the rates for the currencies + rates = {} + for symbol, data in quotes.items(): + currency = symbol[0:3] + rate = data["trade"] + rates[currency] = rate + + # convert all non target currencies + target_currency = invest.CONFIG["currency"] + iter = self.get_iter_first() + while iter != None: + currency = self.get_value(iter, self.CURRENCY) + symbol = self.get_value(iter, self.SYMBOL) + # ignore stocks that are currency conversions + # and only convert stocks that are not in the target currency + # and if we have a conversion rate + if not ( len(symbol) == 8 and symbol[6:8] == "=X" ) and \ + currency != target_currency and rates.has_key(currency): + # first convert the balance, it needs the original value + if not self.get_value(iter, self.TICKER_ONLY): + ticker = self.get_value(iter, self.SYMBOL) + value = self.get_value(iter, self.VALUE) + (balance, change) = self.balance(invest.STOCKS[ticker]["purchases"], value, rates[currency]) + self.set(iter, self.BALANCE, balance) + self.set(iter, self.BALANCE_PCT, change) + self.add_balance_change(balance, change, target_currency) + + # now, convert the value + value = self.get_value(iter, self.VALUE) + value *= rates[currency] + self.set(iter, self.VALUE, value) + self.set(iter, self.CURRENCY, target_currency) + + else: + # consider non-converted stocks here + balance = self.get_value(iter, self.BALANCE) + change = self.get_value(iter, self.BALANCE_PCT) + self.add_balance_change(balance, change, currency) + + iter = self.iter_next(iter) + + def add_balance_change(self, balance, change, currency): + if balance == 0 and change == 0: + return + + if self.statistics.has_key(currency): + self.statistics[currency]["balance"] += balance + self.statistics[currency]["paid"] += balance/change*100 + else: + self.statistics[currency] = { "balance" : balance, "paid" : balance/change*100 } + def set_pb_callback(self, retriever, row): self.set_value(row, self.PB, retriever.image.get_pixbuf()) |